chore: delay unmount child suspense

Author: edison1105

packages/runtime-core/src/components/Suspense.ts

@@ -408,6 +408,7 @@ export interface SuspenseBoundary {
   isInFallback: boolean
   isHydrating: boolean
   isUnmounted: boolean
+  preEffects: Function[]
   effects: Function[]
   resolve(force?: boolean, sync?: boolean): void
   fallback(fallbackVNode: VNode): void
@@ -488,6 +489,7 @@ function createSuspenseBoundary(
     isInFallback: !isHydrating,
     isHydrating,
     isUnmounted: false,
+    preEffects: [],
     effects: [],
 
     resolve(resume = false, sync = false) {
@@ -508,6 +510,7 @@ function createSuspenseBoundary(
         activeBranch,
         pendingBranch,
         pendingId,
+        preEffects,
         effects,
         parentComponent,
         container,
@@ -518,6 +521,10 @@ function createSuspenseBoundary(
       if (suspense.isHydrating) {
         suspense.isHydrating = false
       } else if (!resume) {
+        if (preEffects) {
+          preEffects.forEach(e => e())
+          preEffects.length = 0
+        }
         delayEnter =
           activeBranch &&
           pendingBranch!.transition &&
@@ -723,25 +730,29 @@ function createSuspenseBoundary(
     },
 
     unmount(parentSuspense, doRemove) {
-      if (parentSuspense && parentSuspense.deps > 0) {
-        return
-      }
-      suspense.isUnmounted = true
-      if (suspense.activeBranch) {
-        unmount(
-          suspense.activeBranch,
-          parentComponent,
-          parentSuspense,
-          doRemove,
-        )
+      const performUnmount = () => {
+        suspense.isUnmounted = true
+        if (suspense.activeBranch) {
+          unmount(
+            suspense.activeBranch,
+            parentComponent,
+            parentSuspense,
+            doRemove,
+          )
+        }
+        if (suspense.pendingBranch) {
+          unmount(
+            suspense.pendingBranch,
+            parentComponent,
+            parentSuspense,
+            doRemove,
+          )
+        }
       }
-      if (suspense.pendingBranch) {
-        unmount(
-          suspense.pendingBranch,
-          parentComponent,
-          parentSuspense,
-          doRemove,
-        )
+      if (parentSuspense && parentSuspense.deps > 0) {
+        parentSuspense.preEffects.push(performUnmount)
+      } else {
+        performUnmount()
       }
     },
   }