龙卷风网络套接字客户端:如何异步on_message?(从未等待过协程)

Question

如何使on_message函数在我的Tornado WebSocketClient中异步工作？

我想我需要等待on_message函数，但我不知道如何。

或者，在我尝试实现异步 WebSocketClient 的方式上是否存在根本性的误解？

import tornado.websocket
from tornado.queues import Queue
from tornado import gen
import json

q = Queue()
class WebsocketClient():
def __init__(self, url, connections):
self.url = url
self.connections = connections
print("CLIENT started")
print("CLIENT initial connections: ", len(self.connections))
async def send_message(self):
async for message in q:
try:
msg = json.loads(message)
print(message)
await gen.sleep(0.001)
finally:
q.task_done()
async def update_connections(self, connections):
self.connections = connections
print("CLIENT updated connections: ", len(self.connections))
async def on_message(self, message):
await q.put(message)
await gen.sleep(0.001)
async def connect(self):
client = await tornado.websocket.websocket_connect(url=self.url, on_message_callback=self.on_message)

RuntimeWarning: coroutine 'WebsocketClient.on_message' was never awaited
self._on_message_callback(message)
RuntimeWarning: Enable tracemalloc to get the object allocation traceback

Answer 1

>on_message_callback应该是一个常规函数，而不是协程。当人们使用回调而不是协程时，它旨在用于旧式代码中。

对于较新的异步样式代码，不需要此回调。你可以这样做：

async def connect(self):
client = await tornado.websocket.websocket_connect(url=self.url)
while True:
message = await client.read_message()
if message is None:
# None message means the connection was closed
break
print("Message received:", message)
await q.put(message)
await gen.sleep(0.001)