From 5990b7f8f11941ba43d676f27fe8000b04c93be7 Mon Sep 17 00:00:00 2001
From: Fine <fanxue0830@gmail.com>
Date: Thu, 25 Aug 2022 17:59:18 +0800
Subject: [PATCH] update topology

---
 .../components/Graph/linkProcess.ts           |  1 +
 .../components/Graph/nodeProcess.ts           |  1 +
 .../components/ProcessTopology.vue            | 50 ++++++++++++-------
 3 files changed, 35 insertions(+), 17 deletions(-)

diff --git a/src/views/dashboard/related/network-profiling/components/Graph/linkProcess.ts b/src/views/dashboard/related/network-profiling/components/Graph/linkProcess.ts
index 16cdb7d5..e12abad5 100644
--- a/src/views/dashboard/related/network-profiling/components/Graph/linkProcess.ts
+++ b/src/views/dashboard/related/network-profiling/components/Graph/linkProcess.ts
@@ -53,6 +53,7 @@ export const linkElement = (graph: any) => {
 export const anchorElement = (graph: any, funcs: any, tip: any) => {
   const linkEnter = graph
     .append("g")
+    .attr("class", "topo-line-anchor")
     .attr("style", "cursor: move;")
     .on("mouseover", function (event: unknown, d: unknown) {
       tip.html(funcs.tipHtml).show(d, this);
diff --git a/src/views/dashboard/related/network-profiling/components/Graph/nodeProcess.ts b/src/views/dashboard/related/network-profiling/components/Graph/nodeProcess.ts
index 197b1273..8a823542 100644
--- a/src/views/dashboard/related/network-profiling/components/Graph/nodeProcess.ts
+++ b/src/views/dashboard/related/network-profiling/components/Graph/nodeProcess.ts
@@ -20,6 +20,7 @@ import { Node } from "@/types/topology";
 export default (d3: any, graph: any, funcs: any, tip: any) => {
   const nodeEnter = graph
     .append("g")
+    .attr("class", "topo-node")
     .call(
       d3
         .drag()
diff --git a/src/views/dashboard/related/network-profiling/components/ProcessTopology.vue b/src/views/dashboard/related/network-profiling/components/ProcessTopology.vue
index 2b352403..06e055f9 100644
--- a/src/views/dashboard/related/network-profiling/components/ProcessTopology.vue
+++ b/src/views/dashboard/related/network-profiling/components/ProcessTopology.vue
@@ -95,8 +95,8 @@ async function init() {
   if (!networkProfilingStore.nodes.length) {
     return;
   }
-  drawGraph();
-  createLayout();
+  freshNodes();
+  useThrottleFn(resize, 500)();
 }
 
 function drawGraph() {
@@ -114,10 +114,22 @@ function drawGraph() {
     .attr("class", "svg-graph")
     .attr("transform", `translate(${diff.value[0]}, ${diff.value[1]})`);
   graph.value.call(tip.value);
-  node.value = graph.value.append("g").selectAll(".topo-node");
-  link.value = graph.value.append("g").selectAll(".topo-call");
-  anchor.value = graph.value.append("g").selectAll(".topo-line-anchor");
-  arrow.value = graph.value.append("g").selectAll(".topo-line-arrow");
+  node.value = graph.value
+    .append("g")
+    .attr("class", "nodes")
+    .selectAll(".topo-node");
+  link.value = graph.value
+    .append("g")
+    .attr("class", "calls")
+    .selectAll(".topo-call");
+  anchor.value = graph.value
+    .append("g")
+    .attr("class", "anchors")
+    .selectAll(".topo-line-anchor");
+  arrow.value = graph.value
+    .append("g")
+    .attr("class", "arrows")
+    .selectAll(".topo-line-arrow");
   svg.value.call(zoom(d3, graph.value, diff.value));
   svg.value.on("click", (event: any) => {
     event.stopPropagation();
@@ -126,7 +138,6 @@ function drawGraph() {
     networkProfilingStore.setLink(null);
     dashboardStore.selectWidget(props.config);
   });
-  useThrottleFn(resize, 500)();
 }
 
 function hexGrid(n = 1, radius = 1, origin = [0, 0]) {
@@ -191,7 +202,7 @@ function createLayout() {
   }
   const linePath = d3.line();
   linePath.curve(d3.curveLinearClosed);
-  const hexPolygon = graph.value.append("g");
+  const hexPolygon = graph.value.append("g").attr("class", "hex-polygon");
   hexPolygon
     .append("path")
     .attr("d", linePath(vertices))
@@ -298,14 +309,20 @@ function drawTopology(nodeArr: any[]) {
   ).merge(node.value);
   // line element
   const obj = {} as any;
-  const calls = networkProfilingStore.calls.reduce((prev: any[], next: any) => {
-    if (obj[next.targetId + next.sourceId]) {
-      next.lowerArc = true;
-    }
-    obj[next.sourceId + next.targetId] = true;
-    prev.push(next);
-    return prev;
-  }, []);
+  const calls = networkProfilingStore.calls.reduce(
+    (
+      prev: (Call & { targetId: string; sourceId: string })[],
+      next: Call & { targetId: string; sourceId: string }
+    ) => {
+      if (obj[next.targetId + next.sourceId]) {
+        next.lowerArc = true;
+      }
+      obj[next.sourceId + next.targetId] = true;
+      prev.push(next);
+      return prev;
+    },
+    []
+  );
 
   link.value = link.value.data(calls, (d: Call) => d.id);
   link.value.exit().remove();
@@ -337,7 +354,6 @@ function shuffleArray(array: number[][]) {
     [array[i], array[j]] = [array[j], array[i]];
   }
 }
-
 function handleLinkClick(event: any, d: Call) {
   event.stopPropagation();
   networkProfilingStore.setNode(null);