UNPKG

@td-design/rn-hooks

Version:

从ahooks里面抽取的一些在rn项目里可以使用的hooks。提高开发效率

43 lines (40 loc) 1.25 kB
import { __read } from '../_virtual/_tslib.js'; import { useState } from 'react'; import useMemoizedFn from '../useMemoizedFn/index.js'; /** * 一个可以管理 Map 类型状态的 Hook。 * @param initialValue 初始值 * @returns */ 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); }; var actions = { set: useMemoizedFn(set), setAll: useMemoizedFn(setAll), remove: useMemoizedFn(remove), reset: useMemoizedFn(reset), get: useMemoizedFn(get), }; return [map, actions]; } export { useMap as default };