Vasanko/sanic
2
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

fix: deprecation warning in `asyncio.wait` (#2383)

Browse Source
This commit is contained in:
Ryu juheon 2022-01-19 15:09:17 +09:00 committed by GitHub
parent b8d991420b
commit bb517ddcca
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
sanic/server/websockets/impl.py
View File

@ -518,8 +518,12 @@ class WebsocketImplProtocol:
) )
try: try:
self.recv_cancel = asyncio.Future() self.recv_cancel = asyncio.Future()
tasks = (
self.recv_cancel,
asyncio.ensure_future(self.assembler.get(timeout)),
)
done, pending = await asyncio.wait( done, pending = await asyncio.wait(
(self.recv_cancel, self.assembler.get(timeout)), tasks,
return_when=asyncio.FIRST_COMPLETED, return_when=asyncio.FIRST_COMPLETED,
) )
done_task = next(iter(done)) done_task = next(iter(done))
@ -570,8 +574,12 @@ class WebsocketImplProtocol:
self.can_pause = False self.can_pause = False
self.recv_cancel = asyncio.Future() self.recv_cancel = asyncio.Future()
while True: while True:
tasks = (
self.recv_cancel,
asyncio.ensure_future(self.assembler.get(timeout=0)),
)
done, pending = await asyncio.wait( done, pending = await asyncio.wait(
(self.recv_cancel, self.assembler.get(timeout=0)), tasks,
return_when=asyncio.FIRST_COMPLETED, return_when=asyncio.FIRST_COMPLETED,
) )
done_task = next(iter(done)) done_task = next(iter(done))