fix: closing websocket connection returns invalid status code

AlexStocks · Nov 18, 2023 · 31cf517 · 31cf517
1 parent 929d524
commit 31cf517
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/transport/connection.go b/transport/connection.go
@@ -639,7 +639,8 @@ func (w *gettyWSConn) writePong(message []byte) error {
 // close websocket connection
 func (w *gettyWSConn) close(waitSec int) {
 	w.updateWriteDeadline()
-	w.conn.WriteMessage(websocket.CloseMessage, []byte("bye-bye!!!"))
+	bytes := websocket.FormatCloseMessage(websocket.CloseNormalClosure, "bye-bye!!!")
+	w.conn.WriteMessage(websocket.CloseMessage, bytes)
 	conn := w.conn.UnderlyingConn()
 	if tcpConn, ok := conn.(*net.TCPConn); ok {
 		tcpConn.SetLinger(waitSec)