如何在 Python 中使用 aiohttp 或 asyncio 创建并行循环？

Question

我想使用 rethinkdb .changes() 功能向用户推送一些消息。消息应在用户没有任何请求的情况下发送。

我正在将 rethinkdb 与 aiohttp 和 websockets 一起使用。工作原理：

用户发送消息
服务器放入rethinkdb
我需要什么：一个额外的循环使用 rethinkdb .changes 函数向连接的用户发送更新

我是这样发起申请的：

@asyncio.coroutine
def init(loop):
    app = Application(loop=loop)
    app['sockets'] = []
    app['susers'] = []
    app.router.add_route('GET', '/', wshandler)
    handler = app.make_handler()
    srv = yield from loop.create_server(handler, '127.0.0.1', 9080)
    print("Server started at http://127.0.0.1:9080")
    return app, srv, handler

在 wshandler 我有一个循环，它处理传入的消息：

@asyncio.coroutine
def wshandler(request):
    resp = WebSocketResponse()
    if not resp.can_prepare(request):
        return Response(
            body=bytes(json.dumps({"error_code": 401}), 'utf-8'),
            content_type='application/json'
        )
    yield from resp.prepare(request)
    request.app['sockets'].append(resp)
    print('Someone connected')
    while True:
        msg = yield from resp.receive()
        if msg.tp == MsgType.text:
            runCommand(msg, resp, request)
        else:
            break
    request.app['sockets'].remove(resp)
    print('Someone disconnected.')
    return resp

如何创建第二个循环将消息发送到同一个打开的连接池？如何使其成为线程安全的？

Answer 1

一般来说，您应该尝试在运行事件循环时尽可能避免线程。

遗憾的是 rethinkdb 不支持开箱即用的 asyncio，但它支持 Tornado & Twisted 框架。因此，您可以 bridge Tornado & asyncio 并使其在不使用线程的情况下工作。

编辑:

正如安德鲁指出的那样 rethinkdb 支持 asyncio。在 2.1.0 之后你大概可以做：

rethinkdb.set_loop_type("asyncio")

然后在您的网络处理程序中：

res = await rethinkdb.table(tbl).changes().run(connection)
while await res.fetch_next():
   ...

如何在 Python 中使用 aiohttp 或 asyncio 创建并行循环？

How to create a parallel loop using aiohttp or asyncio in Python?

python

rethinkdb

python-asyncio

rethinkdb-python

aiohttp