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

discovery: properly handle SyncManager shutdown signal

Browse Source
This commit is contained in:
Wilmer Paulino 2019-04-03 19:11:03 -07:00
parent 46ceaf8cf6
commit 00338c5ec2
No known key found for this signature in database
GPG Key ID: 6DF57B9F9514972F
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
discovery/sync_manager.go
View File

@ -286,6 +286,13 @@ func (m *SyncManager) roundRobinHandler() {
current = m.nextPendingActiveSyncer() current = m.nextPendingActiveSyncer()
m.Unlock() m.Unlock()
for current != nil { for current != nil {
// Ensure we properly handle a shutdown signal.
select {
case <-m.quit:
return
default:
}
// We'll avoid performing the transition with the lock // We'll avoid performing the transition with the lock
// as it can potentially stall the SyncManager due to // as it can potentially stall the SyncManager due to
// the syncTransitionTimeout. // the syncTransitionTimeout.
@ -531,6 +538,13 @@ func (m *SyncManager) forceHistoricalSync() {
candidatesChosen := make(map[routing.Vertex]struct{}) candidatesChosen := make(map[routing.Vertex]struct{})
s := m.chooseRandomSyncer(candidatesChosen, true) s := m.chooseRandomSyncer(candidatesChosen, true)
for s != nil { for s != nil {
// Ensure we properly handle a shutdown signal.
select {
case <-m.quit:
return
default:
}
// Blacklist the candidate to ensure it's not chosen again. // Blacklist the candidate to ensure it's not chosen again.
candidatesChosen[s.cfg.peerPub] = struct{}{} candidatesChosen[s.cfg.peerPub] = struct{}{}