fix: skip non-existent elements (#6417)

antvis · Oct 17, 2024 · dab826a · dab826a
1 parent 0ab5f8d
commit dab826a
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/packages/g6/src/behaviors/fix-element-size.ts b/packages/g6/src/behaviors/fix-element-size.ts
@@ -217,7 +217,7 @@ export class FixElementSize extends BaseBehavior<FixElementSizeOptions> {
     const { element, model } = this.context;
     const el = element!.getElement(id) as Node | Combo;
 
-    if (this.skipIfExceedViewport(el)) return;
+    if (!el || this.skipIfExceedViewport(el)) return;
 
     const edges = model.getRelatedEdgesData(id);
     edges.forEach((edge) => this.relatedEdgeToUpdate.add(idOf(edge)));
@@ -234,7 +234,7 @@ export class FixElementSize extends BaseBehavior<FixElementSizeOptions> {
     const id = idOf(datum);
     const el = this.context.element!.getElement(id) as Edge;
 
-    if (this.skipIfExceedViewport(el)) return;
+    if (!el || this.skipIfExceedViewport(el)) return;
 
     const config = this.options.edge;
     if (!config) {

diff --git a/packages/g6/src/plugins/edge-bundling/index.ts b/packages/g6/src/plugins/edge-bundling/index.ts
@@ -284,8 +284,8 @@ export class EdgeBundling extends BasePlugin<EdgeBundlingOptions> {
 
     edges.forEach((edge) => {
       const edgeId = idOf(edge);
-      const edgeEl = element!.getElement(edgeId)!;
-      edgeEl.update({ d: getPolylinePath(this.edgePoints[edgeId]) });
+      const edgeEl = element!.getElement(edgeId);
+      edgeEl?.update({ d: getPolylinePath(this.edgePoints[edgeId]) });
     });
   };
 

diff --git a/packages/g6/src/plugins/fisheye/index.ts b/packages/g6/src/plugins/fisheye/index.ts
@@ -277,7 +277,7 @@ export class Fisheye extends BasePlugin<FisheyeOptions> {
 
     const update = (nodeId: ID, style: NodeStyle) => {
       const node = element!.getElement(nodeId) as Node;
-      node.update(style);
+      node?.update(style);
 
       graph.getRelatedEdgesData(nodeId).forEach((datum) => {
         relatedEdges.add(idOf(datum));