Add channelInactive callback to mark WebSocket as closed. (#60)

vapor · Feb 26, 2020 · e35af05 · e35af05
1 parent 766b4b0
commit e35af05
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/Sources/WebSocketKit/WebSocketHandler.swift b/Sources/WebSocketKit/WebSocketHandler.swift
@@ -42,4 +42,12 @@ private final class WebSocketHandler: ChannelInboundHandler {
         let frame = self.unwrapInboundIn(data)
         self.webSocket.handle(incoming: frame)
     }
+
+    func channelInactive(context: ChannelHandlerContext) {
+        let closedAbnormally = WebSocketErrorCode.unknown(1005)
+        _ = webSocket.close(code: closedAbnormally)
+
+        // We always forward the error on to let others see it.
+        context.fireChannelInactive()
+    }
 }