add hku_to_async utils method

fasiondog · Apr 24, 2024 · 90cc0d9 · 90cc0d9
1 parent 3ec9a92
commit 90cc0d9
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/hikyuu/util/__init__.py b/hikyuu/util/__init__.py
@@ -25,6 +25,7 @@
     'hku_check_throw',
     'hku_check_ignore',
     'hku_catch',
+    'hku_to_async',
     'hku_trace',
     'hku_debug',
     'hku_info',

diff --git a/hikyuu/util/check.py b/hikyuu/util/check.py
@@ -10,6 +10,7 @@
 import sys
 import traceback
 import functools
+import asyncio
 from .mylog import hku_logger
 
 
@@ -142,3 +143,11 @@ def wrappedFunc(*args, **kargs):
         return wrappedFunc
 
     return hku_catch_wrap
+
+
+def hku_to_async(func):
+    @functools.wraps(func)
+    async def async_func(*args, **kwargs):
+        loop = asyncio.get_event_loop()
+        return await loop.run_in_executor(None, func, *args, **kwargs)
+    return async_func