From bf1d020d8939eb97e395812d628b6383efac11f3 Mon Sep 17 00:00:00 2001
From: betaqi <3188864257@qq.com>
Date: Thu, 21 Aug 2025 16:23:13 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20topo=E5=8A=A8=E7=94=BB=E6=B5=81?=
 =?UTF-8?q?=E7=A8=8B=E5=B7=B2=E7=BB=8Fem=E6=8C=87=E5=90=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/views/testG6/components/Node.vue | 11 ++++++
 src/views/testG6/index.vue           | 47 ++++++++++++++----------
 src/views/testG6/utils/MyLineEdge.ts | 54 +++++++++++++++++++---------
 src/views/testG6/utils/MyNode.ts     | 43 ++++++++++++++++++++++
 src/views/testG6/utils/index.ts      | 46 +++++++++++-------------
 vite.config.ts                       |  2 +-
 6 files changed, 141 insertions(+), 62 deletions(-)
 create mode 100644 src/views/testG6/components/Node.vue
 create mode 100644 src/views/testG6/utils/MyNode.ts

diff --git a/src/views/testG6/components/Node.vue b/src/views/testG6/components/Node.vue
new file mode 100644
index 0000000..3499f9b
--- /dev/null
+++ b/src/views/testG6/components/Node.vue
@@ -0,0 +1,11 @@
+<script setup lang="ts">
+
+</script>
+
+<template>
+  <div>node</div>
+</template>
+
+<style scoped lang="scss">
+
+</style>
\ No newline at end of file
diff --git a/src/views/testG6/index.vue b/src/views/testG6/index.vue
index efeacfe..4848380 100644
--- a/src/views/testG6/index.vue
+++ b/src/views/testG6/index.vue
@@ -1,17 +1,18 @@
 <template>
-  <div>Use G6 in Vue</div>
   <div id="container" class="w-full h-full"></div>
 </template>
 
 <script setup lang="ts">
 import DeviceData from './utils/data';
-import { MyLineEdge } from './utils/MyLineEdge';
-import { DeviceType, flattenTree, GenerateGraphData } from "./utils";
+import { EmLineEdge, MyLineEdge } from './utils/MyLineEdge';
+import { DeviceType, flattenTree, GenerateGraphData, NODE_SIZE } from "./utils";
 import { ExtensionCategory, Graph, register } from '@antv/g6';
