36 lines
1.2 KiB
JavaScript
36 lines
1.2 KiB
JavaScript
"use strict";
|
|
|
|
var _interopRequireWildcard = require("@babel/runtime/helpers/interopRequireWildcard").default;
|
|
Object.defineProperty(exports, "__esModule", {
|
|
value: true
|
|
});
|
|
exports.default = void 0;
|
|
var React = _interopRequireWildcard(require("react"));
|
|
const useLazyKVMap = (data, childrenColumnName, getRowKey) => {
|
|
const mapCacheRef = React.useRef({});
|
|
function getRecordByKey(key) {
|
|
var _a;
|
|
if (!mapCacheRef.current || mapCacheRef.current.data !== data || mapCacheRef.current.childrenColumnName !== childrenColumnName || mapCacheRef.current.getRowKey !== getRowKey) {
|
|
const kvMap = new Map();
|
|
function dig(records) {
|
|
records.forEach((record, index) => {
|
|
const rowKey = getRowKey(record, index);
|
|
kvMap.set(rowKey, record);
|
|
if (record && typeof record === 'object' && childrenColumnName in record) {
|
|
dig(record[childrenColumnName] || []);
|
|
}
|
|
});
|
|
}
|
|
dig(data);
|
|
mapCacheRef.current = {
|
|
data,
|
|
childrenColumnName,
|
|
kvMap,
|
|
getRowKey
|
|
};
|
|
}
|
|
return (_a = mapCacheRef.current.kvMap) === null || _a === void 0 ? void 0 : _a.get(key);
|
|
}
|
|
return [getRecordByKey];
|
|
};
|
|
var _default = exports.default = useLazyKVMap; |