antd
Version:
An enterprise-class UI design language and React components implementation
37 lines (36 loc) • 2.03 kB
JavaScript
Object.defineProperty(exports, "__esModule", {
value: true
});
exports.default = void 0;
var React = _interopRequireWildcard(require("react"));
var _transKeys = require("../../_util/transKeys");
function _getRequireWildcardCache(e) { if ("function" != typeof WeakMap) return null; var r = new WeakMap(), t = new WeakMap(); return (_getRequireWildcardCache = function (e) { return e ? t : r; })(e); }
function _interopRequireWildcard(e, r) { if (!r && e && e.__esModule) return e; if (null === e || "object" != typeof e && "function" != typeof e) return { default: e }; var t = _getRequireWildcardCache(r); if (t && t.has(e)) return t.get(e); var n = { __proto__: null }, a = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var u in e) if ("default" !== u && Object.prototype.hasOwnProperty.call(e, u)) { var i = a ? Object.getOwnPropertyDescriptor(e, u) : null; i && (i.get || i.set) ? Object.defineProperty(n, u, i) : n[u] = e[u]; } return n.default = e, t && t.set(e, n), n; }
const useData = (dataSource, rowKey, targetKeys) => {
const mergedDataSource = React.useMemo(() => (dataSource || []).map(record => {
if (rowKey) {
record = Object.assign(Object.assign({}, record), {
key: rowKey(record)
});
}
return record;
}), [dataSource, rowKey]);
const [leftDataSource, rightDataSource] = React.useMemo(() => {
const leftData = [];
const rightData = new Array((targetKeys || []).length);
const targetKeysMap = (0, _transKeys.groupKeysMap)(targetKeys || []);
mergedDataSource.forEach(record => {
// rightData should be ordered by targetKeys
// leftData should be ordered by dataSource
if (targetKeysMap.has(record.key)) {
rightData[targetKeysMap.get(record.key)] = record;
} else {
leftData.push(record);
}
});
return [leftData, rightData];
}, [mergedDataSource, targetKeys, rowKey]);
return [mergedDataSource, leftDataSource, rightDataSource];
};
var _default = exports.default = useData;
;