From d36f2f3b07ca4520108cbff5d26f11b922c6a7b2 Mon Sep 17 00:00:00 2001
From: edison1105 <daiwei521@126.com>
Date: Sun, 22 Oct 2023 11:10:33 +0800
Subject: [PATCH] chore: improve code

chore: improve code

chore: improve code

chore: improve code

chore: improve code
---
 packages/runtime-core/src/hydration.ts | 20 +++++++++-----------
 packages/runtime-core/src/renderer.ts  | 17 +++++++++++++----
 2 files changed, 22 insertions(+), 15 deletions(-)

diff --git a/packages/runtime-core/src/hydration.ts b/packages/runtime-core/src/hydration.ts
index 58abc7a7b7b..ec28e7d4a89 100644
--- a/packages/runtime-core/src/hydration.ts
+++ b/packages/runtime-core/src/hydration.ts
@@ -15,7 +15,7 @@ import { ComponentInternalInstance } from './component'
 import { invokeDirectiveHook } from './directives'
 import { warn } from './warning'
 import { PatchFlags, ShapeFlags, isReservedProp, isOn } from '@vue/shared'
-import { RendererInternals } from './renderer'
+import { needTransition, RendererInternals } from './renderer'
 import { setRef } from './rendererTemplateRef'
 import {
   SuspenseImpl,
@@ -374,22 +374,20 @@ export function createHydrationFunctions(
       let needCallTransitionHooks = false
       if (isTemplateNode(el)) {
         needCallTransitionHooks =
-          (!parentSuspense ||
-            (parentSuspense && !parentSuspense.pendingBranch)) &&
-          transition &&
-          !transition.persisted &&
-          parentComponent?.vnode.props?.appear
+          needTransition(parentSuspense, transition) &&
+          parentComponent &&
+          parentComponent.vnode.props &&
+          parentComponent.vnode.props.appear
 
         const content = (el as HTMLTemplateElement).content
           .firstChild as Element
         needCallTransitionHooks && transition!.beforeEnter(content)
 
-        vnode.el = content
-        dirs && invokeDirectiveHook(vnode, null, parentComponent, 'beforeMount')
-
         // replace <template> node with inner child
-        el = replaceNode(content, el, parentComponent) as Element
-      } else if (dirs) {
+        vnode.el = el = replaceNode(content, el, parentComponent) as Element
+      }
+
+      if (dirs) {
         invokeDirectiveHook(vnode, null, parentComponent, 'beforeMount')
       }
 
diff --git a/packages/runtime-core/src/renderer.ts b/packages/runtime-core/src/renderer.ts
index 383e17fb0f5..8799ecd473c 100644
--- a/packages/runtime-core/src/renderer.ts
+++ b/packages/runtime-core/src/renderer.ts
@@ -72,6 +72,7 @@ import { initFeatureFlags } from './featureFlags'
 import { isAsyncWrapper } from './apiAsyncComponent'
 import { isCompatEnabled } from './compat/compatConfig'
 import { DeprecationTypes } from './compat/compatConfig'
+import { TransitionHooks } from './components/BaseTransition'
 
 export interface Renderer<HostElement = RendererElement> {
   render: RootRenderFunction<HostElement>
@@ -701,10 +702,7 @@ function baseCreateRenderer(
     }
     // #1583 For inside suspense + suspense not resolved case, enter hook should call when suspense resolved
     // #1689 For inside suspense + suspense resolved case, just call it
-    const needCallTransitionHooks =
-      (!parentSuspense || (parentSuspense && !parentSuspense.pendingBranch)) &&
-      transition &&
-      !transition.persisted
+    const needCallTransitionHooks = needTransition(parentSuspense, transition)
     if (needCallTransitionHooks) {
       transition!.beforeEnter(el)
     }
@@ -2365,6 +2363,17 @@ function toggleRecurse(
   effect.allowRecurse = update.allowRecurse = allowed
 }
 
+export function needTransition(
+  parentSuspense: SuspenseBoundary | null,
+  transition: TransitionHooks | null
+) {
+  return (
+    (!parentSuspense || (parentSuspense && !parentSuspense.pendingBranch)) &&
+    transition &&
+    !transition.persisted
+  )
+}
+
 /**
  * #1156
  * When a component is HMR-enabled, we need to make sure that all static nodes