perf: improve the performance of dagre with layers; (#106)

Author: Yanyan-Wang

CHANGELOG.md

@@ -1,5 +1,9 @@
 # ChangeLog
 
+### 0.3.4
+
+- perf: improve the performance of dagre with layers;
+
 ### 0.3.3
 
 - fix: dagre layer failed when the layer number begin with 0;

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@antv/layout",
-  "version": "0.3.3",
+  "version": "0.3.4",
   "description": "graph layout algorithm",
   "main": "lib/index.js",
   "module": "es/index.js",

src/layout/dagre/src/rank/util.ts

@@ -107,6 +107,7 @@ const longestPathWithLayer = (g: Graph) => {
   // minRank += 1; // NOTE: 最小的层级是dummy root，+1
 
   // forward一遍，赋值层级
+  const forwardVisited: Record<string, boolean> = {};
   const dfsForward = (v: string, nextRank: number) => {
     const label = g.node(v)!;
 
@@ -119,6 +120,9 @@ const longestPathWithLayer = (g: Graph) => {
       label.rank = currRank;
     }
 
+    if (forwardVisited[v]) return;
+    forwardVisited[v] = true;
+
     // DFS遍历子节点
     g.outEdges(v)?.map((e) => {
       dfsForward(e.w, currRank + g.edge(e)!.minlen!);