0

我有一个具有同步和非同步代码的应用程序（因为它进行计算）。

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, *args, **kwargs))
            return sync_func

        for func_name, func_addr in coro_methods:
            setattr(self, func_name, sync_method_factory(func_addr))
4

0 回答 0