rc-hooks/es/useCacheState/index.js

React Hooks Library.

import { __read } from "tslib";
import { useCallback, useMemo, useState } from 'react';
import { isUndefined } from 'ut2';
import { Cache } from 'cache2';
import { pkgName } from '../utils/config';
var cacheNamespace = pkgName + '_useCacheState';
/**
 * 获取 `useCacheState` 缓存实例。
 *
 * @see {@link https://github.com/caijf/cache2 | cache2}
 * @param storage 自定义数据存储器，支持 `localStorage` `sessionStorage`。默认使用内存缓存。
 * @returns 缓存实例。
 */
function getCache(storage) {
    return new Cache(cacheNamespace, {
        storage: storage
    });
}
/**
 * 清理 `useCacheState` 缓存。如果不传 `key`，表示清理全部。
 *
 * @see {@link https://github.com/caijf/cache2 | cache2}
 * @param storage 自定义数据存储器，支持 `localStorage` `sessionStorage`。默认使用内存缓存。
 * @param key 键名称。
 */
export function clearCacheState(storage, key) {
    var cache = getCache(storage);
    if (key) {
        cache.del(key);
    }
    else {
        cache.clear();
    }
}
/**
 * 缓存状态。
 *
 * @param key 键名称。
 * @param defaultValue 默认值。
 * @param options 选项。
 * @param options.ttl 过期时间，单位毫秒。默认不过期。
 * @param options.storage 自定义数据存储器，支持 `localStorage` `sessionStorage`。默认使用内存缓存。
 * @returns 状态值和更新状态值的函数。
 */
function useCacheState(key, defaultValue, options) {
    if (options === void 0) { options = {}; }
    var ttl = options.ttl, storage = options.storage;
    var cache = useMemo(function () { return getCache(storage); }, [storage]);
    var _a = __read(useState(function () {
        var cachedValue = cache.get(key);
        return isUndefined(cachedValue) ? defaultValue : cachedValue;
    }), 2), state = _a[0], setState = _a[1];
    var set = useCallback(function (nextState) {
        setState(function (prevState) {
            var newState = nextState instanceof Function ? nextState(prevState) : nextState;
            cache.set(key, newState, ttl);
            return newState;
        });
    }, [cache, key, ttl]);
    return [state, set];
}
export default useCacheState;