✨ add id, source, target on data when check data

antvis · Jan 7, 2020 · 591a705 · 591a705
1 parent ec1b63d
commit 591a705
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/packages/graphin/src/controller/layout/dataChecking.ts b/packages/graphin/src/controller/layout/dataChecking.ts
@@ -29,6 +29,10 @@ const dataChecking = (data: Data = { nodes: [], edges: [] }): Data => {
       return {
         shape: node.shape || 'CircleNode',
         ...node,
+        data: {
+          ...node.data,
+          id: node.id,
+        },
       };
     });
 
@@ -67,6 +71,11 @@ const dataChecking = (data: Data = { nodes: [], edges: [] }): Data => {
           dist: 20,
         },
         ...edge,
+        data: {
+          ...edge.data,
+          source: edge.source,
+          target: edge.target,
+        },
       };
     });