Merge pull request #74 from Jardaliao/fix-close-websocket-invalid-sta…

…tus-code fix: closing websocket connection returns invalid status code
AlexStocks · Nov 20, 2023 · 28c6c73 · 28c6c73
2 parents 929d524 + 31cf517
commit 28c6c73
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)