From afc144c282d0156626432ce167dcf9ac3023ca98 Mon Sep 17 00:00:00 2001
From: GGICE <i@ice.gs>
Date: Fri, 26 Feb 2021 15:43:02 +0800
Subject: [PATCH] fix: when has runtimeLocale.setLocale setLocale not work
 (#543)

---
 .../src/templates/localeExports.tpl           | 48 +++++++++++--------
 1 file changed, 27 insertions(+), 21 deletions(-)

diff --git a/packages/plugin-locale/src/templates/localeExports.tpl b/packages/plugin-locale/src/templates/localeExports.tpl
index 88b832ed3..636ffbd5e 100644
--- a/packages/plugin-locale/src/templates/localeExports.tpl
+++ b/packages/plugin-locale/src/templates/localeExports.tpl
@@ -186,34 +186,40 @@ export const setLocale = (lang: string, realReload: boolean = true) => {
     type: ApplyPluginsType.modify,
     initialValue: {},
   });
+
+  const updater = () => {
+    if (lang !== undefined && !localeExp.test(lang)) {
+      // for reset when lang === undefined
+      throw new Error('setLocale lang format error');
+    }
+    if (getLocale() !== lang) {
+      if (typeof window.localStorage !== 'undefined' && useLocalStorage) {
+        window.localStorage.setItem('umi_locale', lang || '');
+      }
+      setIntl(lang);
+      if (realReload) {
+        window.location.reload();
+      } else {
+        event.emit(LANG_CHANGE_EVENT, lang);
+        // chrome 不支持这个事件。所以人肉触发一下
+        if (window.dispatchEvent) {
+          const event = new Event('languagechange');
+          window.dispatchEvent(event);
+        }
+      }
+    }
+  }
+
   if (typeof runtimeLocale?.setLocale === 'function') {
     runtimeLocale.setLocale({
       lang,
       realReload,
-      updater: (updateLang = lang) => event.emit(LANG_CHANGE_EVENT, updateLang),
+      updater: updater,
     });
     return;
   }
-  if (lang !== undefined && !localeExp.test(lang)) {
-    // for reset when lang === undefined
-    throw new Error('setLocale lang format error');
-  }
-  if (getLocale() !== lang) {
-    if (typeof window.localStorage !== 'undefined' && useLocalStorage) {
-      window.localStorage.setItem('umi_locale', lang || '');
-    }
-    setIntl(lang);
-    if (realReload) {
-      window.location.reload();
-    } else {
-      event.emit(LANG_CHANGE_EVENT, lang);
-      // chrome 不支持这个事件。所以人肉触发一下
-      if (window.dispatchEvent) {
-        const event = new Event('languagechange');
-        window.dispatchEvent(event);
-      }
-    }
-  }
+  
+  updater();
 };
 
 let firstWaring = true;