server: close TCP connection after closing websocket

Fixes #115 and #147.
python-trio · Feb 2, 2021 · cd2d2e8 · cd2d2e8
1 parent dbf819d
commit cd2d2e8
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/trio_websocket/_impl.py b/trio_websocket/_impl.py
@@ -1018,6 +1018,12 @@ async def _close_web_socket(self, code, reason=None):
         exc = ConnectionClosed(self._close_reason)
         logger.debug('%s websocket closed %r', self, exc)
         await self._send_channel.aclose()
+        # RFC: "When a server is instructed to Close the WebSocket Connection
+        #   it SHOULD initiate a TCP Close immediately, and when a client is
+        #   instructed to do the same, it SHOULD wait for a TCP Close from the
+        #   server."
+        if self.is_server:
+            await self._close_stream()
 
     async def _get_request(self):
         '''