@td-design/rn-hooks
Version:
从ahooks里面抽取的一些在rn项目里可以使用的hooks。提高开发效率
43 lines (40 loc) • 1.25 kB
JavaScript
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 };