fix: useLayoutEffect will be warning in ssr mode (#533)

* fix: useLayoutEffect will be warning in ssr mode * format the code
umijs · Feb 20, 2021 · 2b2c169 · 2b2c169
1 parent d931276
commit 2b2c169
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/packages/plugin-locale/src/templates/locale.tpl b/packages/plugin-locale/src/templates/locale.tpl
@@ -27,6 +27,13 @@ export function _onCreate() {
   setIntl(locale);
 }
 
+const useIsomorphicLayoutEffect =
+  typeof window !== 'undefined' &&
+  typeof window.document !== 'undefined' &&
+  typeof window.document.createElement !== 'undefined'
+    ? React.useLayoutEffect
+    : React.useEffect
+
 export const _LocaleContainer = (props:any) => {
   const [locale, setLocale] = React.useState(() => getLocale());
   const [intl, setContainerIntl] = React.useState(() => getIntl(locale, true));
@@ -41,7 +48,7 @@ export const _LocaleContainer = (props:any) => {
     setContainerIntl(getIntl(locale));
   };
 
-  React.useLayoutEffect(() => {
+  useIsomorphicLayoutEffect(() => {
     event.on(LANG_CHANGE_EVENT, handleLangChange);
     {{#Title}}
     // avoid reset route title