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

force close open streaming requests and api calls on shutdown

Browse source
This commit is contained in:
Jack Robison 2019-05-02 16:56:29 -04:00
parent cbe689ea7f
commit 1f7feafb67
No known key found for this signature in database
GPG key ID: DF25C68FE0239BB2
2 changed files with 9 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
lbrynet/extras/daemon/Daemon.py
View file

@ -439,6 +439,7 @@ class Daemon(metaclass=JSONRPCServerType):
await self.component_manager.stop()
else:
self.component_startup_task.cancel()
await self.runner.shutdown()
await self.runner.cleanup()
if self.analytics_manager.is_started:
self.analytics_manager.stop()

13
lbrynet/stream/managed_stream.py
View file

@ -94,7 +94,7 @@ class ManagedStream:
self.fully_reflected = asyncio.Event(loop=self.loop)
self.file_output_task: typing.Optional[asyncio.Task] = None
self.delayed_stop_task: typing.Optional[asyncio.Task] = None
self.streaming_responses: typing.List[StreamResponse] = []
self.streaming_responses: typing.List[typing.Tuple[Request, StreamResponse]] = []
self.streaming = asyncio.Event(loop=self.loop)
self._running = asyncio.Event(loop=self.loop)
self.saving = asyncio.Event(loop=self.loop)
@ -311,7 +311,7 @@ class ManagedStream:
headers=headers
)
await response.prepare(request)
self.streaming_responses.append(response)
self.streaming_responses.append((request, response))
self.streaming.set()
try:
wrote = 0
@ -329,8 +329,9 @@ class ManagedStream:
return response
finally:
response.force_close()
if response in self.streaming_responses:
self.streaming_responses.remove(response)
if (request, response) in self.streaming_responses:
self.streaming_responses.remove((request, response))
if not self.streaming_responses:
self.streaming.clear()
async def _save_file(self, output_path: str):
@ -394,7 +395,9 @@ class ManagedStream:
self.file_output_task.cancel()
self.file_output_task = None
while self.streaming_responses:
self.streaming_responses.pop().force_close()
req, response = self.streaming_responses.pop()
response.force_close()
req.transport.close()
self.downloader.stop()
self._running.clear()