mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-31 23:41:22 +00:00
b015dc793a
* HeadFetcher data race fix * bazel run //:gazelle -- fix * add the db teardown to test * add TestChainService_SaveHead_DataRace test * split race and norace tests * change testset name * test CI with 'spectest' tag instead of 'raceon' * one more test CI with 'spectest' tag instead of 'raceon' * bazel run //:gazelle -- fix * set test tag to 'race_on' * not clone the state
105 lines
2.7 KiB
Go
105 lines
2.7 KiB
Go
package blockchain
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/gogo/protobuf/proto"
|
|
pb "github.com/prysmaticlabs/prysm/proto/beacon/p2p/v1"
|
|
ethpb "github.com/prysmaticlabs/prysm/proto/eth/v1alpha1"
|
|
"github.com/prysmaticlabs/prysm/shared/params"
|
|
)
|
|
|
|
// ChainInfoFetcher defines a common interface for methods in blockchain service which
|
|
// directly retrieves chain info related data.
|
|
type ChainInfoFetcher interface {
|
|
HeadFetcher
|
|
CanonicalRootFetcher
|
|
FinalizationFetcher
|
|
}
|
|
|
|
// GenesisTimeFetcher retrieves the Eth2 genesis timestamp.
|
|
type GenesisTimeFetcher interface {
|
|
GenesisTime() time.Time
|
|
}
|
|
|
|
// HeadFetcher defines a common interface for methods in blockchain service which
|
|
// directly retrieves head related data.
|
|
type HeadFetcher interface {
|
|
HeadSlot() uint64
|
|
HeadRoot() []byte
|
|
HeadBlock() *ethpb.BeaconBlock
|
|
HeadState() *pb.BeaconState
|
|
}
|
|
|
|
// CanonicalRootFetcher defines a common interface for methods in blockchain service which
|
|
// directly retrieves canonical roots related data.
|
|
type CanonicalRootFetcher interface {
|
|
CanonicalRoot(slot uint64) []byte
|
|
}
|
|
|
|
// FinalizationFetcher defines a common interface for methods in blockchain service which
|
|
// directly retrieves finalization related data.
|
|
type FinalizationFetcher interface {
|
|
FinalizedCheckpt() *ethpb.Checkpoint
|
|
}
|
|
|
|
// FinalizedCheckpt returns the latest finalized checkpoint tracked in fork choice service.
|
|
func (s *Service) FinalizedCheckpt() *ethpb.Checkpoint {
|
|
cp := s.forkChoiceStore.FinalizedCheckpt()
|
|
if cp != nil {
|
|
return cp
|
|
}
|
|
|
|
return ðpb.Checkpoint{Root: params.BeaconConfig().ZeroHash[:]}
|
|
}
|
|
|
|
// HeadSlot returns the slot of the head of the chain.
|
|
func (s *Service) HeadSlot() uint64 {
|
|
s.headLock.RLock()
|
|
defer s.headLock.RUnlock()
|
|
|
|
return s.headSlot
|
|
}
|
|
|
|
// HeadRoot returns the root of the head of the chain.
|
|
func (s *Service) HeadRoot() []byte {
|
|
s.headLock.RLock()
|
|
defer s.headLock.RUnlock()
|
|
|
|
root := s.canonicalRoots[s.headSlot]
|
|
if len(root) != 0 {
|
|
return root
|
|
}
|
|
|
|
return params.BeaconConfig().ZeroHash[:]
|
|
}
|
|
|
|
// HeadBlock returns the head block of the chain.
|
|
func (s *Service) HeadBlock() *ethpb.BeaconBlock {
|
|
s.headLock.RLock()
|
|
defer s.headLock.RUnlock()
|
|
|
|
return proto.Clone(s.headBlock).(*ethpb.BeaconBlock)
|
|
}
|
|
|
|
// HeadState returns the head state of the chain.
|
|
func (s *Service) HeadState() *pb.BeaconState {
|
|
s.headLock.RLock()
|
|
defer s.headLock.RUnlock()
|
|
|
|
return proto.Clone(s.headState).(*pb.BeaconState)
|
|
}
|
|
|
|
// CanonicalRoot returns the canonical root of a given slot.
|
|
func (s *Service) CanonicalRoot(slot uint64) []byte {
|
|
s.headLock.RLock()
|
|
defer s.headLock.RUnlock()
|
|
|
|
return s.canonicalRoots[slot]
|
|
}
|
|
|
|
// GenesisTime returns the genesis time of beacon chain.
|
|
func (s *Service) GenesisTime() time.Time {
|
|
return s.genesisTime
|
|
}
|