handle more cases

dunglas · Nov 14, 2021 · d4fc309 · d4fc309
1 parent 21cd1d2
commit d4fc309
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/subscriber.go b/subscriber.go
@@ -68,8 +68,8 @@ func (s *Subscriber) Dispatch(u *Update, fromHistory bool) bool {
 		s.liveMutex.Unlock()
 	}
 
-	s.outMutex.RLock()
-	defer s.outMutex.RUnlock()
+	s.outMutex.Lock()
+	defer s.outMutex.Unlock()
 	if atomic.LoadInt32(&s.disconnected) > 0 {
 		return false
 	}
@@ -83,6 +83,8 @@ func (s *Subscriber) Dispatch(u *Update, fromHistory bool) bool {
 func (s *Subscriber) Ready() int {
 	s.liveMutex.Lock()
 	defer s.liveMutex.Unlock()
+	s.outMutex.Lock()
+	defer s.outMutex.Unlock()
 
 	n := len(s.liveQueue)
 	for _, u := range s.liveQueue {