为套接字服务器使用asyncio的推荐方法是:
import asyncio
async def handle_client(reader, writer):
request = (await reader.read(100)).decode()
response = "Data received."
writer.write(response.encode())
async def main():
loop.create_task(asyncio.start_server(handle_client, 'localhost', 15555))
loop = asyncio.get_event_loop()
loop.create_task(main())
loop.run_forever()
这很好用,但现在我需要接收适当的客户端请求,然后使用aiohttp库从第三方restful API获取数据。
这需要创建一个会话变量,如下所示:
from aiohttp import ClientSession
session = ClientSession()
但这也应该在协程本身内部,所以我将它放在main中:
async def main():
session = ClientSession()
loop.create_task(asyncio.start_server(handle_client, '', 55555))
现在我需要将会话变量传递给aiohttp get coroutine以获取其余的API数据:
async with session.get(url, params=params) as r:
try:
return await r.json(content_type='application/json')
except aiohttp.client_exceptions.ClientResponseError:
....
我的问题是如何将会话变量传递给handle_client coroutine,如果它坚持只有读取器,编写器参数和全局变量不帮助我,因为会话必须存在于协同程序中?
答案 0 :(得分:4)
您可以使用临时函数或lambda:
async def main():
session = aiohttp.ClientSession()
await asyncio.start_server(lambda r, w: handle_client(r, w, session),
'', 55555)
这是有效的,因为即使lambda
在技术上不是协程,它的行为就像一个 - 它是一个可调用的,在调用时返回coroutine object。
对于较大的程序,您可能更喜欢基于类的方法,其中一个类封装了多个客户端共享的状态,而不必将其从协程传递到协程。例如:
class ClientContext:
def __init__(self, session):
self.session = session
async def handle_client(self, reader, writer):
# ... here you get reader and writer, but also have
# session etc as self.session ...
async def main():
ctx = ClientContext(aiohttp.ClientSession())
await asyncio.start_server(ctx.handle_client), '', 55555)