⚡️ Make useMutation reset use batched updates

reduxjs · Oct 16, 2021 · f2c3606 · f2c3606
1 parent 8b62c14
commit f2c3606
Showing 1 changed file with 13 additions and 11 deletions.
diff --git a/packages/toolkit/src/query/react/buildHooks.ts b/packages/toolkit/src/query/react/buildHooks.ts
@@ -825,17 +825,19 @@ export function buildHooks<Definitions extends EndpointDefinitions>({
       const originalArgs =
         fixedCacheKey == null ? promise?.arg.originalArgs : undefined
       const reset = useCallback(() => {
-        if (promise) {
-          setPromise(undefined)
-        }
-        if (fixedCacheKey) {
-          dispatch(
-            api.internalActions.removeMutationResult({
-              requestId,
-              fixedCacheKey,
-            })
-          )
-        }
+        batch(() => {
+          if (promise) {
+            setPromise(undefined)
+          }
+          if (fixedCacheKey) {
+            dispatch(
+              api.internalActions.removeMutationResult({
+                requestId,
+                fixedCacheKey,
+              })
+            )
+          }
+        })
       }, [dispatch, fixedCacheKey, promise, requestId])
       const finalState = useMemo(
         () => ({ ...currentState, originalArgs, reset }),