feat: update graphin

antvis · Nov 21, 2023 · d69bce7 · d69bce7
1 parent b15405b
commit d69bce7
Showing 1 changed file with 20 additions and 3 deletions.
diff --git a/packages/graphin/src/compatible/graph.ts b/packages/graphin/src/compatible/graph.ts
@@ -37,10 +37,20 @@ export default (graph: IGraph) => {
     },
     paint: () => {},
     getNodes: () => {
-      return graph.getAllNodesData();
+      return graph.getAllNodesData().map(item => {
+        return {
+          ...item,
+          getModel: () => item,
+        };
+      });
     },
     getEdges: () => {
-      return graph.getAllEdgesData();
+      return graph.getAllEdgesData().map(item => {
+        return {
+          ...item,
+          getModel: () => item,
+        };
+      });
     },
     clearItemStates: node => {
       const isString = typeof node === 'string';
@@ -51,7 +61,14 @@ export default (graph: IGraph) => {
       }
     },
     findById: id => {
-      return graph.getItemById(id);
+      const { graphCore } = graph.dataController;
+      return {
+        getNeighbors: () => {
+          return graphCore.getNeighbors(id);
+        },
+        getEdges: () => graphCore.bothEdgesMap.get(id),
+        ...graph.getItemById(id),
+      };
     },
   });
   return graph;