Madiator2011/lbry-sdk
1
0
Fork 0
forked from LBRYCommunity/lbry-sdk
Code Pull requests Activity

feedback

Browse source
This commit is contained in:
Jack Robison 2020-07-20 12:20:00 -04:00
parent 1f82a8b99e
commit df5b6a8380
No known key found for this signature in database
GPG key ID: DF25C68FE0239BB2
1 changed files with 5 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
lbry/stream/reflector/server.py
View file

@ -16,9 +16,8 @@ log = logging.getLogger(__name__)
class ReflectorServerProtocol(asyncio.Protocol): class ReflectorServerProtocol(asyncio.Protocol):
def __init__(self, blob_manager: 'BlobManager', response_chunk_size: int = 10000, def __init__(self, blob_manager: 'BlobManager', response_chunk_size: int = 10000,
stop_event: typing.Optional[asyncio.Event] = None, stop_event: asyncio.Event = None, incoming_event: asyncio.Event = None,
incoming_event: typing.Optional[asyncio.Event] = None, not_incoming_event: asyncio.Event = None):
not_incoming_event: typing.Optional[asyncio.Event] = None):
self.loop = asyncio.get_event_loop() self.loop = asyncio.get_event_loop()
self.blob_manager = blob_manager self.blob_manager = blob_manager
self.server_task: asyncio.Task = None self.server_task: asyncio.Task = None
@ -45,7 +44,7 @@ class ReflectorServerProtocol(asyncio.Protocol):
self.transport = transport self.transport = transport
self.wait_for_stop_task = self.loop.create_task(self.wait_for_stop()) self.wait_for_stop_task = self.loop.create_task(self.wait_for_stop())
def connection_lost(self, exc: typing.Optional[Exception]) -> None: def connection_lost(self, exc):
if self.wait_for_stop_task: if self.wait_for_stop_task:
self.wait_for_stop_task.cancel() self.wait_for_stop_task.cancel()
self.wait_for_stop_task = None self.wait_for_stop_task = None
@ -153,9 +152,8 @@ class ReflectorServerProtocol(asyncio.Protocol):
class ReflectorServer: class ReflectorServer:
def __init__(self, blob_manager: 'BlobManager', response_chunk_size: int = 10000, def __init__(self, blob_manager: 'BlobManager', response_chunk_size: int = 10000,
stop_event: typing.Optional[asyncio.Event] = None, stop_event: asyncio.Event = None, incoming_event: asyncio.Event = None,
incoming_event: typing.Optional[asyncio.Event] = None, not_incoming_event: asyncio.Event = None):
not_incoming_event: typing.Optional[asyncio.Event] = None):
self.loop = asyncio.get_event_loop() self.loop = asyncio.get_event_loop()
self.blob_manager = blob_manager self.blob_manager = blob_manager
self.server_task: typing.Optional[asyncio.Task] = None self.server_task: typing.Optional[asyncio.Task] = None