mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 21:07:18 +00:00
5aac06f04e
* begin move * use same import path * imports * regen protos * regen * no rename * generate ssz * gaz * fmt * edit build file * imports * modify * remove generated files * remove protos * edit imports in prysm * beacon chain all builds * edit script * add generated pbs * add replace rules * license for ethereumapis protos * change visibility * fmt * update build files to gaz ignore * use proper form * edit imports * wrap block * revert scripts * revert go mod
116 lines
3.5 KiB
Go
116 lines
3.5 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"
|
|
pb "github.com/prysmaticlabs/prysm/proto/beacon/p2p/v1"
|
|
pbrpc "github.com/prysmaticlabs/prysm/proto/beacon/rpc/v1"
|
|
ethpb "github.com/prysmaticlabs/prysm/proto/eth/v1alpha1"
|
|
"github.com/prysmaticlabs/prysm/shared/interfaces"
|
|
)
|
|
|
|
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 interfaces.Metadata
|
|
ChainState *pb.Status
|
|
ChainStateLastUpdated time.Time
|
|
ChainStateValidationError error
|
|
// Scorers internal data.
|
|
BadResponses int
|
|
ProcessedBlocks uint64
|
|
BlockProviderUpdated time.Time
|
|
// Gossip Scoring data.
|
|
TopicScores map[string]*pbrpc.TopicScoreSnapshot
|
|
GossipScore float64
|
|
BehaviourPenalty float64
|
|
}
|
|
|
|
// 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
|
|
}
|