peer: allow msgConsumers to also exit on peer quit send
This commit is contained in:
parent
33c5c9661e
commit
5130949604
4
peer.go
4
peer.go
@ -753,6 +753,8 @@ func (ms *msgStream) msgConsumer() {
|
||||
// Otherwise, we'll check the message queue for any new
|
||||
// items.
|
||||
select {
|
||||
case <-ms.peer.quit:
|
||||
return
|
||||
case <-ms.quit:
|
||||
ms.msgCond.L.Unlock()
|
||||
return
|
||||
@ -777,6 +779,8 @@ func (ms *msgStream) msgConsumer() {
|
||||
// grow indefinitely.
|
||||
select {
|
||||
case ms.producerSema <- struct{}{}:
|
||||
case <-ms.peer.quit:
|
||||
return
|
||||
case <-ms.quit:
|
||||
return
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user