Update peer.go

This commit is contained in:
jagdeep sidhu 2024-10-07 20:51:29 -07:00
parent dceecbd541
commit 0057742b1e

View file

@ -37,10 +37,10 @@ const (
) )
var ( var (
errAlreadyRegistered = errors.New("peer is already registered")
errNotRegistered = errors.New("peer is not registered")
// SYSCOIN // SYSCOIN
errPeerSetClosed = errors.New("peerset closed") errPeerSetClosed = errors.New("peerset closed")
errAlreadyRegistered = errors.New("peer is already registered")
errNotRegistered = errors.New("peer is not registered")
) )
// peerConnection represents an active peer from which hashes and blocks are retrieved. // peerConnection represents an active peer from which hashes and blocks are retrieved.
@ -212,6 +212,11 @@ func (ps *peerSet) Reset() {
func (ps *peerSet) Register(p *peerConnection) error { func (ps *peerSet) Register(p *peerConnection) error {
// Register the new peer with some meaningful defaults // Register the new peer with some meaningful defaults
ps.lock.Lock() ps.lock.Lock()
// SYSCOIN
if ps.closed {
ps.lock.Unlock()
return errPeerSetClosed
}
if _, ok := ps.peers[p.id]; ok { if _, ok := ps.peers[p.id]; ok {
ps.lock.Unlock() ps.lock.Unlock()
return errAlreadyRegistered return errAlreadyRegistered
@ -227,6 +232,7 @@ func (ps *peerSet) Register(p *peerConnection) error {
ps.events.Send(&peeringEvent{peer: p, join: true}) ps.events.Send(&peeringEvent{peer: p, join: true})
return nil return nil
} }
// SYSCOIN // SYSCOIN
func (ps *peerSet) Close() { func (ps *peerSet) Close() {
ps.lock.Lock() ps.lock.Lock()
@ -240,6 +246,7 @@ func (ps *peerSet) Open() {
ps.closed = false ps.closed = false
} }
// Unregister removes a remote peer from the active set, disabling any further // Unregister removes a remote peer from the active set, disabling any further
// actions to/from that particular entity. // actions to/from that particular entity.
func (ps *peerSet) Unregister(id string) error { func (ps *peerSet) Unregister(id string) error {