Ver código fonte

fix: connection.go

lijian 2 anos atrás
pai
commit
656d5c2e31
2 arquivos alterados com 3 adições e 1 exclusões
  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() {
 func (c *Connection) SendMsgToClient() {
+	c.mLock.Lock()
+	c.mLock.Unlock()
 	if c.Conn == nil {
 	if c.Conn == nil {
 		return
 		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 := byte(msgType) << 4
-	val |= (boolToByte(hdr.DupFlag) << 3)
+	val |= boolToByte(hdr.DupFlag) << 3
 	val |= byte(hdr.QosLevel) << 1
 	val |= byte(hdr.QosLevel) << 1
 	val |= boolToByte(hdr.Retain)
 	val |= boolToByte(hdr.Retain)
 	buf.WriteByte(val)
 	buf.WriteByte(val)