prysm-pulse/beacon-chain/operations/attestations/kv/kv.go

package kv

import (
	"time"

	"github.com/patrickmn/go-cache"
	"github.com/prysmaticlabs/go-bitfield"
	"github.com/prysmaticlabs/prysm/shared/params"
)

// 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 {
	aggregatedAtt   *cache.Cache
	unAggregatedAtt *cache.Cache
	attInBlock      *cache.Cache
}

// NewAttCaches initializes a new attestation pool consists of multiple KV store in cache for
// various kind of attestations.
func NewAttCaches() *AttCaches {
	secsInEpoch := time.Duration(params.BeaconConfig().SlotsPerEpoch * params.BeaconConfig().SecondsPerSlot)

	// Create caches with default expiration time of one epoch and which
	// purges expired items every other epoch.
	pool := &AttCaches{
		unAggregatedAtt: cache.New(secsInEpoch*time.Second, 2*secsInEpoch*time.Second),
		aggregatedAtt:   cache.New(secsInEpoch*time.Second, 2*secsInEpoch*time.Second),
		attInBlock:      cache.New(secsInEpoch*time.Second, 2*secsInEpoch*time.Second),
	}

	return pool
}

func aggregated(bits bitfield.Bitlist) bool {
	if bits.Count() > 1 {
		return true
	}
	return false
}