我越来越熟悉这些库,但我被以下情况难住了:
我希望持续处理来自两个不同网站的 websocket 的更新消息。
但我在使用单个 session 变量实现这一点时遇到了麻烦,因为 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 版本放在哪里。
最佳答案
由于每个 websocket 都需要自己的无限循环,因此您可以将其抽象为一个协程,为该 websocket 提供服务并接受来自其调用者的 session 。调用协程将使用 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()
关于python - aiohttp - 多个 websockets,单个 session ?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/49150853/