mirror/erigon-pulse
1
0
Fork 0
mirror of https://gitlab.com/pulsechaincom/erigon-pulse.git synced 2024-12-25 13:07:17 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Close peerInfo tasks channel (#4429)

Browse Source 
Co-authored-by: Alexey Sharp <alexeysharp@Alexeys-iMac.local>
This commit is contained in:
ledgerwatch 2022-06-10 11:59:31 +01:00 committed by GitHub
parent 1e3c90ba80
commit 43f5f0ae27
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
cmd/sentry/sentry/sentry_grpc_server.go
View File

@ -78,6 +78,14 @@ func NewPeerInfo(peer *p2p.Peer, rw p2p.MsgReadWriter) *PeerInfo {
return p
}
func (pi *PeerInfo) Close() {
pi.lock.Lock()
defer pi.lock.Unlock()
if pi.tasks != nil {
close(pi.tasks)
}
}
func (pi *PeerInfo) ID() [64]byte {
return pi.peer.Pubkey()
}
@ -494,6 +502,7 @@ func NewGrpcServer(ctx context.Context, dialCandidates enode.Iterator, readNodeI
log.Trace(fmt.Sprintf("[%s] Start with peer", peerID))
peerInfo := NewPeerInfo(peer, rw)
defer peerInfo.Close()
defer ss.GoodPeers.Delete(peerID)
err := handShake(ctx, ss.GetStatus(), peerID, rw, protocol, protocol, func(bestHash common.Hash) error {