@flowgram.ai/free-layout-core 0.5.2 → 0.5.3

package/dist/esm/index.js

@@ -2536,12 +2536,14 @@ var WorkflowDocument = class extends FlowDocument {
    * 批量添加节点
    */
   batchAddFromJSON(json, options) {
-    const { parent = this.root } = options ?? {};
+    const { parent = this.root, onNodeCreated, onEdgeCreated } = options ?? {};
     const parentID = this.getNodeSubCanvas(parent)?.canvasNode.id ?? parent.id;
     const processedJSON = buildGroupJSON(json);
     const nodes = processedJSON.nodes.map(
       (nodeJSON) => this._createWorkflowNode(nodeJSON, {
-        parentID
+        parentID,
+        onNodeCreated,
+        onEdgeCreated
       })
     );
     const edges = processedJSON.edges.map((edge) => this.createWorkflowLine(edge, parentID)).filter(Boolean);
@@ -3563,6 +3565,7 @@ var WorkflowOperationBaseServiceImpl = class extends FlowOperationBaseServiceImp
       edges: json.edges ?? []
     };
     const oldNodes = this.document.getAllNodes();
+    const oldEdges = this.linesManager.getAllLines();
     const oldPositionMap = new Map(
       oldNodes.map((node) => [
         node.id,
@@ -3574,11 +3577,17 @@ var WorkflowOperationBaseServiceImpl = class extends FlowOperationBaseServiceImp
     );
     const newNodes = [];
     const newEdges = [];
-    this.linesManager.getAllLines().map((line) => line.dispose());
     this.document.batchAddFromJSON(workflowJSON, {
       onNodeCreated: (node) => newNodes.push(node),
       onEdgeCreated: (edge) => newEdges.push(edge)
     });
+    const newEdgeIDSet = new Set(newEdges.map((edge) => edge.id));
+    oldEdges.forEach((edge) => {
+      if (!newEdgeIDSet.has(edge.id)) {
+        edge.dispose();
+        return;
+      }
+    });
     const newNodeIDSet = new Set(newNodes.map((node) => node.id));
     oldNodes.forEach((node) => {
       if (!newNodeIDSet.has(node.id)) {