Another wrapping test with global ThreadPoolExecutor

back/boxtribute_server/graph_ql/loaders.py

@@ -1,5 +1,6 @@
 import asyncio
 from collections import defaultdict
+from concurrent.futures import ThreadPoolExecutor
 from datetime import datetime
 from functools import partial
 
@@ -38,6 +39,9 @@ def load(self, key):
         return super().load(key)
 
 
+executor = ThreadPoolExecutor(max_workers=5)
+
+
 async def select(
     model, /, *conditions, fields=None, join_kwargs=None, group_field=None
 ):
@@ -53,9 +57,8 @@ def utility_function(*conditions, fields, join_kwargs):
             query = query.group_by(group_field)
         return list(query.iterator())
 
-    return await asyncio.to_thread(
-        utility_function, *conditions, fields=fields, join_kwargs=join_kwargs
-    )
+    f = partial(utility_function, *conditions, fields=fields, join_kwargs=join_kwargs)
+    return await asyncio.get_running_loop().run_in_executor(executor, f)
 
 
 class SimpleDataLoader(DataLoader):