99 lines
2.9 KiB
JavaScript
99 lines
2.9 KiB
JavaScript
"use strict";
|
|
|
|
var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault").default;
|
|
Object.defineProperty(exports, "__esModule", {
|
|
value: true
|
|
});
|
|
exports.default = void 0;
|
|
var _classCallCheck2 = _interopRequireDefault(require("@babel/runtime/helpers/classCallCheck"));
|
|
var _createClass2 = _interopRequireDefault(require("@babel/runtime/helpers/createClass"));
|
|
var _react = _interopRequireDefault(require("react"));
|
|
const BEAT_LIMIT = 1000 * 60 * 10;
|
|
/**
|
|
* A helper class to map keys to values.
|
|
* It supports both primitive keys and object keys.
|
|
*/
|
|
let ArrayKeyMap = /*#__PURE__*/function () {
|
|
function ArrayKeyMap() {
|
|
(0, _classCallCheck2.default)(this, ArrayKeyMap);
|
|
this.map = new Map();
|
|
// Use WeakMap to avoid memory leak
|
|
this.objectIDMap = new WeakMap();
|
|
this.nextID = 0;
|
|
this.lastAccessBeat = new Map();
|
|
// We will clean up the cache when reach the limit
|
|
this.accessBeat = 0;
|
|
}
|
|
return (0, _createClass2.default)(ArrayKeyMap, [{
|
|
key: "set",
|
|
value: function set(keys, value) {
|
|
// New set will trigger clear
|
|
this.clear();
|
|
// Set logic
|
|
const compositeKey = this.getCompositeKey(keys);
|
|
this.map.set(compositeKey, value);
|
|
this.lastAccessBeat.set(compositeKey, Date.now());
|
|
}
|
|
}, {
|
|
key: "get",
|
|
value: function get(keys) {
|
|
const compositeKey = this.getCompositeKey(keys);
|
|
const cache = this.map.get(compositeKey);
|
|
this.lastAccessBeat.set(compositeKey, Date.now());
|
|
this.accessBeat += 1;
|
|
return cache;
|
|
}
|
|
}, {
|
|
key: "getCompositeKey",
|
|
value: function getCompositeKey(keys) {
|
|
const ids = keys.map(key => {
|
|
if (key && typeof key === 'object') {
|
|
return `obj_${this.getObjectID(key)}`;
|
|
}
|
|
return `${typeof key}_${key}`;
|
|
});
|
|
return ids.join('|');
|
|
}
|
|
}, {
|
|
key: "getObjectID",
|
|
value: function getObjectID(obj) {
|
|
if (this.objectIDMap.has(obj)) {
|
|
return this.objectIDMap.get(obj);
|
|
}
|
|
const id = this.nextID;
|
|
this.objectIDMap.set(obj, id);
|
|
this.nextID += 1;
|
|
return id;
|
|
}
|
|
}, {
|
|
key: "clear",
|
|
value: function clear() {
|
|
if (this.accessBeat > 10000) {
|
|
const now = Date.now();
|
|
this.lastAccessBeat.forEach((beat, key) => {
|
|
if (now - beat > BEAT_LIMIT) {
|
|
this.map.delete(key);
|
|
this.lastAccessBeat.delete(key);
|
|
}
|
|
});
|
|
this.accessBeat = 0;
|
|
}
|
|
}
|
|
}]);
|
|
}();
|
|
const uniqueMap = new ArrayKeyMap();
|
|
/**
|
|
* Like `useMemo`, but this hook result will be shared across all instances.
|
|
*/
|
|
function useUniqueMemo(memoFn, deps) {
|
|
return _react.default.useMemo(() => {
|
|
const cachedValue = uniqueMap.get(deps);
|
|
if (cachedValue) {
|
|
return cachedValue;
|
|
}
|
|
const newValue = memoFn();
|
|
uniqueMap.set(deps, newValue);
|
|
return newValue;
|
|
}, deps);
|
|
}
|
|
var _default = exports.default = useUniqueMemo; |