akovalenko/lnd.xprv
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

peer: allow msgConsumers to also exit on peer quit send

Browse Source
This commit is contained in:
Olaoluwa Osuntokun 2018-08-24 17:58:24 -07:00
parent 33c5c9661e
commit 5130949604
No known key found for this signature in database
GPG Key ID: 964EA263DD637C21
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
peer.go
View File

@ -753,6 +753,8 @@ func (ms *msgStream) msgConsumer() {
// Otherwise, we'll check the message queue for any new // Otherwise, we'll check the message queue for any new
// items. // items.
select { select {
case <-ms.peer.quit:
return
case <-ms.quit: case <-ms.quit:
ms.msgCond.L.Unlock() ms.msgCond.L.Unlock()
return return
@ -777,6 +779,8 @@ func (ms *msgStream) msgConsumer() {
// grow indefinitely. // grow indefinitely.
select { select {
case ms.producerSema <- struct{}{}: case ms.producerSema <- struct{}{}:
case <-ms.peer.quit:
return
case <-ms.quit: case <-ms.quit:
return return
} }