瀏覽代碼

fix: connection.go

lijian 2 年之前
父節點
當前提交
656d5c2e31
共有 2 個文件被更改,包括 3 次插入1 次删除
  1. 2 0
      pkg/mqtt/connection.go
  2. 1 1
      pkg/mqtt/message.go

+ 2 - 0
pkg/mqtt/connection.go

@@ -315,6 +315,8 @@ func (c *Connection) RcvMsgFromClient() {
 }
 
 func (c *Connection) SendMsgToClient() {
+	c.mLock.Lock()
+	c.mLock.Unlock()
 	if c.Conn == nil {
 		return
 	}

+ 1 - 1
pkg/mqtt/message.go

@@ -114,7 +114,7 @@ func (hdr *Header) EncodeInto(buf *bytes.Buffer, msgType TagMessageType, remaini
 	}
 
 	val := byte(msgType) << 4
-	val |= (boolToByte(hdr.DupFlag) << 3)
+	val |= boolToByte(hdr.DupFlag) << 3
 	val |= byte(hdr.QosLevel) << 1
 	val |= boolToByte(hdr.Retain)
 	buf.WriteByte(val)