fix: extract common function for set operations

alibaba · Oct 23, 2024 · ed3eda8 · ed3eda8
1 parent c7bb04c
commit ed3eda8
Showing 1 changed file with 10 additions and 8 deletions.
diff --git a/packages/hooks/src/useSet/index.ts b/packages/hooks/src/useSet/index.ts
@@ -5,25 +5,27 @@ function useSet<K>(initialValue?: Iterable<K>) {
   const getInitValue = () => new Set(initialValue);
   const [set, setSet] = useState<Set<K>>(getInitValue);
 
+  const updateSet = (updater: (set: Set<K>) => Set<K>) => {
+    setSet((prevSet) => updater(new Set(prevSet)));
+  };
+
   const add = (key: K) => {
     if (set.has(key)) {
       return;
     }
-    setSet((prevSet) => {
-      const temp = new Set(prevSet);
-      temp.add(key);
-      return temp;
+    updateSet((newSet) => {
+      newSet.add(key);
+      return newSet;
     });
   };
 
   const remove = (key: K) => {
     if (!set.has(key)) {
       return;
     }
-    setSet((prevSet) => {
-      const temp = new Set(prevSet);
-      temp.delete(key);
-      return temp;
+    updateSet((newSet) => {
+      newSet.delete(key);
+      return newSet;
     });
   };