mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 21:07:18 +00:00
095c4d5dd5
* define and enforce minimum scorer interface * better decoupling of multiple scorers in service * removes redundant weight * adds peer_status scorer * minir re-arrangement * rely on scorer in peer status service * gazelle * updates rpc_status * fix build * better interface verifying * remove unnecessary locks * mark todo * simplify service * remove redundant references * avoid passing contexts * remove unused context * refactor errors to p2p package * refactor goodbye codes into p2p * simplify status api * remove isbad method from peers * update scoring service * introduce validation error * gazelle * add score * restore isbad method * resolve dep cycle * gazelle * peer status scorer: test score calculation * bad responses scorer: bad peer score * remove redundant type checks * pass nil config * add rounding * test IsBadPeer * test bad peers list * more tests * check validation error on non-existent peer * max peer slot -> highest peer slot * remove redundant comment * combine * combine * introduce var * fix tests * remove redundant update * minor fix * Nishant's suggestion Co-authored-by: prylabs-bulldozer[bot] <58059840+prylabs-bulldozer[bot]@users.noreply.github.com>
110 lines
3.3 KiB
Go
110 lines
3.3 KiB
Go
package peerdata
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/ethereum/go-ethereum/p2p/enr"
|
|
"github.com/libp2p/go-libp2p-core/network"
|
|
"github.com/libp2p/go-libp2p-core/peer"
|
|
ma "github.com/multiformats/go-multiaddr"
|
|
ethpb "github.com/prysmaticlabs/ethereumapis/eth/v1"
|
|
pb "github.com/prysmaticlabs/prysm/proto/beacon/p2p/v1"
|
|
)
|
|
|
|
var (
|
|
// ErrPeerUnknown is returned when there is an attempt to obtain data from a peer that is not known.
|
|
ErrPeerUnknown = errors.New("peer unknown")
|
|
)
|
|
|
|
// PeerConnectionState is the state of the connection.
|
|
type PeerConnectionState ethpb.ConnectionState
|
|
|
|
// StoreConfig holds peer store parameters.
|
|
type StoreConfig struct {
|
|
MaxPeers int
|
|
}
|
|
|
|
// Store is a container for various peer related data (both protocol and app level).
|
|
// Container implements RWMutex, so data access can be restricted on the container level. This allows
|
|
// different components rely on the very same peer map container.
|
|
// Note: access to data is controlled by clients i.e. client code is responsible for locking/unlocking
|
|
// the mutex when accessing data.
|
|
type Store struct {
|
|
sync.RWMutex
|
|
ctx context.Context
|
|
config *StoreConfig
|
|
peers map[peer.ID]*PeerData
|
|
}
|
|
|
|
// PeerData aggregates protocol and application level info about a single peer.
|
|
type PeerData struct {
|
|
// Network related data.
|
|
Address ma.Multiaddr
|
|
Direction network.Direction
|
|
ConnState PeerConnectionState
|
|
Enr *enr.Record
|
|
NextValidTime time.Time
|
|
// Chain related data.
|
|
MetaData *pb.MetaData
|
|
ChainState *pb.Status
|
|
ChainStateLastUpdated time.Time
|
|
ChainStateValidationError error
|
|
// Scorers internal data.
|
|
BadResponses int
|
|
ProcessedBlocks uint64
|
|
BlockProviderUpdated time.Time
|
|
}
|
|
|
|
// NewStore creates new peer data store.
|
|
func NewStore(ctx context.Context, config *StoreConfig) *Store {
|
|
return &Store{
|
|
ctx: ctx,
|
|
config: config,
|
|
peers: make(map[peer.ID]*PeerData),
|
|
}
|
|
}
|
|
|
|
// PeerData returns data associated with a given peer, if any.
|
|
// Important: it is assumed that store mutex is locked when calling this method.
|
|
func (s *Store) PeerData(pid peer.ID) (*PeerData, bool) {
|
|
peerData, ok := s.peers[pid]
|
|
return peerData, ok
|
|
}
|
|
|
|
// PeerDataGetOrCreate returns data associated with a given peer.
|
|
// If no data has been associated yet, newly created and associated data object is returned.
|
|
// Important: it is assumed that store mutex is locked when calling this method.
|
|
func (s *Store) PeerDataGetOrCreate(pid peer.ID) *PeerData {
|
|
if peerData, ok := s.peers[pid]; ok {
|
|
return peerData
|
|
}
|
|
s.peers[pid] = &PeerData{}
|
|
return s.peers[pid]
|
|
}
|
|
|
|
// SetPeerData updates data associated with a given peer.
|
|
// Important: it is assumed that store mutex is locked when calling this method.
|
|
func (s *Store) SetPeerData(pid peer.ID, data *PeerData) {
|
|
s.peers[pid] = data
|
|
}
|
|
|
|
// DeletePeerData removes data associated with a given peer.
|
|
// Important: it is assumed that store mutex is locked when calling this method.
|
|
func (s *Store) DeletePeerData(pid peer.ID) {
|
|
delete(s.peers, pid)
|
|
}
|
|
|
|
// Peers returns map of peer data objects.
|
|
// Important: it is assumed that store mutex is locked when calling this method.
|
|
func (s *Store) Peers() map[peer.ID]*PeerData {
|
|
return s.peers
|
|
}
|
|
|
|
// Config exposes store configuration params.
|
|
func (s *Store) Config() *StoreConfig {
|
|
return s.config
|
|
}
|