Add PubMessge

mqtt.go

@@ -19,22 +19,22 @@ type Message interface {
 	Ack()
 }
 
-type message struct {
+type PubMessage interface {
-	duplicate bool
+	Topic() string
-	qos       byte
+	Payload() []byte
-	retained  bool
+	Qos() byte
-	topic     string
+	Retained() bool
-	messageID uint16
-	payload   []byte
-	ack       func()
 }
 
-func (m *message) Duplicate() bool {
+type message struct {
-	return m.duplicate
+	qos      int
+	retained bool
+	topic    string
+	payload  []byte
 }
 
 func (m *message) Qos() byte {
-	return m.qos
+	return byte(m.qos)
 }
 
 func (m *message) Retained() bool {
@@ -45,19 +45,17 @@ func (m *message) Topic() string {
 	return m.topic
 }
 
-func (m *message) MessageID() uint16 {
-	return m.messageID
-}
-
 func (m *message) Payload() []byte {
 	return m.payload
 }
 
-func (m *message) Ack() {
+func NewPubMessage(topic string, payload []byte, qos int, retained bool) *message {
+	return &message{
+		topic:    topic,
+		payload:  payload,
+		qos:      qos,
+		retained: retained,
 	}
-
-func NewSimpleMessage(topic string, payload []byte, retained bool) *message {
-	return &message{topic: topic, payload: payload, retained: retained}
 }
 
 type Config struct {