fix(useSetState): memoize setState callback

streamich · Aug 23, 2019 · 16f023f · 16f023f
1 parent c73e92f
commit 16f023f
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/src/useSetState.ts b/src/useSetState.ts
@@ -1,12 +1,15 @@
-import { useState } from 'react';
+import { useState, useCallback } from 'react';
 
 const useSetState = <T extends object>(
   initialState: T = {} as T
 ): [T, (patch: Partial<T> | ((prevState: T) => Partial<T>)) => void] => {
   const [state, set] = useState<T>(initialState);
-  const setState = patch => {
-    set(prevState => Object.assign({}, prevState, patch instanceof Function ? patch(prevState) : patch));
-  };
+  const setState = useCallback(
+    patch => {
+      set(prevState => Object.assign({}, prevState, patch instanceof Function ? patch(prevState) : patch));
+    },
+    [set]
+  );
 
   return [state, setState];
 };