Merge pull request #27 from gky360/fix/cleanup-old-component

Add cleanup function to remove old component
gky360 · Nov 10, 2021 · 57f984f · 57f984f
2 parents a776398 + 4cc4498
commit 57f984f
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/components/Component/Component.tsx b/src/components/Component/Component.tsx
@@ -72,6 +72,14 @@ export const Component: React.FC<ComponentProps> = ({
     }
   }, [component, onLoad]);
 
+  useEffect(() => {
+    return () => {
+      if (component) {
+        stage.removeComponent(component);
+      }
+    };
+  }, [component, stage]);
+
   useEffect(() => {
     if (component) {
       const nextParams = mergeParams(params, ComponentDefaultParameters);