fix: comment reply mutation

Signed-off-by: Innei <i@innei.in>
Innei · Jan 21, 2024 · bbd6f90 · bbd6f90
1 parent 59d1944
commit bbd6f90
Showing 1 changed file with 15 additions and 4 deletions.
diff --git a/src/queries/definition/comment.ts b/src/queries/definition/comment.ts
@@ -109,13 +109,24 @@ export const useReplyCommentMutation = () => {
         },
       })
     },
-    onSuccess: () => {
-      queryClient.invalidateQueries({
-        queryKey: commentAdmin.byState(state).queryKey,
-      })
+    onMutate({ id }) {
+      queryClient.setQueryData<InfiniteData<PaginateResult<CommentModel>>>(
+        commentAdmin.byState(state).queryKey,
+        produce((draft) => {
+          draft?.pages.forEach((page) => {
+            page.data = page.data.filter((comment) => comment.id !== id)
+          })
+        }),
+      )
+    },
+    onSuccess: (_) => {
       toast.success('回复成功')
     },
     onError: () => {
+      queryClient.invalidateQueries({
+        queryKey: commentAdmin.byState(state).queryKey,
+      })
+
       toast.error('回复失败')
     },
   })