refactor channel.send to avoid queue.put

This commit is contained in:
Timothy Pogue 2022-11-02 17:57:11 -06:00
parent 2dc55e89e6
commit cbede2e27d

View File

@ -1,5 +1,6 @@
import asyncio
import logging
import time
from threading import RLock
from typing import Any, Dict, List, Optional, Type, Union
from uuid import uuid4
@ -73,16 +74,23 @@ class WebSocketChannel:
Add the data to the queue to be sent.
:returns: True if data added to queue, False otherwise
"""
try:
await asyncio.wait_for(
self.queue.put(data),
timeout=self.drain_timeout
)
except asyncio.TimeoutError:
return False
except RuntimeError:
pass
# This block only runs if the queue is full, it will wait
# until self.drain_timeout for the relay to drain the outgoing
# queue
start = time.time()
while self.queue.full():
await asyncio.sleep(1)
if (time.time() - start) > self.drain_timeout:
return False
# If for some reason the queue is still full, just return False
try:
self.queue.put_nowait(data)
except asyncio.QueueFull:
return False
# If we got here everything is ok
return True
async def recv(self):