-
+import { UserCardNode } from "@/views/testG6/utils/MyNode";
 
 
 onMounted(() => {
+  register(ExtensionCategory.EDGE, 'em-line-edge', EmLineEdge)
+  register(ExtensionCategory.NODE, 'user-card-node', UserCardNode);
   register(ExtensionCategory.EDGE, 'my-line-edge', MyLineEdge);
   const device = flattenTree(DeviceData)
   const graphData = new GenerateGraphData(device)
@@ -26,8 +27,15 @@ onMounted(() => {
       nodesep: 50,
       ranksep: 50,
     },
-    edge: {
-      type: 'my-line-edge',
+    // edge: {
+    //   type: 'my-line-edge',
+    // },
+    node: {
+      type: 'user-card-node',
+      style: {
+        size: NODE_SIZE,
+        customData: (d: any) => d.data,
+      },
     },
     data: {
       nodes: graphData.getNodes(),
@@ -48,21 +56,22 @@ onMounted(() => {
     );
 
     const [x, y] = graph.getElementPosition(root.id);
-    console.log(x, y)
-    console.log(emNodes)
 
     graph.updateNodeData(
-      emNodes.map((n, idx) => ({
-        id: n.id,
-        style: Object.assign(
-          {},
-          n.style,
-          { // 更新Em 节点位置
-            x: x + 100,
-            y: -y - (emNodes.length * 10) + (idx * 60),
-          },
-        )
-      }))
+      emNodes.map((n, idx) => {
+        // 获取节点大小
+        return {
+          id: n.id,
+          style: Object.assign(
+            {},
+            n.style,
+            { // 更新Em 节点位置
+              x: x + NODE_SIZE[0] * 2,
+              y: -y - (emNodes.length * 10) + (idx * 60),
+            },
+          )
+        }
+      })
     );
 
     await graph.draw();
diff --git a/src/views/testG6/utils/MyLineEdge.ts b/src/views/testG6/utils/MyLineEdge.ts
index 35d8e0c..8e57171 100644
--- a/src/views/testG6/utils/MyLineEdge.ts
+++ b/src/views/testG6/utils/MyLineEdge.ts
@@ -2,7 +2,7 @@ import { BaseEdge } from '@antv/g6';
 import type { PathArray } from "@antv/util";
 import { Circle } from '@antv/g';
 import { subStyleProps } from '@antv/g6';
-import type { Device } from "./index";
+import { type Device, NODE_SIZE } from "./index";
 
 
 export class MyLineEdge extends BaseEdge {
@@ -38,28 +38,14 @@ export class MyLineEdge extends BaseEdge {
     // });
   }
 
-  // 这里保持你的路径逻辑
   protected getKeyPath(attrs: any): PathArray {
-    // const { sourceNode, targetNode } = this;
-    const [sourcePoint, targetPoint] = this.getEndpoints(attrs,false)
+    const [sourcePoint, targetPoint] = this.getEndpoints(attrs, false)
     const [sx, sy] = [sourcePoint[0], sourcePoint[1]];
     const [tx, ty] = [targetPoint[0], targetPoint[1]];
-    // const prev = this.context.model.getRelatedEdgesData(this.sourceNode.id) // 获取源节点的相关边数据
-    //   .find((edge) => edge.target === this.targetNode.id) as Device;
-    // console.log(prev)
-    // const [sx, sy] = sourceNode.getPosition();
-    // const [tx, ty] = targetNode.getPosition();
-    // const [x1, y1] = sourceNode.getPosition();
-    // const [x2, y2] = targetNode.getPosition();
 
     // 固定的主干 X
     const busX = attrs.busX ?? tx;
 
-    // return [
-    //   ['M', x1, y1],
-    //   ['L', x2, y2],
-    // ];
-
     return [
       ['M', sx, sy],
       ['L', busX, sy],
@@ -70,3 +56,39 @@ export class MyLineEdge extends BaseEdge {
 
 }
 
+
+export class EmLineEdge extends BaseEdge {
+
+  getMarkerStyle(attributes: object) {
+    return {
+      r: 4,
+      fill: '#c3d5f9',
+      offsetPath: this.shapeMap.key, ...subStyleProps(attributes, 'marker')
+    };
+  }
+
+  onCreate() {
+    const marker = this.upsert('marker', Circle, this.getMarkerStyle(this.attributes), this)!;
+
+    const delay = 0;
+    marker.animate(
+      [{ offsetDistance: 0 }, { offsetDistance: 1 }],
+      {
+        duration: 3000,
+        iterations: Infinity,
+        delay,
+      }
+    );
+
+  }
+
+  getKeyPath(attributes: any): PathArray {
+    const [sourcePoint, targetPoint] = this.getEndpoints(attributes,false);
+    return [
+      ['M', sourcePoint[0], sourcePoint[1]],
+      ['L', targetPoint[0] / 2 + (1 / 2) * sourcePoint[0], sourcePoint[1]],
+      ['L', targetPoint[0] / 2 + (1 / 2) * sourcePoint[0], targetPoint[1]],
+      ['L', targetPoint[0], targetPoint[1]],
+    ];
+  }
+}
diff --git a/src/views/testG6/utils/MyNode.ts b/src/views/testG6/utils/MyNode.ts
new file mode 100644
index 0000000..49bdb4c
--- /dev/null
+++ b/src/views/testG6/utils/MyNode.ts
@@ -0,0 +1,43 @@
+import { Graph, register, Rect, ExtensionCategory } from '@antv/g6';
+import { NODE_SIZE } from "./index";
+
+export class UserCardNode extends Rect {
+  get nodeData() {
+    return this.context.graph.getNodeData(this.id);
+  }
+
+  get data() {
+    return this.nodeData.data || {};
+  }
+
+  // 用户名样式
+  getUsernameStyle(attributes: any) {
+    return {
+      // x: -20,
+      text: this.data.name || '',
+      fontSize: 12,
+      fill: '#262626',
+      fontWeight: 'bold',
+      textAlign: 'center',
+      textBaseline: 'middle',
+    };
+  }
+
+  drawUsernameShape(attributes: any, container: any) {
+    if (!attributes.customData.name) return;
+
+    const usernameStyle = this.getUsernameStyle(attributes);
+    this.upsert('username', 'text', usernameStyle, container);
+  }
+
+
+  render(attributes: any, container: any) {
+    // 渲染基础矩形
+    super.render(attributes, container);
+
+
+    this.drawUsernameShape(attributes, container);
+  }
+}
+
+
diff --git a/src/views/testG6/utils/index.ts b/src/views/testG6/utils/index.ts
index b254aa6..d8cd484 100644
--- a/src/views/testG6/utils/index.ts
+++ b/src/views/testG6/utils/index.ts
@@ -5,6 +5,8 @@ import type { NodeStyle } from "@antv/g6/lib/spec/element/node";
 
 export type Device = any
 
+export const NODE_SIZE: [number, number] = [100, 36]
+
 export enum DeviceType {
   Ems,
   Ecu,
@@ -22,23 +24,17 @@ export enum DeviceType {
 
 const VIRTUAL_NODE = 'virtual-node'
 
-const edgeDefaultStyle: EdgeStyle = {
-  router: {
-    type: 'orth',
-  },
-}
-
-const nodeDefaultStyle: NodeStyle = {
-  ports: [
-    { key: 'top', placement: [0.5, 0], r: 0 },
-    { key: 'bottom', placement: [0.5, 0.98], r: 0 },
-    { key: 'left', placement: [0, 0.5], r: 0 },
-    { key: 'right', placement: [1, 0.5], r: 0 }
-  ]
-}
+// const nodeDefaultStyle: NodeStyle = {
+//   ports: [
+//     { key: 'top', placement: [0.5, 0], r: 0 },
+//     // { key: 'bottom', placement: [0.5, 1], r: 0 },
+//     { key: 'left', placement: [0, 0.5], r: 0 },
+//     { key: 'right', placement: [1, 0.5], r: 0 }
+//   ]
+// }
 
 
-export function flattenTree(tree: Device[], depth = 0): Device[] {
+export function flattenTree(tree: Device[], depth = 1): Device[] {
   return flatMap(tree, (node: Device,) => {
     const { children, ...rest } = node
     const current = { ...rest, depth }
@@ -64,11 +60,14 @@ export class GenerateGraphData {
       const isNeedPoint = device.type === DeviceType.Ems || device.type === DeviceType['Em-Measure'] || device.type === DeviceType.Em
       this.nodes.push({
         id: String(device.id),
+        type: 'user-card-node',
         label: device.name,
         data: device,
-        style: Object.assign(isNeedPoint ? nodeDefaultStyle : {},
+        style: Object.assign(
           {
-            labelText: device.name
+            // labelText: device.name,
+            opacity: .3,
+            deviceName: (d: any) => d.data.name,
           })
       })
 
@@ -102,12 +101,12 @@ export class GenerateGraphData {
         this.edges.push({
           source: customData.parentId,
           target: String(node.id),
-          type: 'polyline',
+          type: 'my-line-edge',
           data: customData,
-          style: edgeDefaultStyle,
         })
         continue;
       }
+      //
 
       if (findParentNode) {
         if (this.isEms(customData.type)) continue;
@@ -116,12 +115,8 @@ export class GenerateGraphData {
           this.edges.push({
             target: String(customData.parentId),
             source: String(node.id),
-            type: 'polyline',
+            type: 'em-line-edge',
             data: customData,
-            style: Object.assign(edgeDefaultStyle, {
-              sourcePort: 'bottom',
-              targetPort: 'right',
-            }),
           })
           continue;
         }
@@ -129,9 +124,8 @@ export class GenerateGraphData {
         this.edges.push({
           source: virtualId,
           target: String(node.id),
-          type: 'polyline',
+          type: 'my-line-edge',
           data: customData,
-          style: edgeDefaultStyle,
         })
       }
     }
diff --git a/vite.config.ts b/vite.config.ts
index a944842..cc92116 100644
--- a/vite.config.ts
+++ b/vite.config.ts
@@ -65,7 +65,7 @@ export default defineConfig({
     host: '0.0.0.0',
     proxy: {
       '/remoteServer': {
-        target: 'http://192.168.1.99:8080/',
+        target: 'http://192.168.1.199:8080/',
         changeOrigin: true,
         secure: false,
         ws: true,