apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-05-03 21:02:40 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

feat: add logs cache keys

Browse source
This commit is contained in:
CREDO23 2025-12-26 17:05:06 +02:00
parent b0214c901d
commit 10f3baca0c
2 changed files with 55 additions and 30 deletions
Download patch file Download diff file Expand all files Collapse all files

80
surfsense_web/atoms/logs/log-mutation.atoms.ts
View file

@ -1,10 +1,9 @@
import { atomWithMutation } from "jotai-tanstack-query"; import { atomWithMutation } from "jotai-tanstack-query";
import { toast } from "sonner"; import { activeSearchSpaceIdAtom } from "@/atoms/search-spaces/search-space-query.atoms";
import type { import type {
CreateLogRequest, CreateLogRequest,
DeleteLogRequest, DeleteLogRequest,
UpdateLogRequest, UpdateLogRequest,
GetLogSummaryRequest,
} from "@/contracts/types/log.types"; } from "@/contracts/types/log.types";
import { logsApiService } from "@/lib/apis/logs-api.service"; import { logsApiService } from "@/lib/apis/logs-api.service";
import { cacheKeys } from "@/lib/query-client/cache-keys"; import { cacheKeys } from "@/lib/query-client/cache-keys";
@ -13,36 +12,57 @@ import { queryClient } from "@/lib/query-client/client";
/** /**
* Create Log Mutation * Create Log Mutation
*/ */
export const createLogMutationAtom = atomWithMutation(() => ({ export const createLogMutationAtom = atomWithMutation((get) => {
mutationKey: cacheKeys.logs.create(), const searchSpaceId = get(activeSearchSpaceIdAtom);
mutationFn: async (request: CreateLogRequest) => logsApiService.createLog(request), return {
onSuccess: () => { mutationKey: cacheKeys.logs.list(searchSpaceId ?? undefined),
queryClient.invalidateQueries({ queryKey: cacheKeys.logs.list() }); enabled: !!searchSpaceId,
}, mutationFn: async (request: CreateLogRequest) => logsApiService.createLog(request),
})); onSuccess: () => {
queryClient.invalidateQueries({ queryKey: cacheKeys.logs.list(searchSpaceId ?? undefined) });
queryClient.invalidateQueries({
queryKey: cacheKeys.logs.summary(searchSpaceId ?? undefined),
});
},
};
});
/** /**
* Update Log Mutation * Update Log Mutation
*/ */
export const updateLogMutationAtom = atomWithMutation(() => ({ export const updateLogMutationAtom = atomWithMutation((get) => {
mutationKey: cacheKeys.logs.update(), const searchSpaceId = get(activeSearchSpaceIdAtom);
mutationFn: async ({ logId, data }: { logId: number; data: UpdateLogRequest }) => return {
logsApiService.updateLog(logId, data), mutationKey: cacheKeys.logs.list(searchSpaceId ?? undefined),
onSuccess: (_data, variables) => { enabled: !!searchSpaceId,
queryClient.invalidateQueries({ queryKey: cacheKeys.logs.detail(variables.logId) }); mutationFn: async ({ logId, data }: { logId: number; data: UpdateLogRequest }) =>
queryClient.invalidateQueries({ queryKey: cacheKeys.logs.list() }); logsApiService.updateLog(logId, data),
}, onSuccess: (_data, variables) => {
})); queryClient.invalidateQueries({ queryKey: cacheKeys.logs.detail(variables.logId) });
queryClient.invalidateQueries({ queryKey: cacheKeys.logs.list(searchSpaceId ?? undefined) });
queryClient.invalidateQueries({
queryKey: cacheKeys.logs.summary(searchSpaceId ?? undefined),
});
},
};
});
/** /**
* Delete Log Mutation * Delete Log Mutation
*/ */
export const deleteLogMutationAtom = atomWithMutation(() => ({ export const deleteLogMutationAtom = atomWithMutation((get) => {
mutationKey: cacheKeys.logs.delete(), const searchSpaceId = get(activeSearchSpaceIdAtom);
mutationFn: async (request: DeleteLogRequest) => logsApiService.deleteLog(request), return {
onSuccess: (_data, request) => { mutationKey: cacheKeys.logs.list(searchSpaceId ?? undefined),
queryClient.invalidateQueries({ queryKey: cacheKeys.logs.list() }); enabled: !!searchSpaceId,
if (request?.id) queryClient.invalidateQueries({ queryKey: cacheKeys.logs.detail(request.id) }); mutationFn: async (request: DeleteLogRequest) => logsApiService.deleteLog(request),
}, onSuccess: (_data, request) => {
})); queryClient.invalidateQueries({ queryKey: cacheKeys.logs.list(searchSpaceId ?? undefined) });
queryClient.invalidateQueries({
queryKey: cacheKeys.logs.summary(searchSpaceId ?? undefined),
});
if (request?.id)
queryClient.invalidateQueries({ queryKey: cacheKeys.logs.detail(request.id) });
},
};
});

5
surfsense_web/lib/query-client/cache-keys.ts
View file

@ -18,6 +18,11 @@ export const cacheKeys = {
typeCounts: (searchSpaceId?: string) => ["documents", "type-counts", searchSpaceId] as const, typeCounts: (searchSpaceId?: string) => ["documents", "type-counts", searchSpaceId] as const,
byChunk: (chunkId: string) => ["documents", "by-chunk", chunkId] as const, byChunk: (chunkId: string) => ["documents", "by-chunk", chunkId] as const,
}, },
logs: {
list: (searchSpaceId?: number | string) => ["logs", "list", searchSpaceId] as const,
detail: (logId: number | string) => ["logs", "detail", logId] as const,
summary: (searchSpaceId?: number | string) => ["logs", "summary", searchSpaceId] as const,
},
newLLMConfigs: { newLLMConfigs: {
all: (searchSpaceId: number) => ["new-llm-configs", searchSpaceId] as const, all: (searchSpaceId: number) => ["new-llm-configs", searchSpaceId] as const,
byId: (configId: number) => ["new-llm-configs", "detail", configId] as const, byId: (configId: number) => ["new-llm-configs", "detail", configId] as const,