mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-26 13:18:57 +00:00
3df2980cba
* DB: add block roots test * Pool: add seen atts map * Pool: use seen atts map * Pool: clear seen map * Gazelle * Fixed an existing test Co-authored-by: rauljordan <raul@prysmaticlabs.com> Co-authored-by: prylabs-bulldozer[bot] <58059840+prylabs-bulldozer[bot]@users.noreply.github.com>
45 lines
1.5 KiB
Go
45 lines
1.5 KiB
Go
// Package kv includes a key-value store implementation
|
|
// of an attestation cache used to satisfy important use-cases
|
|
// such as aggregation in a beacon node runtime.
|
|
package kv
|
|
|
|
import (
|
|
"sync"
|
|
|
|
ethpb "github.com/prysmaticlabs/ethereumapis/eth/v1alpha1"
|
|
"github.com/prysmaticlabs/go-bitfield"
|
|
"github.com/prysmaticlabs/prysm/shared/hashutil"
|
|
)
|
|
|
|
var hashFn = hashutil.HashProto
|
|
|
|
// AttCaches defines the caches used to satisfy attestation pool interface.
|
|
// These caches are KV store for various attestations
|
|
// such are unaggregated, aggregated or attestations within a block.
|
|
type AttCaches struct {
|
|
aggregatedAttLock sync.RWMutex
|
|
aggregatedAtt map[[32]byte][]*ethpb.Attestation
|
|
unAggregateAttLock sync.RWMutex
|
|
unAggregatedAtt map[[32]byte]*ethpb.Attestation
|
|
forkchoiceAttLock sync.RWMutex
|
|
forkchoiceAtt map[[32]byte]*ethpb.Attestation
|
|
blockAttLock sync.RWMutex
|
|
blockAtt map[[32]byte][]*ethpb.Attestation
|
|
seenAggregatedAttLock sync.RWMutex
|
|
seenAggregatedAtt map[[32]byte][]bitfield.Bitlist
|
|
}
|
|
|
|
// NewAttCaches initializes a new attestation pool consists of multiple KV store in cache for
|
|
// various kind of attestations.
|
|
func NewAttCaches() *AttCaches {
|
|
pool := &AttCaches{
|
|
unAggregatedAtt: make(map[[32]byte]*ethpb.Attestation),
|
|
aggregatedAtt: make(map[[32]byte][]*ethpb.Attestation),
|
|
forkchoiceAtt: make(map[[32]byte]*ethpb.Attestation),
|
|
blockAtt: make(map[[32]byte][]*ethpb.Attestation),
|
|
seenAggregatedAtt: make(map[[32]byte][]bitfield.Bitlist),
|
|
}
|
|
|
|
return pool
|
|
}
|