prevent a subscriber from blocking all others

bluenviron · Aug 18, 2023 · 401c29a · 401c29a
1 parent c3c2995
commit 401c29a
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/publisher_sub.go b/publisher_sub.go
@@ -9,6 +9,10 @@ import (
 	"github.com/bluenviron/goroslib/v2/pkg/prototcp"
 )
 
+const (
+	writeBufferSize = 16
+)
+
 type publisherSubscriber struct {
 	pub      *Publisher
 	callerID string
@@ -40,7 +44,7 @@ func newPublisherSubscriber(
 		udpAddr:        udpAddr,
 		ctx:            ctx,
 		ctxCancel:      ctxCancel,
-		chWriteMessage: make(chan interface{}),
+		chWriteMessage: make(chan interface{}, writeBufferSize),
 	}
 
 	pub.subscribersWg.Add(1)
@@ -197,5 +201,6 @@ func (ps *publisherSubscriber) writeMessage(msg interface{}) {
 	select {
 	case ps.chWriteMessage <- msg:
 	case <-ps.ctx.Done():
+	default: // buffer is full
 	}
 }