aiohttp - 多个 websockets，单个会话？答案

【问题标题】：aiohttp - multiple websockets, single session?aiohttp - 多个 websockets，单个会话？
【发布时间】：2018-03-07 19:45:44
【问题描述】：

我越来越熟悉这些库，但我被以下情况难住了：

我希望不断处理来自两个不同网站的 websocket 的更新消息。

但是考虑到 aiohttp.ClientSession() 对象应该存在于协程中，我无法使用单个会话变量来实现这一点。

import asyncio
import aiohttp

url1 = 'wss://example.com'

async def main():
    session = aiohttp.ClientSession()
    async with session.ws_connect(url1) as ws1:
        async for msg in ws1:
            # do some processing

loop = asyncio.get_event_loop()
loop.create_task(main())
loop.run_forever()

以上内容适用于单个 websocket 连接。但是因为 async for msg in ws: 是一个无限循环，所以我看不到我可以将这个异步循环的 ws2 版本放在哪里。

【问题讨论】：

标签： python python-asyncio aiohttp

【解决方案1】：

由于每个 websocket 都需要自己的无限循环，因此您可以将其抽象为一个协程，该协程为该 websocket 提供服务并接受来自其调用者的会话。调用协程将使用loop.create_task“在后台”创建服务任务，也可以从协程调用：

async def setup():
    session = aiohttp.ClientSession()
    loop = asyncio.get_event_loop()
    loop.create_task(serve(url1, session))
    loop.create_task(serve(url2, session))

async def serve(url, session):
    async with session.ws_connect(url) as ws:
        async for msg in ws:
            ...

loop = asyncio.get_event_loop()
loop.run_until_complete(setup())
loop.run_forever()

【讨论】：