@aloudata/ink-lineage 0.0.1-beta.4 → 0.0.1-beta.40

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

@@ -1,27 +1,27 @@
-import { __assign, __extends, __read, __spreadArray, __values } from "tslib";
-import { getNodeHeight, groupNodesByEdgesAndDirection, } from '@aloudata/ink-graph-new';
-import { EEntityType, EDirection, EDashType, } from '../types';
+import { __assign, __extends, __read, __values } from "tslib";
+import { EDirection, getNodeHeight, groupNodesByEdgesAndDirection, } from '@aloudata/ink-graph-new';
+import { EEntityType, EDashType, } from '../types';
 import _ from 'lodash';
 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(lineageManager) {
-        return _super.call(this, lineageManager) || this;
+var TableDataProcessor = /** @class */ (function (_super) {
+    __extends(TableDataProcessor, _super);
+    function TableDataProcessor() {
+        return _super !== null && _super.apply(this, arguments) || this;
     }
-    DataProcessor.parseData = function (data, startNodes) {
-        DataProcessor.startNodes = startNodes;
+    TableDataProcessor.parseData = function (data, startNodes) {
+        TableDataProcessor.startNodes = startNodes;
         //  parse data here
         var entities = data.entities, relations = data.relations;
-        var nodes = DataProcessor.parseEntities(entities, startNodes);
-        var edges = DataProcessor.parseRelations(relations);
+        var nodes = TableDataProcessor.parseEntities(entities, startNodes);
+        var edges = TableDataProcessor.parseRelations(relations);
         var res = {
             nodes: nodes,
             edges: edges,
         };
         return res;
     };
-    DataProcessor.parseEntities = function (entities, startNodes) {
+    TableDataProcessor.parseEntities = function (entities, startNodes) {
         var nodesId = [];
         var nodes = [];
         entities.forEach(function (entity) {
@@ -32,10 +32,10 @@ var DataProcessor = /** @class */ (function (_super) {
             switch (entity.typeCode) {
                 case 'Table':
                 case 'View':
-                    node = DataProcessor.parseTableEntity(entity, isStartNode);
+                    node = TableDataProcessor.parseTableEntity(entity, isStartNode);
                     break;
                 default:
-                    node = DataProcessor.parseCustomEntity(entity, isStartNode);
+                    node = TableDataProcessor.parseCustomEntity(entity, isStartNode);
                     break;
             }
             if (node) {
@@ -45,7 +45,7 @@ var DataProcessor = /** @class */ (function (_super) {
         });
         return nodes;
     };
-    DataProcessor.parseRelations = function (relations) {
+    TableDataProcessor.parseRelations = function (relations) {
         var existedEdgesMap = new Map();
         var edges = [];
         relations.forEach(function (relation) {
@@ -84,19 +84,19 @@ var DataProcessor = /** @class */ (function (_super) {
         });
         return edges;
     };
-    DataProcessor.parseTableEntity = function (entity, isStartNode) {
+    TableDataProcessor.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 DataProcessor.parseColumn(col, isStartNode, entity.guid, (_a = entity.properties) === null || _a === void 0 ? void 0 : _a.name);
+                return TableDataProcessor.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 DataProcessor.parseRelatedColumn(col, isStartNode, entity.guid, (_a = entity.properties) === null || _a === void 0 ? void 0 : _a.name);
+                return TableDataProcessor.parseRelatedColumn(col, isStartNode, entity.guid, (_a = entity.properties) === null || _a === void 0 ? void 0 : _a.name);
             })) || []
             : [];
         var data = {
@@ -137,7 +137,7 @@ var DataProcessor = /** @class */ (function (_super) {
         data.style.height = getNodeHeight(data);
         return data;
     };
-    DataProcessor.parseColumn = function (column, isStartNode, tableId, tableName) {
+    TableDataProcessor.parseColumn = function (column, isStartNode, tableId, tableName) {
         return {
             data: __assign({}, column),
             type: '',
@@ -162,7 +162,7 @@ var DataProcessor = /** @class */ (function (_super) {
             hasMoreOutput: true,
         };
     };
-    DataProcessor.parseRelatedColumn = function (column, isStartNode, tableId, tableName) {
+    TableDataProcessor.parseRelatedColumn = function (column, isStartNode, tableId, tableName) {
         return {
             data: __assign({}, column),
             type: '',
@@ -188,7 +188,7 @@ var DataProcessor = /** @class */ (function (_super) {
             hasMoreOutput: true,
         };
     };
-    DataProcessor.parseCustomEntity = function (entity, isStartNode) {
+    TableDataProcessor.parseCustomEntity = function (entity, isStartNode) {
         var _a, _b, _c, _d;
         var data = {
             id: entity.guid,
@@ -216,7 +216,7 @@ var DataProcessor = /** @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 DataProcessor.parseColumn(col, isStartNode, entity.guid, (_a = entity.properties) === null || _a === void 0 ? void 0 : _a.name);
+                    return TableDataProcessor.parseColumn(col, isStartNode, entity.guid, (_a = entity.properties) === null || _a === void 0 ? void 0 : _a.name);
                 })) || []
                 : [],
             pagination: {
@@ -230,13 +230,227 @@ var DataProcessor = /** @class */ (function (_super) {
         data.style.height = getNodeHeight(data);
         return data;
     };
-    DataProcessor.prototype.updateSideIconVisible = function (data) {
-        var e_1, _a, e_2, _b;
+    // 刷新相关节点和边
+    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;
         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)
+        if (!nodesConfig.length)
             return;
         var startNodeIndex = nodesConfig.findIndex(function (node) { return node.isStartNode; });
         if (startNodeIndex < 0)
@@ -245,7 +459,7 @@ var DataProcessor = /** @class */ (function (_super) {
         if (!startNodeCfg)
             return;
         var nodesConfigMap = new Map();
-        nodesConfig === null || nodesConfig === void 0 ? void 0 : nodesConfig.forEach(function (n) {
+        nodesConfig.forEach(function (n) {
             var _a;
             nodesConfigMap.set(n.id, n);
             (_a = n.children) === null || _a === void 0 ? void 0 : _a.forEach(function (child) {
@@ -258,46 +472,66 @@ var DataProcessor = /** @class */ (function (_super) {
             return;
         var inputGroupNodes = inputGroupedData.groupedNodes;
         var outputGroupNodes = outputGroupedData.groupedNodes;
-        // 两个map合并
-        var groupedNodes = new Map();
-        if (inputGroupNodes) {
+        // 优化Map合并：直接使用inputGroupNodes作为基础，避免重复has()调用
+        var groupedNodes = new Map(inputGroupNodes || []);
+        if (outputGroupNodes) {
             try {
-                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);
+                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);
+                    }
                 }
             }
             catch (e_1_1) { e_1 = { error: e_1_1 }; }
             finally {
                 try {
-                    if (inputGroupNodes_1_1 && !inputGroupNodes_1_1.done && (_a = inputGroupNodes_1.return)) _a.call(inputGroupNodes_1);
+                    if (outputGroupNodes_1_1 && !outputGroupNodes_1_1.done && (_a = outputGroupNodes_1.return)) _a.call(outputGroupNodes_1);
                 }
                 finally { if (e_1) throw e_1.error; }
             }
         }
-        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);
+        var dstNodeIdsSet = new Set();
+        var srcNodeIdsSet = new Set();
+        edgesConfig.forEach(function (item) {
+            var _a, _b, _c, _d, _e, _f;
+            var target = item.target, source = item.source;
+            var dstNode = _this.lineageManager.getNodeConfigById(target);
+            var srcNode = _this.lineageManager.getNodeConfigById(source);
+            // 列
+            if ((srcNode === null || srcNode === void 0 ? void 0 : srcNode.tableId) || (dstNode === null || dstNode === void 0 ? void 0 : dstNode.tableId)) {
+                var parentSrcNode = _this.lineageManager.getNodeConfigById(srcNode === null || srcNode === void 0 ? void 0 : srcNode.tableId);
+                var parentDstNode = _this.lineageManager.getNodeConfigById(dstNode === null || dstNode === void 0 ? void 0 : dstNode.tableId);
+                // 列所在父节点正常
+                if (((_a = parentSrcNode === null || parentSrcNode === void 0 ? void 0 : parentSrcNode.position) === null || _a === void 0 ? void 0 : _a.x) < ((_b = parentDstNode === null || parentDstNode === void 0 ? void 0 : parentDstNode.position) === null || _b === void 0 ? void 0 : _b.x)) {
+                    srcNodeIdsSet.add(source);
+                    dstNodeIdsSet.add(target);
+                }
+                else {
+                    if ((srcNode === null || srcNode === void 0 ? void 0 : srcNode.tableId) && source === ((_c = _this.lineageManager.activeNodes[0]) === null || _c === void 0 ? void 0 : _c.id)) {
+                        srcNodeIdsSet.add(source);
+                    }
+                    if ((dstNode === null || dstNode === void 0 ? void 0 : dstNode.tableId) && target === ((_d = _this.lineageManager.activeNodes[0]) === null || _d === void 0 ? void 0 : _d.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);
+            else if (srcNode && dstNode && srcNode.position.x < dstNode.position.x) {
+                srcNodeIdsSet.add(source);
+                dstNodeIdsSet.add(target);
+            }
+            else {
+                if (source === ((_e = _this.lineageManager.activeNodes[0]) === null || _e === void 0 ? void 0 : _e.id)) {
+                    srcNodeIdsSet.add(source);
+                }
+                if (target === ((_f = _this.lineageManager.activeNodes[0]) === null || _f === void 0 ? void 0 : _f.id)) {
+                    dstNodeIdsSet.add(target);
                 }
-                finally { if (e_2) throw e_2.error; }
             }
-        }
-        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) {
+        });
+        // 优化节点处理逻辑
+        nodesConfig.forEach(function (n) {
+            // 重置状态
             n.canExpandInput = false;
             n.canExpandOutput = false;
             n.canFoldInput = false;
@@ -307,17 +541,18 @@ var DataProcessor = /** @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 = !dstNodeIds.includes(n.id);
-                isOutputLeaf = !srcNodeIds.includes(n.id);
+                isInputLeaf = !dstNodeIdsSet.has(n.id);
+                isOutputLeaf = !srcNodeIdsSet.has(n.id);
             }
             else if (currNodeLevel < 0) {
-                isInputLeaf = !dstNodeIds.includes(n.id);
+                isInputLeaf = !dstNodeIdsSet.has(n.id);
             }
             else {
-                isOutputLeaf = !srcNodeIds.includes(n.id);
+                isOutputLeaf = !srcNodeIdsSet.has(n.id);
             }
             // input node exist & left of start node
             if (currNodeLevel <= 0 && !isInputLeaf) {
@@ -337,25 +572,29 @@ var DataProcessor = /** @class */ (function (_super) {
             }
             // column:  always true if isStartNode === true
             var columnRefreshList = n.isStartNode
-                ? __spreadArray(__spreadArray([], __read(n.children), false), __read(n.relatedColumns), false) : n.relatedColumns;
-            columnRefreshList === null || columnRefreshList === void 0 ? void 0 : columnRefreshList.forEach(function (col) {
+                ? (n.children || []).concat(n.relatedColumns || [])
+                : n.relatedColumns || [];
+            // 处理列的展开/折叠状态
+            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 = !dstNodeIds.includes(col.id);
-                    isOutputLeaf_col = !srcNodeIds.includes(col.id);
+                    isInputLeaf_col = !dstNodeIdsSet.has(col.id);
+                    isOutputLeaf_col = !srcNodeIdsSet.has(col.id);
                 }
                 else if (currNodeLevel < 0) {
-                    isInputLeaf_col = !dstNodeIds.includes(col.id);
+                    isInputLeaf_col = !dstNodeIdsSet.has(col.id);
                 }
                 else {
-                    isOutputLeaf_col = !srcNodeIds.includes(col.id);
+                    isOutputLeaf_col = !srcNodeIdsSet.has(col.id);
                 }
                 // input node exist & left of start node
                 if (currNodeLevel <= 0 && !isInputLeaf_col) {
@@ -376,186 +615,8 @@ var DataProcessor = /** @class */ (function (_super) {
             });
         });
     };
-    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;
-        data.edges.forEach(function (edge) {
-            edge.visible = _this.getEdgeVisible(edge, edgeDashVisible, onlyShowRelated, isFoldChecked);
-        });
-    };
-    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) {
-        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;
-            }
-        }
-        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;
+    TableDataProcessor.startNodes = [];
+    return TableDataProcessor;
 }(BaseManager));
-export { DataProcessor };
-//# sourceMappingURL=DataProcessor.js.map
+export { TableDataProcessor };
+//# sourceMappingURL=TableDataProcessor.js.map