Improve dht shutdown. Notice writing pauses.

magneticod/magneticod/bittorrent.py

@@ -52,6 +52,7 @@ class DisposablePeer:
 
         self._metadata_future = loop.create_future()
         self._writer = None
+        self._is_paused = False
 
         try:
             self._reader, self._writer = await asyncio.open_connection(
@@ -88,6 +89,12 @@ class DisposablePeer:
             self._writer.close()
         return self._metadata_future.result()
 
+    def pause_writing(self):
+        self._is_paused = True
+
+    def resume_writing(self):
+        self._is_paused = False
+
     def __on_message(self, message: bytes) -> None:
         length = len(message)
 

magneticod/magneticod/dht.py

@@ -84,6 +84,9 @@ class SybilNode:
         if addr[1] == 0:
             return
 
+        if self._transport.is_closing():
+            return
+
         try:
             message = bencode.loads(data)
         except bencode.BencodeDecodingError: