Extract a common _reset() method.

This commit is contained in:
Yuxin Wang 2018-10-10 20:33:32 -04:00
parent 81c58c540b
commit fbd2b1f699

View file

@ -24,14 +24,17 @@ class Tracker(MessageServer):
def peers(self):
return tuple(self._peers.values())
def _reset(self):
self._peers = {}
self._file_list = {}
self._chunkinfo = {}
async def stop(self):
await super().stop()
if len(self._peers) != 0:
logger.warning('Peers dict not fully cleared {}'.format(self._peers))
self._peers = {}
self._file_list = {}
self._chunkinfo = {}
self._reset()
async def _process_connection(self, reader, writer):
assert isinstance(reader, asyncio.StreamReader) and isinstance(writer, asyncio.StreamWriter)