aiohttp - 多个websockets,单个会话?

时间:2018-03-07 11:31:19

标签: python python-asyncio aiohttp

我对这些图书馆越来越熟悉,但我对以下情况感到困惑:

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

但是由于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连接。但是因为ws:中msg的 async是一个无限循环,我无法看到我可以把这个异步循环的ws2版本放在哪里。

1 个答案:

答案 0 :(得分:3)

由于每个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()