如何解决如何使 asyncpg 驱动程序通过守护线程从同步代码工作?
我有一个同时包含同步和非同步代码的应用(因为它会进行计算)。
asyncpg 声称比 psycopg2 快得多,这里: https://magic.io/blog/asyncpg-1m-rows-from-postgres-to-python/
我想出了动态修补 DatabaseManager 本身使用 asyncpg 和协程方法的代码。目标是使同步代码可以使用相同的界面。有没有更好的方法来做到这一点?
AsyncioWorkerThread 只是在守护线程中运行的事件循环。 (见这个链接https://stackoverflow.com/questions/67199459/asyncio-loop-in-daemon-thread)
class DatabaseManager(DatabaseManager):
async def test():
asyncio.sleep(1)
print('pretend we are doing work with asyncpg')
class SyncDatabaseManager(DatabaseManager):
"""A sync version of the DatabaseManager."""
def __init__(self,*args,**kwargs):
"""Construct."""
super().__init__(*args,**kwargs)
self.workerThread = AsyncioWorkerThread(daemon=True)
self.workerThread.start()
self.__patch_all_coroutine_methods()
def __patch_all_coroutine_methods(self):
methods = inspect.getmembers(DatabaseManager,predicate=inspect.isfunction)
coro_methods = [(func_name,func_addr) for (func_name,func_addr) in methods if inspect.iscoroutinefunction(func_addr)]
def sync_method_factory(func_addr):
@functools.wraps(func_addr)
def sync_func(*args,**kwargs):
return self.workerThread.submit(func_addr(self,**kwargs))
return sync_func
for func_name,func_addr in coro_methods:
setattr(self,func_name,sync_method_factory(func_addr))
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。