Buffer ack receive chan (#119)

at-wat · May 30, 2020 · 88550ac · 88550ac
1 parent 218a69c
commit 88550ac
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/publish.go b/publish.go
@@ -141,7 +141,7 @@ func (c *BaseClient) Publish(ctx context.Context, message *Message) error {
 		chPubAck = make(chan *pktPubAck, 1)
 		c.sig.mu.Lock()
 		if c.sig.chPubAck == nil {
-			c.sig.chPubAck = make(map[uint16]chan *pktPubAck)
+			c.sig.chPubAck = make(map[uint16]chan *pktPubAck, 1)
 		}
 		c.sig.chPubAck[message.ID] = chPubAck
 		c.sig.mu.Unlock()
@@ -150,11 +150,11 @@ func (c *BaseClient) Publish(ctx context.Context, message *Message) error {
 		chPubComp = make(chan *pktPubComp, 1)
 		c.sig.mu.Lock()
 		if c.sig.chPubRec == nil {
-			c.sig.chPubRec = make(map[uint16]chan *pktPubRec)
+			c.sig.chPubRec = make(map[uint16]chan *pktPubRec, 1)
 		}
 		c.sig.chPubRec[message.ID] = chPubRec
 		if c.sig.chPubComp == nil {
-			c.sig.chPubComp = make(map[uint16]chan *pktPubComp)
+			c.sig.chPubComp = make(map[uint16]chan *pktPubComp, 1)
 		}
 		c.sig.chPubComp[message.ID] = chPubComp
 		c.sig.mu.Unlock()

diff --git a/subscribe.go b/subscribe.go
@@ -70,7 +70,7 @@ func (c *BaseClient) Subscribe(ctx context.Context, subs ...Subscription) error
 	chSubAck := make(chan *pktSubAck, 1)
 	c.sig.mu.Lock()
 	if c.sig.chSubAck == nil {
-		c.sig.chSubAck = make(map[uint16]chan *pktSubAck)
+		c.sig.chSubAck = make(map[uint16]chan *pktSubAck, 1)
 	}
 	c.sig.chSubAck[id] = chSubAck
 	c.sig.mu.Unlock()

diff --git a/unsubscribe.go b/unsubscribe.go
@@ -46,7 +46,7 @@ func (c *BaseClient) Unsubscribe(ctx context.Context, subs ...string) error {
 	chUnsubAck := make(chan *pktUnsubAck, 1)
 	c.sig.mu.Lock()
 	if c.sig.chUnsubAck == nil {
-		c.sig.chUnsubAck = make(map[uint16]chan *pktUnsubAck)
+		c.sig.chUnsubAck = make(map[uint16]chan *pktUnsubAck, 1)
 	}
 	c.sig.chUnsubAck[id] = chUnsubAck
 	c.sig.mu.Unlock()