@aloudata/ink-lineage 0.0.1-beta.34 → 0.0.1-beta.36

package/dist/manager/{dataProcessor/TableDataProcessor.js → DataProcessor.js}

@@ -1,28 +1,27 @@
-import { __assign, __extends, __read, __values } from "tslib";
-import { EDirection, getNodeHeight, groupNodesByEdgesAndDirection, } from '@aloudata/ink-graph-new';
-import { EEntityType, EDashType, } from '../../types';
+import { __assign, __extends, __read, __spreadArray, __values } from "tslib";
+import { getNodeHeight, groupNodesByEdgesAndDirection, } from '@aloudata/ink-graph-new';
+import { EEntityType, EDirection, EDashType, } from '../types';
 import _ from 'lodash';
-import { BaseManager } from '../BaseManager';
-import { CLICK_EDGE, COL_HEIGHT, CUSTOM_NODE, LINEAGE_EDGE, NODE_WIDTH, TABLE_HEADER_HEIGHT, TABLE_NODE, } from '../../constant';
-var TableDataProcessor = /** @class */ (function (_super) {
-    __extends(TableDataProcessor, _super);
-    function TableDataProcessor() {
+import { BaseManager } from './BaseManager';
+import { COL_HEIGHT, CUSTOM_NODE, LINEAGE_EDGE, NODE_WIDTH, TABLE_HEADER_HEIGHT, TABLE_NODE, } from '../constant';
+var DataProcessor = /** @class */ (function (_super) {
+    __extends(DataProcessor, _super);
+    function DataProcessor() {
         return _super !== null && _super.apply(this, arguments) || this;
     }
-    TableDataProcessor.parseData = function (data, startNodes) {
-        TableDataProcessor.startNodes = startNodes;
+    DataProcessor.parseData = function (data, startNodes) {
+        DataProcessor.startNodes = startNodes;
         //  parse data here
         var entities = data.entities, relations = data.relations;
-        var nodes = TableDataProcessor.parseEntities(entities, startNodes);
-        var edges = TableDataProcessor.parseRelations(relations);
+        var nodes = DataProcessor.parseEntities(entities, startNodes);
+        var edges = DataProcessor.parseRelations(relations);
         var res = {
             nodes: nodes,
             edges: edges,
         };
         return res;
     };
-    TableDataProcessor.parseEntities = function (entities, startNodes) {
-        var _this = this;
+    DataProcessor.parseEntities = function (entities, startNodes) {
         var nodesId = [];
         var nodes = [];
         entities.forEach(function (entity) {
@@ -33,22 +32,20 @@ var TableDataProcessor = /** @class */ (function (_super) {
             switch (entity.typeCode) {
                 case 'Table':
                 case 'View':
-                    node = TableDataProcessor.parseTableEntity(entity, isStartNode);
+                    node = DataProcessor.parseTableEntity(entity, isStartNode);
                     break;
                 default:
-                    node = TableDataProcessor.parseCustomEntity(entity, isStartNode);
+                    node = DataProcessor.parseCustomEntity(entity, isStartNode);
                     break;
             }
             if (node) {
                 nodes.push(node);
                 nodesId.push(node.id);
-                _this.nodesIdMap.set(node.id, node);
             }
         });
         return nodes;
     };
-    TableDataProcessor.parseRelations = function (relations) {
-        var _this = this;
+    DataProcessor.parseRelations = function (relations) {
         var existedEdgesMap = new Map();
         var edges = [];
         relations.forEach(function (relation) {
@@ -64,16 +61,9 @@ var TableDataProcessor = /** @class */ (function (_super) {
                 return;
             }
             var isIndirect = relation.relationTypeCode.includes('Indirect');
-            var edgeType = LINEAGE_EDGE;
-            var srcGuid = relation.srcGuid, dstGuid = relation.dstGuid;
-            var srcNode = _this.nodesIdMap.get(srcGuid);
-            var tgtNode = _this.nodesIdMap.get(dstGuid);
-            if (srcNode || tgtNode) {
-                edgeType = CLICK_EDGE;
-            }
             var edge = {
                 id: edgeId,
-                type: edgeType,
+                type: LINEAGE_EDGE,
                 data: __assign({}, relation),
                 source: relation.srcGuid,
                 target: relation.dstGuid,
@@ -94,25 +84,25 @@ var TableDataProcessor = /** @class */ (function (_super) {
         });
         return edges;
     };
-    TableDataProcessor.parseTableEntity = function (entity, isStartNode) {
+    DataProcessor.parseTableEntity = function (entity, isStartNode) {
         var _a, _b, _c, _d, _e, _f, _g;
         var cols = Array.isArray((_a = entity.properties) === null || _a === void 0 ? void 0 : _a.columns)
             ? ((_c = (_b = entity.properties) === null || _b === void 0 ? void 0 : _b.columns) === null || _c === void 0 ? void 0 : _c.map(function (col) {
                 var _a;
-                return TableDataProcessor.parseColumn(col, isStartNode, entity.guid, (_a = entity.properties) === null || _a === void 0 ? void 0 : _a.name);
+                return DataProcessor.parseColumn(col, isStartNode, entity.guid, (_a = entity.properties) === null || _a === void 0 ? void 0 : _a.name);
             })) || []
             : [];
         // 只有起始资产的relatedCols一开始有元素
         var relatedCols = Array.isArray((_d = entity.properties) === null || _d === void 0 ? void 0 : _d.columns)
             ? ((_f = (_e = entity.properties) === null || _e === void 0 ? void 0 : _e.columns) === null || _f === void 0 ? void 0 : _f.map(function (col) {
                 var _a;
-                return TableDataProcessor.parseRelatedColumn(col, isStartNode, entity.guid, (_a = entity.properties) === null || _a === void 0 ? void 0 : _a.name);
+                return DataProcessor.parseRelatedColumn(col, isStartNode, entity.guid, (_a = entity.properties) === null || _a === void 0 ? void 0 : _a.name);
             })) || []
             : [];
         var data = {
             id: entity.guid,
             type: TABLE_NODE,
-            data: __assign(__assign({}, entity.properties), { type: ((_g = entity.properties) === null || _g === void 0 ? void 0 : _g.type) || EEntityType.TABLE, guid: entity.guid, typeCode: entity.typeCode, isAggregationTable: true }),
+            data: __assign(__assign({}, entity.properties), { type: ((_g = entity.properties) === null || _g === void 0 ? void 0 : _g.type) || EEntityType.TABLE, guid: entity.guid, typeCode: entity.typeCode }),
             style: {
                 width: NODE_WIDTH + 2,
                 height: TABLE_HEADER_HEIGHT,
@@ -147,7 +137,7 @@ var TableDataProcessor = /** @class */ (function (_super) {
         data.style.height = getNodeHeight(data);
         return data;
     };
-    TableDataProcessor.parseColumn = function (column, isStartNode, tableId, tableName) {
+    DataProcessor.parseColumn = function (column, isStartNode, tableId, tableName) {
         return {
             data: __assign({}, column),
             type: '',
@@ -172,7 +162,7 @@ var TableDataProcessor = /** @class */ (function (_super) {
             hasMoreOutput: true,
         };
     };
-    TableDataProcessor.parseRelatedColumn = function (column, isStartNode, tableId, tableName) {
+    DataProcessor.parseRelatedColumn = function (column, isStartNode, tableId, tableName) {
         return {
             data: __assign({}, column),
             type: '',
@@ -198,7 +188,7 @@ var TableDataProcessor = /** @class */ (function (_super) {
             hasMoreOutput: true,
         };
     };
-    TableDataProcessor.parseCustomEntity = function (entity, isStartNode) {
+    DataProcessor.parseCustomEntity = function (entity, isStartNode) {
         var _a, _b, _c, _d;
         var data = {
             id: entity.guid,
@@ -226,7 +216,7 @@ var TableDataProcessor = /** @class */ (function (_super) {
             children: Array.isArray((_b = entity.properties) === null || _b === void 0 ? void 0 : _b.columns)
                 ? ((_d = (_c = entity.properties) === null || _c === void 0 ? void 0 : _c.columns) === null || _d === void 0 ? void 0 : _d.map(function (col) {
                     var _a;
-                    return TableDataProcessor.parseColumn(col, isStartNode, entity.guid, (_a = entity.properties) === null || _a === void 0 ? void 0 : _a.name);
+                    return DataProcessor.parseColumn(col, isStartNode, entity.guid, (_a = entity.properties) === null || _a === void 0 ? void 0 : _a.name);
                 })) || []
                 : [],
             pagination: {
@@ -240,227 +230,13 @@ var TableDataProcessor = /** @class */ (function (_super) {
         data.style.height = getNodeHeight(data);
         return data;
     };
-    // 刷新相关节点和边
-    TableDataProcessor.prototype.refreshRelatedNodesAndEdges = function (relatedNodes, relatedEdges) {
-        var _this = this;
-        relatedNodes.forEach(function (n) {
-            if (!n)
-                return;
-            var nodeConfig = _this.lineageManager.getNodeConfigById(n.id);
-            if (!nodeConfig)
-                return;
-            nodeConfig.isRelated = true;
-            if (nodeConfig.style.zIndex < 4) {
-                nodeConfig.style.zIndex = 3;
-                nodeConfig.style.stroke = '#3271C9';
-                if (!nodeConfig.isActive) {
-                    nodeConfig.style.fill = '#E4EFFF';
-                }
-                else {
-                    nodeConfig.style.fill = '#A9CFFB';
-                }
-            }
-        });
-        relatedEdges.forEach(function (e) {
-            if (!e)
-                return;
-            var edge = _this.lineageManager.getEdgeConfigById(e.id);
-            if (!edge)
-                return;
-            if ((edge.inFoldLink &&
-                _this.lineageManager.foldLineageManager.isFoldChecked) ||
-                edge.isExtraFoldEdge)
-                return;
-            edge.style.stroke = '#3271C9';
-        });
-    };
-    // 更新节点和边可见性
-    TableDataProcessor.prototype.updateNodesAndEdgesVisible = function (data, edgeDashVisible, onlyShowRelated, isFoldChecked) {
-        this.updateEdgesVisible(data, edgeDashVisible, onlyShowRelated, isFoldChecked);
-        // 更新节点本身、列、节点children的可见性
-        this.updateNodesVisible(data, edgeDashVisible, onlyShowRelated, isFoldChecked);
-    };
-    // 获取边可见性
-    TableDataProcessor.prototype.getEdgeVisible = function (edgeCfg, edgeDashVisible, onlyShowRelated, isFoldChecked, relatedNodes) {
-        var _a, _b, _c;
-        // 如果边被折叠了，那么该边就隐藏
-        if (edgeCfg.isFolded)
-            return false;
-        var visible = false;
-        // 折叠
-        if (isFoldChecked) {
-            var foldStatus = (_c = (_b = (_a = this.lineageManager) === null || _a === void 0 ? void 0 : _a.foldLineageManager) === null || _b === void 0 ? void 0 : _b.foldStatusMap) === null || _c === void 0 ? void 0 : _c.get(edgeCfg.foldTableId);
-            if (foldStatus) {
-                visible = !!edgeCfg.isExtraFoldEdge;
-                if (!visible) {
-                    return visible;
-                }
-            }
-        }
-        // 仅相关
-        if (onlyShowRelated) {
-            var source = edgeCfg.source, target = edgeCfg.target;
-            if (relatedNodes[source] && relatedNodes[target]) {
-                visible = true;
-            }
-            else {
-                return false;
-            }
-        }
-        switch (edgeDashVisible) {
-            case EDashType.DIRECT:
-                visible =
-                    edgeCfg.relationType === EDashType.DIRECT ||
-                        edgeCfg.relationType === EDashType.ALL;
-                break;
-            case EDashType.INDIRECT:
-                visible = true;
-                break;
-        }
-        return visible;
-    };
-    // 获取相关节点可见性
-    TableDataProcessor.prototype.getRelatedNodeVisible = function (nodeConfig, data, edgeDashVisible) {
-        var _a, _b, _c, _d, _e, _f;
-        // 仅相关开启的情况下，如果高亮链路是表链路，隐藏非相关的列
-        if ((_b = (_a = this.lineageManager) === null || _a === void 0 ? void 0 : _a.rightKeyMenuManager) === null || _b === void 0 ? void 0 : _b.onlyShowRelated) {
-            if (!((_e = (_d = (_c = this.lineageManager) === null || _c === void 0 ? void 0 : _c.activeNodes) === null || _d === void 0 ? void 0 : _d[0]) === null || _e === void 0 ? void 0 : _e.tableName)) {
-                if (!_.some((_f = this.lineageManager) === null || _f === void 0 ? void 0 : _f.relatedNodes, { id: nodeConfig.id }))
-                    return false;
-            }
-        }
-        var cacheBit = this.nodeEdgeRelationCache.get(nodeConfig.id);
-        if (!cacheBit)
-            return false;
-        var isAllIndirectEdges = cacheBit < 0;
-        // 开关只有true false，直接、间接
-        if (edgeDashVisible === EDashType.INDIRECT) {
-            return true;
-        }
-        else {
-            // 如果边都是间接边，那么该节点就隐藏
-            if (isAllIndirectEdges)
-                return false;
-            return true;
-        }
-    };
-    // 获取节点可见性
-    TableDataProcessor.prototype.getNodeVisible = function (nodeConfig, data, edgeDashVisible, onlyShowRelated, isFoldChecked) {
-        if (nodeConfig.isStartNode)
-            return true;
-        if (nodeConfig.isFolded)
-            return false;
-        // 只显示相关节点 & 该节点不是相关节点 & 该节点不是表，则该节点不可见
-        if (onlyShowRelated && !nodeConfig.isRelated && !nodeConfig.tableId)
-            return false;
-        if (nodeConfig.canBeFold && isFoldChecked)
-            return false;
-        var cacheBit = this.nodeEdgeRelationCache.get(nodeConfig.id);
-        if (!cacheBit)
-            return false;
-        var isAllIndirectEdges = cacheBit < 0;
-        // 开关只有true false，直接、间接
-        if (edgeDashVisible === EDashType.INDIRECT) {
-            return true;
-        }
-        else {
-            // 如果边都是间接边，那么该节点就隐藏
-            if (isAllIndirectEdges)
-                return false;
-            return true;
-        }
-    };
-    // 更新边可见性
-    TableDataProcessor.prototype.updateEdgesVisible = function (data, edgeDashVisible, onlyShowRelated, isFoldChecked) {
-        var _this = this;
-        if (_.isNil(data))
-            return;
-        var relatedNodes = {};
-        data.nodes.forEach(function (node) {
-            var _a;
-            if (node.isRelated) {
-                relatedNodes[node.id] = node;
-                (_a = node.relatedColumns) === null || _a === void 0 ? void 0 : _a.forEach(function (col) {
-                    if (col.visible) {
-                        relatedNodes[col.id] = col;
-                    }
-                });
-            }
-        });
-        data.edges.forEach(function (edge) {
-            edge.visible = _this.getEdgeVisible(edge, edgeDashVisible, onlyShowRelated, isFoldChecked, relatedNodes);
-        });
-    };
-    // 构建节点/列可见性缓存。根据关联的边类型和当前配置决定是否可见
-    TableDataProcessor.prototype.buildVisibilityCache = function (data, isFoldChecked) {
-        var nodeIdToNodeMap = new Map();
-        var columnIdToTableIdMap = new Map();
-        data.nodes.forEach(function (node) {
-            var _a, _b;
-            (_a = node.children) === null || _a === void 0 ? void 0 : _a.forEach(function (child) {
-                columnIdToTableIdMap.set(child.id, child.tableId);
-            });
-            (_b = node.relatedColumns) === null || _b === void 0 ? void 0 : _b.forEach(function (col) {
-                columnIdToTableIdMap.set(col.id, col.tableId);
-            });
-            nodeIdToNodeMap.set(node.id, node);
-        });
-        this.nodeEdgeRelationCache = new Map();
-        var cache = this.nodeEdgeRelationCache;
-        var updateBit = function (id, isIndirect) {
-            if (!id)
-                return;
-            var bit = cache.get(id);
-            if (!bit) {
-                cache.set(id, isIndirect ? -1 : 1);
-            }
-            if (bit < 0 && !isIndirect) {
-                cache.set(id, 1);
-            }
-        };
-        data.edges.forEach(function (edge) {
-            var srcNode = nodeIdToNodeMap.get(edge.source);
-            var tgtNode = nodeIdToNodeMap.get(edge.target);
-            if (!edge.visible)
-                return;
-            // 合并链路，折叠的边忽略
-            if (((srcNode === null || srcNode === void 0 ? void 0 : srcNode.canBeFold) || (tgtNode === null || tgtNode === void 0 ? void 0 : tgtNode.canBeFold)) && isFoldChecked) {
-                return;
-            }
-            var isIndirect = edge.relationType === EDashType.INDIRECT;
-            updateBit(edge.source, isIndirect);
-            updateBit(edge.target, isIndirect);
-            // 更新列对应的表（如果是列
-            updateBit(columnIdToTableIdMap.get(edge.source), isIndirect);
-            updateBit(columnIdToTableIdMap.get(edge.target), isIndirect);
-        });
-    };
-    // 更新节点的可见性
-    TableDataProcessor.prototype.updateNodesVisible = function (data, edgeDashVisible, onlyShowRelated, isFoldChecked) {
-        var _this = this;
-        if (_.isNil(data))
-            return;
-        this.buildVisibilityCache(data, isFoldChecked);
-        data.nodes.forEach(function (node) {
-            var _a, _b;
-            node.visible = _this.getNodeVisible(node, data, edgeDashVisible, onlyShowRelated, isFoldChecked);
-            (_a = node.relatedColumns) === null || _a === void 0 ? void 0 : _a.forEach(function (col) {
-                col.visible = _this.getRelatedNodeVisible(col, data, edgeDashVisible);
-            });
-            (_b = node.children) === null || _b === void 0 ? void 0 : _b.forEach(function (child) {
-                child.visible = _this.getNodeVisible(child, data, edgeDashVisible, onlyShowRelated, isFoldChecked);
-            });
-        });
-    };
-    // 更新icon的可见性
-    TableDataProcessor.prototype.updateSideIconVisible = function (data) {
-        var e_1, _a;
-        var _this = this;
+    DataProcessor.prototype.updateSideIconVisible = function (data) {
+        var e_1, _a, e_2, _b;
         if (_.isNil(data))
             return;
         var nodesConfig = data.nodes.filter(function (cfg) { return cfg.visible; });
         var edgesConfig = data.edges.filter(function (cfg) { return cfg.visible; });
-        if (!nodesConfig.length)
+        if (!nodesConfig)
             return;
         var startNodeIndex = nodesConfig.findIndex(function (node) { return node.isStartNode; });
         if (startNodeIndex < 0)
@@ -469,7 +245,7 @@ var TableDataProcessor = /** @class */ (function (_super) {
         if (!startNodeCfg)
             return;
         var nodesConfigMap = new Map();
-        nodesConfig.forEach(function (n) {
+        nodesConfig === null || nodesConfig === void 0 ? void 0 : nodesConfig.forEach(function (n) {
             var _a;
             nodesConfigMap.set(n.id, n);
             (_a = n.children) === null || _a === void 0 ? void 0 : _a.forEach(function (child) {
@@ -482,48 +258,46 @@ var TableDataProcessor = /** @class */ (function (_super) {
             return;
         var inputGroupNodes = inputGroupedData.groupedNodes;
         var outputGroupNodes = outputGroupedData.groupedNodes;
-        // 优化Map合并：直接使用inputGroupNodes作为基础，避免重复has()调用
-        var groupedNodes = new Map(inputGroupNodes || []);
-        if (outputGroupNodes) {
+        // 两个map合并
+        var groupedNodes = new Map();
+        if (inputGroupNodes) {
             try {
-                for (var outputGroupNodes_1 = __values(outputGroupNodes), outputGroupNodes_1_1 = outputGroupNodes_1.next(); !outputGroupNodes_1_1.done; outputGroupNodes_1_1 = outputGroupNodes_1.next()) {
-                    var _b = __read(outputGroupNodes_1_1.value, 2), key = _b[0], value = _b[1];
-                    if (!groupedNodes.has(key)) {
-                        groupedNodes.set(key, value);
-                    }
+                for (var inputGroupNodes_1 = __values(inputGroupNodes), inputGroupNodes_1_1 = inputGroupNodes_1.next(); !inputGroupNodes_1_1.done; inputGroupNodes_1_1 = inputGroupNodes_1.next()) {
+                    var _c = __read(inputGroupNodes_1_1.value, 2), key = _c[0], value = _c[1];
+                    if (groupedNodes.has(key))
+                        continue;
+                    groupedNodes.set(key, value);
                 }
             }
             catch (e_1_1) { e_1 = { error: e_1_1 }; }
             finally {
                 try {
-                    if (outputGroupNodes_1_1 && !outputGroupNodes_1_1.done && (_a = outputGroupNodes_1.return)) _a.call(outputGroupNodes_1);
+                    if (inputGroupNodes_1_1 && !inputGroupNodes_1_1.done && (_a = inputGroupNodes_1.return)) _a.call(inputGroupNodes_1);
                 }
                 finally { if (e_1) throw e_1.error; }
             }
         }
-        var dstNodeIdsSet = new Set();
-        var srcNodeIdsSet = new Set();
-        edgesConfig.forEach(function (item) {
-            var _a, _b;
-            var target = item.target, source = item.source;
-            var dstNode = _this.lineageManager.getNodeConfigById(target);
-            var srcNode = _this.lineageManager.getNodeConfigById(source);
-            if (srcNode.position.x < dstNode.position.x) {
-                srcNodeIdsSet.add(source);
-                dstNodeIdsSet.add(target);
-            }
-            else {
-                if (source === ((_a = _this.lineageManager.activeNodes[0]) === null || _a === void 0 ? void 0 : _a.id)) {
-                    srcNodeIdsSet.add(source);
+        if (outputGroupNodes) {
+            try {
+                for (var outputGroupNodes_1 = __values(outputGroupNodes), outputGroupNodes_1_1 = outputGroupNodes_1.next(); !outputGroupNodes_1_1.done; outputGroupNodes_1_1 = outputGroupNodes_1.next()) {
+                    var _d = __read(outputGroupNodes_1_1.value, 2), key = _d[0], value = _d[1];
+                    if (groupedNodes.has(key))
+                        continue;
+                    groupedNodes.set(key, value);
                 }
-                if (target === ((_b = _this.lineageManager.activeNodes[0]) === null || _b === void 0 ? void 0 : _b.id)) {
-                    dstNodeIdsSet.add(target);
+            }
+            catch (e_2_1) { e_2 = { error: e_2_1 }; }
+            finally {
+                try {
+                    if (outputGroupNodes_1_1 && !outputGroupNodes_1_1.done && (_b = outputGroupNodes_1.return)) _b.call(outputGroupNodes_1);
                 }
+                finally { if (e_2) throw e_2.error; }
             }
-        });
-        // 优化节点处理逻辑
-        nodesConfig.forEach(function (n) {
-            // 重置状态
+        }
+        var dstNodeIds = edgesConfig.map(function (edge) { return edge.target; });
+        var srcNodeIds = edgesConfig.map(function (edge) { return edge.source; });
+        // table
+        nodesConfig === null || nodesConfig === void 0 ? void 0 : nodesConfig.forEach(function (n) {
             n.canExpandInput = false;
             n.canExpandOutput = false;
             n.canFoldInput = false;
@@ -533,18 +307,17 @@ var TableDataProcessor = /** @class */ (function (_super) {
                 console.warn("get node level error '".concat(n.id, "'"));
                 return;
             }
-            // 优化：使用Set的has()方法进行O(1)查找
             var isInputLeaf = false;
             var isOutputLeaf = false;
             if (currNodeLevel === 0) {
-                isInputLeaf = !dstNodeIdsSet.has(n.id);
-                isOutputLeaf = !srcNodeIdsSet.has(n.id);
+                isInputLeaf = !dstNodeIds.includes(n.id);
+                isOutputLeaf = !srcNodeIds.includes(n.id);
             }
             else if (currNodeLevel < 0) {
-                isInputLeaf = !dstNodeIdsSet.has(n.id);
+                isInputLeaf = !dstNodeIds.includes(n.id);
             }
             else {
-                isOutputLeaf = !srcNodeIdsSet.has(n.id);
+                isOutputLeaf = !srcNodeIds.includes(n.id);
             }
             // input node exist & left of start node
             if (currNodeLevel <= 0 && !isInputLeaf) {
@@ -564,29 +337,25 @@ var TableDataProcessor = /** @class */ (function (_super) {
             }
             // column:  always true if isStartNode === true
             var columnRefreshList = n.isStartNode
-                ? (n.children || []).concat(n.relatedColumns || [])
-                : (n.relatedColumns || []);
-            // 处理列的展开/折叠状态
-            columnRefreshList.forEach(function (col) {
-                // 重置列状态
+                ? __spreadArray(__spreadArray([], __read(n.children), false), __read(n.relatedColumns), false) : n.relatedColumns;
+            columnRefreshList === null || columnRefreshList === void 0 ? void 0 : columnRefreshList.forEach(function (col) {
                 col.canExpandInput = false;
                 col.canExpandOutput = false;
                 col.canFoldInput = false;
                 col.canFoldOutput = false;
                 if (!col.isStartNode && !col.isRelatedColumn)
                     return;
-                // 优化：使用Set进行O(1)查找
                 var isInputLeaf_col = false;
                 var isOutputLeaf_col = false;
                 if (currNodeLevel === 0) {
-                    isInputLeaf_col = !dstNodeIdsSet.has(col.id);
-                    isOutputLeaf_col = !srcNodeIdsSet.has(col.id);
+                    isInputLeaf_col = !dstNodeIds.includes(col.id);
+                    isOutputLeaf_col = !srcNodeIds.includes(col.id);
                 }
                 else if (currNodeLevel < 0) {
-                    isInputLeaf_col = !dstNodeIdsSet.has(col.id);
+                    isInputLeaf_col = !dstNodeIds.includes(col.id);
                 }
                 else {
-                    isOutputLeaf_col = !srcNodeIdsSet.has(col.id);
+                    isOutputLeaf_col = !srcNodeIds.includes(col.id);
                 }
                 // input node exist & left of start node
                 if (currNodeLevel <= 0 && !isInputLeaf_col) {
@@ -607,10 +376,212 @@ var TableDataProcessor = /** @class */ (function (_super) {
             });
         });
     };
-    TableDataProcessor.startNodes = [];
-    // 放在这里，是因为parseData相当于一个工具方法，可以在lineageManager没有初始化的时候，就调用
-    TableDataProcessor.nodesIdMap = new Map();
-    return TableDataProcessor;
+    DataProcessor.prototype.getNodeType = function (typeCode) {
+        return typeCode === 'Table' ? TABLE_NODE : CUSTOM_NODE;
+    };
+    DataProcessor.prototype.updateNodesVisible = function (data, edgeDashVisible, onlyShowRelated, isFoldChecked) {
+        var _this = this;
+        if (_.isNil(data))
+            return;
+        data.nodes.forEach(function (node) {
+            var _a, _b;
+            node.visible = _this.getNodeVisible(node, data, edgeDashVisible, onlyShowRelated, isFoldChecked);
+            (_a = node.relatedColumns) === null || _a === void 0 ? void 0 : _a.forEach(function (col) {
+                col.visible = _this.getRelatedNodeVisible(col, data, edgeDashVisible);
+            });
+            (_b = node.children) === null || _b === void 0 ? void 0 : _b.forEach(function (child) {
+                child.visible = _this.getNodeVisible(child, data, edgeDashVisible, onlyShowRelated, isFoldChecked);
+            });
+        });
+    };
+    DataProcessor.prototype.updateEdgesVisible = function (data, edgeDashVisible, onlyShowRelated, isFoldChecked) {
+        var _this = this;
+        if (_.isNil(data))
+            return;
+        var relatedNodes = {};
+        data.nodes.forEach(function (node) {
+            var _a;
+            if (node.isRelated) {
+                relatedNodes[node.id] = node;
+                (_a = node.relatedColumns) === null || _a === void 0 ? void 0 : _a.forEach(function (col) {
+                    if (col.visible) {
+                        relatedNodes[col.id] = col;
+                    }
+                });
+            }
+        });
+        data.edges.forEach(function (edge) {
+            edge.visible = _this.getEdgeVisible(edge, edgeDashVisible, onlyShowRelated, isFoldChecked, relatedNodes);
+        });
+    };
+    DataProcessor.prototype.getNodeVisible = function (nodeConfig, data, edgeDashVisible, onlyShowRelated, isFoldChecked) {
+        if (nodeConfig.isStartNode)
+            return true;
+        if (nodeConfig.isFolded)
+            return false;
+        // 只显示相关节点 & 该节点不是相关节点 & 该节点不是表，则该节点不可见
+        if (onlyShowRelated && !nodeConfig.isRelated && !nodeConfig.tableId)
+            return false;
+        if (nodeConfig.canBeFold && isFoldChecked)
+            return false;
+        // 找出所有target、source为该节点的边
+        var relatedEdges = data.edges.filter(function (edge) {
+            var _a;
+            return (
+            // 上游表-当前表、当前表-下游表
+            edge.target === nodeConfig.id ||
+                edge.source === nodeConfig.id ||
+                (
+                // 上游列-当前列、当前列-下游列
+                (_a = nodeConfig === null || nodeConfig === void 0 ? void 0 : nodeConfig.relatedColumns) === null || _a === void 0 ? void 0 : _a.some(function (col) {
+                    return edge.target === col.id || edge.source === col.id;
+                })));
+        });
+        if (!relatedEdges.length)
+            return false;
+        var allRelatedNodeInvisible = relatedEdges.every(function (edgeCfg) {
+            var srcNode = data.nodes.find(function (node) { return node.id === edgeCfg.source; });
+            var tgtNode = data.nodes.find(function (node) { return node.id === edgeCfg.target; });
+            if (((srcNode === null || srcNode === void 0 ? void 0 : srcNode.canBeFold) || (tgtNode === null || tgtNode === void 0 ? void 0 : tgtNode.canBeFold)) && isFoldChecked)
+                return true;
+        });
+        if (allRelatedNodeInvisible)
+            return false;
+        var isAllIndirectEdges = relatedEdges.every(function (edge) { return edge.relationType === EDashType.INDIRECT; });
+        // 开关只有true false，直接、间接
+        if (edgeDashVisible === EDashType.INDIRECT) {
+            return true;
+        }
+        else {
+            // 如果边都是间接边，那么该节点就隐藏
+            if (isAllIndirectEdges)
+                return false;
+            return true;
+        }
+    };
+    DataProcessor.prototype.getRelatedNodeVisible = function (nodeConfig, data, edgeDashVisible) {
+        var _a, _b, _c, _d, _e, _f;
+        // 仅相关开启的情况下，如果高亮链路是表链路，隐藏非相关的列
+        if ((_b = (_a = this.lineageManager) === null || _a === void 0 ? void 0 : _a.rightKeyMenuManager) === null || _b === void 0 ? void 0 : _b.onlyShowRelated) {
+            if (!((_e = (_d = (_c = this.lineageManager) === null || _c === void 0 ? void 0 : _c.activeNodes) === null || _d === void 0 ? void 0 : _d[0]) === null || _e === void 0 ? void 0 : _e.tableName)) {
+                if (!_.some((_f = this.lineageManager) === null || _f === void 0 ? void 0 : _f.relatedNodes, { id: nodeConfig.id }))
+                    return false;
+            }
+        }
+        // 找出所有target、source为该节点的边
+        var relatedEdges = data.edges.filter(function (edge) {
+            return (edge.target === nodeConfig.id || edge.source === nodeConfig.id) &&
+                edge.visible;
+        });
+        if (!relatedEdges.length)
+            return false;
+        var isAllIndirectEdges = relatedEdges.every(function (edge) { return edge.relationType === EDashType.INDIRECT; });
+        // 开关只有true false，直接、间接
+        if (edgeDashVisible === EDashType.INDIRECT) {
+            return true;
+        }
+        else {
+            // 如果边都是间接边，那么该节点就隐藏
+            if (isAllIndirectEdges)
+                return false;
+            return true;
+        }
+    };
+    DataProcessor.prototype.getEdgeVisible = function (edgeCfg, edgeDashVisible, onlyShowRelated, isFoldChecked, relatedNodes) {
+        var _a, _b, _c;
+        // 如果边被折叠了，那么该边就隐藏
+        if (edgeCfg.isFolded)
+            return false;
+        var visible = false;
+        // 折叠
+        if (isFoldChecked) {
+            var foldStatus = (_c = (_b = (_a = this.lineageManager) === null || _a === void 0 ? void 0 : _a.foldLineageManager) === null || _b === void 0 ? void 0 : _b.foldStatusMap) === null || _c === void 0 ? void 0 : _c.get(edgeCfg.foldTableId);
+            if (foldStatus) {
+                visible = !!edgeCfg.isExtraFoldEdge;
+                if (!visible) {
+                    return visible;
+                }
+            }
+        }
+        // 仅相关
+        if (onlyShowRelated) {
+            var source = edgeCfg.source, target = edgeCfg.target;
+            if (relatedNodes[source] && relatedNodes[target]) {
+                visible = true;
+            }
+            else {
+                return false;
+            }
+        }
+        switch (edgeDashVisible) {
+            case EDashType.DIRECT:
+                visible =
+                    edgeCfg.relationType === EDashType.DIRECT ||
+                        edgeCfg.relationType === EDashType.ALL;
+                break;
+            case EDashType.INDIRECT:
+                visible = true;
+                break;
+        }
+        return visible;
+    };
+    DataProcessor.prototype.updateNodesAndEdgesVisible = function (data, edgeDashVisible, onlyShowRelated, isFoldChecked) {
+        this.updateEdgesVisible(data, edgeDashVisible, onlyShowRelated, isFoldChecked);
+        // 更新节点本身、列、节点children的可见性
+        this.updateNodesVisible(data, edgeDashVisible, onlyShowRelated, isFoldChecked);
+    };
+    DataProcessor.prototype.refreshRelatedNodesAndEdges = function (relatedNodes, relatedEdges) {
+        var _this = this;
+        relatedNodes.forEach(function (n) {
+            if (!n)
+                return;
+            var nodeConfig = _this.lineageManager.getNodeConfigById(n.id);
+            if (!nodeConfig)
+                return;
+            nodeConfig.isRelated = true;
+            nodeConfig.style.stroke = '#3271C9';
+            if (!nodeConfig.isActive) {
+                nodeConfig.style.fill = '#E4EFFF';
+            }
+            nodeConfig.style.zIndex = 3;
+        });
+        relatedEdges.forEach(function (e) {
+            if (!e)
+                return;
+            var edge = _this.lineageManager.getEdgeConfigById(e.id);
+            if (!edge)
+                return;
+            if ((edge.inFoldLink &&
+                _this.lineageManager.foldLineageManager.isFoldChecked) ||
+                edge.isExtraFoldEdge)
+                return;
+            edge.style.stroke = '#3271C9';
+        });
+    };
+    DataProcessor.prototype.updateNodeHeight = function (data) {
+        var _this = this;
+        data.nodes.forEach(function (node) {
+            node.style.height = _this.calcNodeHeight(node);
+        });
+    };
+    DataProcessor.prototype.updateColumnPosition = function (data) {
+        data.nodes.forEach(function (node) {
+            var _a, _b;
+            var visibleRelatedColumns = (_a = node.relatedColumns) === null || _a === void 0 ? void 0 : _a.filter(function (col) { return col.visible; });
+            visibleRelatedColumns === null || visibleRelatedColumns === void 0 ? void 0 : visibleRelatedColumns.forEach(function (col, i) {
+                col.position.y = i * COL_HEIGHT;
+            });
+            var visibleColumns = (_b = node.children) === null || _b === void 0 ? void 0 : _b.filter(function (col) { return col.visible; });
+            visibleColumns === null || visibleColumns === void 0 ? void 0 : visibleColumns.forEach(function (col, i) {
+                col.position.y = node.position.y + COL_HEIGHT * i;
+            });
+        });
+    };
+    DataProcessor.prototype.calcNodeHeight = function (node) {
+        return getNodeHeight(node);
+    };
+    DataProcessor.startNodes = [];
+    return DataProcessor;
 }(BaseManager));
-export { TableDataProcessor };
-//# sourceMappingURL=TableDataProcessor.js.map
+export { DataProcessor };
+//# sourceMappingURL=DataProcessor.js.map