5

我有一个提供网络套接字的课程，并听 PostgreSQL。使用 asyncpg，当我尝试使用 add_listener 时，出现错误：RuntimeWarning: coroutine was never awaited。如何异步/等待回调。我尝试添加“等待 self.listener”，但它不起作用。

有没有办法以另一种方式处理这个问题？

import asyncio
import http
import websockets
import asyncpg

class App(object):

    def __init__(self, loop):
        self.loop = loop
        self.ws_list = []
        self.conn = None

    async def ws_handler(self, ws, path):
        if self.conn is None:
            self.conn = await asyncpg.connect(user='xxx', password='xxx', database='pgws', host='127.0.0.1')
            await self.conn.add_listener('todo_updates', self.listener)
        print('new socket!!!')
        self.ws_list.append(ws)
        while True:
            await asyncio.sleep(1)

    async def listener(self, conn, pid, channel, payload):
        print(payload)
        for ws in self.ws_list:
            task = asyncio.create_task()
            await ws.send(payload)

if __name__ == "__main__":
    loop = asyncio.get_event_loop()
    app = App(loop)
    start_server = websockets.serve(app.ws_handler, 'localhost', 8766)
    app.loop.run_until_complete(start_server)
    app.loop.run_forever()
4

1 回答 1

7

问题是您传递给的回调asyncpg.Connection.add_listener()是一个协程函数，但它应该是一个简单的同步函数。asyncpg 不会引发错误，因为从技术上讲，它仍然是一个可调用的，它需要一个连接、pid、通道和有效负载，但它在被调用时的行为并不像你期望的那样。

要从同步回调中​​调用异步函数（当事件循环已经在运行时），您需要使用类似asyncio.create_task()（在 Python >=3.7 中）或loop.create_task()（在 Python >=3.4.2 中）或asyncio.ensure_future()（在 Python >=3.4 中.4），像这样：

class App:
    ...  # Your other code here
    def listener(self, conn, pid, channel, payload):
        print(payload)
        for ws in self.ws_list:
            asyncio.create_task(ws.send(payload))

请注意asyncio.create_task()（和其他上述功能）将立即返回，并且不会等待任务完成。该任务将被安排在其他地方一个或多个awaits 之后运行。

于 2019-08-13T01:18:20.570 回答