websocket: fix callback for nil close message

lesismal · Jul 16, 2024 · ebb0bdb · ebb0bdb
1 parent 882d08e
commit ebb0bdb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nbhttp/websocket/conn.go b/nbhttp/websocket/conn.go
@@ -482,7 +482,7 @@ func (c *Conn) Parse(data []byte) error {
 			c.handleDataFrame(msgType, fin, frame)
 			frame = nil
 		}
-		if protocolMessage != nil {
+		if fin && opcode == CloseMessage { //protocolMessage != nil {
 			c.handleProtocolMessage(opcode, protocolMessage)
 			protocolMessage = nil
 		}