ahooks/es/useMap/index.js

react hooks library

import { __read } from "tslib";
import { useState } from 'react';
import useMemoizedFn from '../useMemoizedFn';
function useMap(initialValue) {
  var getInitValue = function () {
    return new Map(initialValue);
  };
  var _a = __read(useState(getInitValue), 2),
    map = _a[0],
    setMap = _a[1];
  var set = function (key, entry) {
    setMap(function (prev) {
      var temp = new Map(prev);
      temp.set(key, entry);
      return temp;
    });
  };
  var setAll = function (newMap) {
    setMap(new Map(newMap));
  };
  var remove = function (key) {
    setMap(function (prev) {
      var temp = new Map(prev);
      temp.delete(key);
      return temp;
    });
  };
  var reset = function () {
    return setMap(getInitValue());
  };
  var get = function (key) {
    return map.get(key);
  };
  return [map, {
    set: useMemoizedFn(set),
    setAll: useMemoizedFn(setAll),
    remove: useMemoizedFn(remove),
    reset: useMemoizedFn(reset),
    get: useMemoizedFn(get)
  }];
}
export default useMap;