Merge pull request #89 from peterferguson/fix-atom-with-mutation-stat…

…e-unsubscribe fix: `atomWithMutationState` should unsubscribe on unmount
jotaijs · Aug 18, 2024 · 7d6ee8e · 7d6ee8e
2 parents 4171f5d + 4a9f618
commit 7d6ee8e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/atomWithMutationState.ts b/src/atomWithMutationState.ts
@@ -46,9 +46,10 @@ export const atomWithMutationState = <TResult = MutationState>(
 
     const mutationCache = queryClient.getMutationCache()
     resultsAtom.onMount = (set) => {
-      mutationCache.subscribe(() => {
+      const unsubscribe = mutationCache.subscribe(() => {
         set(getResult(getQueryClient(get).getMutationCache(), getOptions(get)))
       })
+      return unsubscribe
     }
   })
   if (process.env.NODE_ENV !== 'production') {