Cancel plumbing request task when request is canceled

chrysn · Mar 24, 2020 · 81a9e99 · 81a9e99
1 parent 538afa8
commit 81a9e99
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/aiocoap/protocol.py b/aiocoap/protocol.py
@@ -594,10 +594,16 @@ def __init__(self, plumbing_request, loop, log):
         else:
             self.observation = None
 
-        loop.create_task(self._run())
+        self._runner = loop.create_task(self._run())
 
         self.log = log
 
+        self.response.add_done_callback(self._response_cancellation_handler)
+
+    def _response_cancellation_handler(self, response):
+        if self.response.cancelled() and not self._runner.cancelled():
+            self._runner.cancel()
+
     @staticmethod
     def _add_response_properties(response, request):
         response.request = request