Websocket+sockjs transports implementing closed signal

Author: prawnsalad

pkg/webircgateway/transport_sockjs.go

@@ -92,5 +92,9 @@ func (t *TransportSockjs) sessionHandler(session sockjs.Session) {
 			client.Log(1, "->ws: %s", line)
 			session.Send(line)
 		}
+
+		if signal[0] == "state" && signal[1] == "closed" {
+			session.Close(0, "Closed")
+		}
 	}
 }

pkg/webircgateway/transport_websocket.go

@@ -115,6 +115,10 @@ func (t *TransportWebsocket) websocketHandler(ws *websocket.Conn) {
 			client.Log(1, "->ws: %s", line)
 			ws.Write([]byte(line))
 		}
+
+		if signal[0] == "state" && signal[1] == "closed" {
+			ws.Close()
+		}
 	}
 
 	sendDrained.Wait()