perf: 优化解析脚本导入

O-Jiangweidong · Sep 12, 2024 · 6b53bf4 · 6b53bf4
1 parent 94ef440
commit 6b53bf4
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/apps/behemoth/libs/parser/handle.py b/apps/behemoth/libs/parser/handle.py
@@ -9,17 +9,20 @@
 from common.utils import get_logger
 
 
-if platform.system() == "Darwin":
+if platform.system() == 'Darwin':
     so_dir = 'parse_darwin.so'
 else:
     so_dir = 'parse_linux.so'
 
 
 logger = get_logger(__file__)
 
-lib = ctypes.CDLL(os.path.join(settings.APPS_DIR, 'behemoth', 'libs', 'parser', so_dir))
-lib.Parse.argtypes = [ctypes.c_char_p] # noqa
-lib.Parse.restype = ctypes.c_char_p
+try:
+    lib = ctypes.CDLL(os.path.join(settings.APPS_DIR, 'behemoth', 'libs', 'parser', so_dir))
+    lib.Parse.argtypes = [ctypes.c_char_p] # noqa
+    lib.Parse.restype = ctypes.c_char_p
+except Exception as error:
+    logger.error('Failed to load parse lib')
 
 
 def parse_sql(sql: str) -> List[str]: