Add support for QoS-2

Signed-off-by: Chaitanya Munukutla <[email protected]>

README.md

@@ -5,7 +5,7 @@ A tiny Websocket-based event broker.
 - [x] Grouped subscriptions
 - [ ] Offline messages
 - [ ] Clustering
-- [x] QoS (0, 1)
+- [x] QoS (0, 1, 2)
 - [x] Topics & Patterns
 
 ### Subscriptions

events/pubcomp.go

@@ -0,0 +1,8 @@
+package events
+
+const PubComp = "pubcomp"
+
+type PubCompEvent struct {
+	Kind     string `json:"kind"`
+	PacketId string `json:"packet_id"`
+}

events/pubrec.go

@@ -0,0 +1,8 @@
+package events
+
+const PubRec = "pubrec"
+
+type PubRecEvent struct {
+	Kind     string `json:"kind"`
+	PacketId string `json:"packet_id"`
+}

events/pubrel.go

@@ -0,0 +1,8 @@
+package events
+
+const PubRel = "pubrel"
+
+type PubRelEvent struct {
+	Kind     string `json:"kind"`
+	PacketId string `json:"packet_id"`
+}

handlers/base.go

@@ -10,6 +10,8 @@ func HandleMessage(client *broker.ConnectedClient, broker *broker.Broker, messag
 	switch kind {
 	case events.Pub:
 		return handlePublish(message, client, broker)
+	case events.PubRel:
+		return handlePubrel(message, client)
 	case events.Sub:
 		return handleSubscribe(message, client)
 	case events.Unsub:

handlers/pub.go

@@ -20,6 +20,13 @@ func handlePublish(message []byte, client *broker.ConnectedClient, broker *broke
 		}
 		client.WriteInterface(pubAckEvent)
 	}
+	if event.QoS == 2 {
+		pubRecEvent := &events.PubRecEvent{
+			Kind:     events.PubRec,
+			PacketId: uuid.NewString(),
+		}
+		client.WriteInterface(pubRecEvent)
+	}
 	go broker.Broadcast(event)
 	return nil
 }

handlers/pubrel.go

@@ -0,0 +1,21 @@
+package handlers
+
+import (
+	"encoding/json"
+	"github.com/c16a/microq/broker"
+	"github.com/c16a/microq/events"
+)
+
+func handlePubrel(message []byte, client *broker.ConnectedClient) error {
+	var event events.PubRelEvent
+	err := json.Unmarshal(message, &event)
+	if err != nil {
+		return err
+	}
+	pubCompEvent := &events.PubCompEvent{
+		Kind:     events.PubComp,
+		PacketId: event.PacketId,
+	}
+	client.WriteInterface(pubCompEvent)
+	return nil
+}