feat: 支持 umi3 主应用 (#65)

umijs · Jan 20, 2022 · 0629dec · 0629dec
1 parent ca1ee71
commit 0629dec
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/slave/lifecycles.ts b/src/slave/lifecycles.ts
@@ -53,12 +53,14 @@ export function genMount() {
 }
 
 export function genUnmount(mountElementId: string) {
-  return async (...args: any[]) => {
-    const container = document.getElementById(mountElementId);
+  return async (props: any) => {
+    const container = props?.container
+      ? props.container.querySelector(`#${mountElementId}`)
+      : document.getElementById(mountElementId);
     if (container) {
       ReactDOM.unmountComponentAtNode(container);
     }
     const slaveRuntime = getSlaveRuntime();
-    if (slaveRuntime.unmount) await slaveRuntime.unmount(...args);
+    if (slaveRuntime.unmount) await slaveRuntime.unmount(props);
   };
 }