From 6c7fe6a0f76479e8e143b17f56cf69c9fd21f57e Mon Sep 17 00:00:00 2001
From: Marc Magnin <marc.magnin@gmail.com>
Date: Mon, 7 Jan 2019 12:56:00 +0100
Subject: [PATCH] simple fix (#35)

---
 broker/broker.go | 2 ++
 broker/client.go | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/broker/broker.go b/broker/broker.go
index de1aa73..53518ff 100644
--- a/broker/broker.go
+++ b/broker/broker.go
@@ -617,7 +617,9 @@ func (b *Broker) removeClient(c *client) {
 func (b *Broker) PublishMessage(packet *packets.PublishPacket) {
 	var subs []interface{}
 	var qoss []byte
+	b.mu.Lock()
 	err := b.topicsMgr.Subscribers([]byte(packet.TopicName), packet.Qos, &subs, &qoss)
+	b.mu.Unlock()
 	if err != nil {
 		log.Error("search sub client error,  ", zap.Error(err))
 		return
diff --git a/broker/client.go b/broker/client.go
index 27ab9be..65b7ffa 100644
--- a/broker/client.go
+++ b/broker/client.go
@@ -213,7 +213,9 @@ func (c *client) ProcessPublishMessage(packet *packets.PublishPacket) {
 		}
 	}
 
+	c.mu.Lock()
 	err := c.topicsMgr.Subscribers([]byte(packet.TopicName), packet.Qos, &c.subs, &c.qoss)
+	c.mu.Unlock()
 	if err != nil {
 		log.Error("Error retrieving subscribers list: ", zap.String("ClientID", c.info.clientID))
 		return