feat: use useMounted

HomyeeKing · Mar 15, 2024 · 1df7f2e · 1df7f2e
1 parent 7dc77fd
commit 1df7f2e
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/packages/runtime/src/dynamic.tsx b/packages/runtime/src/dynamic.tsx
@@ -1,5 +1,6 @@
 import type { ReactNode } from 'react';
-import React, { Suspense, lazy, useEffect, useState } from 'react';
+import React, { Suspense, lazy } from 'react';
+import useMounted from './useMounted.js';
 
 const isServer = import.meta.renderer === 'server';
 
@@ -41,11 +42,7 @@ export function dynamic<P = {}>(loader: Loader<P>, option?: DynamicOptions) {
 
   const LazyComp = lazy(() => realLoader().then(convertModule));
   return (props) => {
-    const [hasMounted, setHasMounted] = useState(false);
-
-    useEffect(() => {
-      setHasMounted(true);
-    }, []);
+    const hasMounted = useMounted();
 
     return ssr || (!ssr && hasMounted) ? (
       <Suspense fallback={<Fallback />}>