[runtime/fix] fix locale wstring conversion error (#1921)

wenet-e2e · Jul 21, 2023 · a247e18 · a247e18
1 parent f97c460
commit a247e18
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/runtime/core/utils/string.cc b/runtime/core/utils/string.cc
@@ -148,13 +148,17 @@ std::string ProcessBlank(const std::string& str, bool lowercase) {
     }
     // NOTE: convert string to wstring
     //       see issue 745: https://github.com/wenet-e2e/wenet/issues/745
-    std::locale loc("");
-    std::wstring_convert<std::codecvt_utf8<wchar_t>, wchar_t> converter;
-    std::wstring wsresult = converter.from_bytes(result);
-    for (auto& c : wsresult) {
-      c = lowercase ? tolower(c, loc) : toupper(c, loc);
+    try {
+      std::locale loc("");
+      std::wstring_convert<std::codecvt_utf8<wchar_t>, wchar_t> converter;
+      std::wstring wsresult = converter.from_bytes(result);
+      for (auto& c : wsresult) {
+        c = lowercase ? tolower(c, loc) : toupper(c, loc);
+      }
+      result = converter.to_bytes(wsresult);
+    } catch (std::exception& e) {
+      LOG(ERROR) << "convert wstring error " << e.what();
     }
-    result = converter.to_bytes(wsresult);
   }
   return result;
 }