fix: add defense for view in transform plugin (#3092)

antvis · Dec 24, 2022 · fb8098c · fb8098c
1 parent c250cab
commit fb8098c
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/packages/x6-plugin-transform/src/transform.ts b/packages/x6-plugin-transform/src/transform.ts
@@ -107,7 +107,11 @@ export class TransformImpl extends View<TransformImpl.EventArgs> {
 
   render() {
     this.renderHandles()
-    this.view.addClass(Private.NODE_CLS)
+
+    if (this.view) {
+      this.view.addClass(Private.NODE_CLS)
+    }
+
     Dom.addClass(this.container, this.containerClassName)
     Dom.toggleClass(
       this.container,
@@ -153,7 +157,9 @@ export class TransformImpl extends View<TransformImpl.EventArgs> {
   }
 
   remove() {
-    this.view.removeClass(Private.NODE_CLS)
+    if (this.view) {
+      this.view.removeClass(Private.NODE_CLS)
+    }
     return super.remove()
   }