perf(useMap): reduce re-render

packages/hooks/src/useMap/__tests__/index.test.ts

@@ -98,6 +98,28 @@ describe('useMap', () => {
     ]);
   });
 
+  it('should not render if set same key-value pair', () => {
+    let count = 0;
+    const { result } = renderHook(() => {
+      count++;
+
+      return useMap([['hello', 'world']]);
+    });
+    const [, utils] = result.current;
+
+    expect(count).toBe(1);
+
+    act(() => {
+      utils.set('hello', 'world');
+    });
+    expect(count).toBe(1);
+
+    act(() => {
+      utils.set('hello', 'ahooks');
+    });
+    expect(count).toBe(2);
+  });
+
   it('should override current value if setting existing key', () => {
     const { result } = setup([
       ['foo', 'bar'],
@@ -169,6 +191,28 @@ describe('useMap', () => {
     ]);
   });
 
+  it('should not render if remove non-existing key', () => {
+    let count = 0;
+    const { result } = renderHook(() => {
+      count++;
+
+      return useMap([['hello', 'world']]);
+    });
+    const [, utils] = result.current;
+
+    expect(count).toBe(1);
+
+    act(() => {
+      utils.remove('hi');
+    });
+    expect(count).toBe(1);
+
+    act(() => {
+      utils.remove('hello');
+    });
+    expect(count).toBe(2);
+  });
+
   it('reset should be work', () => {
     const { result } = setup([['msg', 'hello']]);
     const { set, reset } = result.current[1];

packages/hooks/src/useMap/index.ts

@@ -6,6 +6,9 @@ function useMap<K, T>(initialValue?: Iterable<readonly [K, T]>) {
   const [map, setMap] = useState<Map<K, T>>(getInitValue);
 
   const set = (key: K, entry: T) => {
+    if (map.has(key) && map.get(key) === entry) {
+      return;
+    }
     setMap((prev) => {
       const temp = new Map(prev);
       temp.set(key, entry);
@@ -18,6 +21,9 @@ function useMap<K, T>(initialValue?: Iterable<readonly [K, T]>) {
   };
 
   const remove = (key: K) => {
+    if (!map.has(key)) {
+      return;
+    }
     setMap((prev) => {
       const temp = new Map(prev);
       temp.delete(key);