Browse Source

Merge pull request #110 from tabone/close-frame-response

Close Frame Response
Gary Burd 9 năm trước cách đây
mục cha
commit
e85d897656
1 tập tin đã thay đổi với 3 bổ sung1 xóa
  1. 3 1
      conn.go

+ 3 - 1
conn.go

@@ -726,13 +726,15 @@ func (c *Conn) advanceFrame() (int, error) {
 			return noFrame, err
 		}
 	case CloseMessage:
-		c.WriteControl(CloseMessage, []byte{}, time.Now().Add(writeWait))
+		echoMessage := []byte{}
 		closeCode := CloseNoStatusReceived
 		closeText := ""
 		if len(payload) >= 2 {
+			echoMessage = payload[:2]
 			closeCode = int(binary.BigEndian.Uint16(payload))
 			closeText = string(payload[2:])
 		}
+		c.WriteControl(CloseMessage,  echoMessage, time.Now().Add(writeWait))
 		return noFrame, &CloseError{Code: closeCode, Text: closeText}
 	}