perf(node, p2p): avoid unnecessary check len of map (#1995)

This commit is contained in:
Daniel Liu 2026-02-05 17:17:48 +08:00 committed by GitHub
parent 9dba15a673
commit 67a6e34188
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 7 additions and 13 deletions

View file

@ -72,10 +72,8 @@ func (api *adminAPI) AddPeer(url string) (bool, error) {
}
}
// reject the node which is in peer blacklist
if len(server.BlackPeers) > 0 {
if _, ok := server.BlackPeers[node.ID]; ok {
return false, fmt.Errorf("peer is in blacklist: %v, ID: %s", url, node.ID)
}
if _, ok := server.BlackPeers[node.ID]; ok {
return false, fmt.Errorf("peer is in blacklist: %v, ID: %s", url, node.ID)
}
server.AddPeer(node)
return true, nil
@ -115,10 +113,8 @@ func (api *adminAPI) AddTrustedPeer(url string) (bool, error) {
}
}
// reject the node which is in peer blacklist
if len(server.BlackPeers) > 0 {
if _, ok := server.BlackPeers[node.ID]; ok {
return false, fmt.Errorf("trusted peer is in blacklist: %v, ID: %s", url, node.ID)
}
if _, ok := server.BlackPeers[node.ID]; ok {
return false, fmt.Errorf("trusted peer is in blacklist: %v, ID: %s", url, node.ID)
}
server.AddTrustedPeer(node)
return true, nil

View file

@ -905,11 +905,9 @@ func (srv *Server) setupConn(c *conn, flags connFlag, dialDest *discover.Node) e
return DiscNonWhitelistedPeer
}
}
if len(srv.BlackPeers) > 0 {
if _, ok := srv.BlackPeers[c.id]; ok {
clog.Debug("Reject blacklisted peer")
return DiscBlacklistedPeer
}
if _, ok := srv.BlackPeers[c.id]; ok {
clog.Debug("Reject blacklisted peer")
return DiscBlacklistedPeer
}
// For dialed connections, check that the remote public key matches.
if dialDest != nil && c.id != dialDest.ID {