feat: setQueryData after collecting and uncollecting

RSSNext · May 22, 2024 · 02573a6 · 02573a6
1 parent b3e4270
commit 02573a6
Showing 1 changed file with 14 additions and 6 deletions.
diff --git a/src/renderer/src/hooks/useEntryActions.tsx b/src/renderer/src/hooks/useEntryActions.tsx
@@ -35,9 +35,14 @@ export const useEntryActions = ({
         },
       }),
     onSuccess: () => {
-      queryClient.invalidateQueries({
-        queryKey: ["entries"],
-      })
+      queryClient.setQueryData(
+        ["entry", entry.id],
+        Object.assign({}, entry, {
+          collections: {
+            createdAt: new Date().toISOString(),
+          },
+        }),
+      )
       toast({
         duration: 1000,
         description: "Collected.",
@@ -53,9 +58,12 @@ export const useEntryActions = ({
         },
       }),
     onSuccess: () => {
-      queryClient.invalidateQueries({
-        queryKey: ["entries"],
-      })
+      queryClient.setQueryData(
+        ["entry", entry.id],
+        Object.assign({}, entry, {
+          collections: null,
+        }),
+      )
       toast({
         duration: 1000,
         description: "Uncollected.",