prysm-pulse/beacon-chain/db/kv/state_summary_cache.go
Raul Jordan 9145310647
Eliminate Proto V2 Namespace (#9297)
* get rid of v2 in prysm codebase

* replace block2

* builds

* terence comments

* gazelle
2021-07-28 21:23:44 +00:00

75 lines
2.3 KiB
Go

package kv
import (
"sync"
statepb "github.com/prysmaticlabs/prysm/proto/prysm/v1alpha1"
)
const stateSummaryCachePruneCount = 128
// stateSummaryCache caches state summary object.
type stateSummaryCache struct {
initSyncStateSummaries map[[32]byte]*statepb.StateSummary
initSyncStateSummariesLock sync.RWMutex
}
// newStateSummaryCache creates a new state summary cache.
func newStateSummaryCache() *stateSummaryCache {
return &stateSummaryCache{
initSyncStateSummaries: make(map[[32]byte]*statepb.StateSummary),
}
}
// put saves a state summary to the initial sync state summaries cache.
func (c *stateSummaryCache) put(r [32]byte, b *statepb.StateSummary) {
c.initSyncStateSummariesLock.Lock()
defer c.initSyncStateSummariesLock.Unlock()
c.initSyncStateSummaries[r] = b
}
// has checks if a state summary exists in the initial sync state summaries cache using the root
// of the block.
func (c *stateSummaryCache) has(r [32]byte) bool {
c.initSyncStateSummariesLock.RLock()
defer c.initSyncStateSummariesLock.RUnlock()
_, ok := c.initSyncStateSummaries[r]
return ok
}
// get retrieves a state summary from the initial sync state summaries cache using the root of
// the block.
func (c *stateSummaryCache) get(r [32]byte) *statepb.StateSummary {
c.initSyncStateSummariesLock.RLock()
defer c.initSyncStateSummariesLock.RUnlock()
b := c.initSyncStateSummaries[r]
return b
}
// len retrieves the state summary count from the state summaries cache.
func (c *stateSummaryCache) len() int {
c.initSyncStateSummariesLock.RLock()
defer c.initSyncStateSummariesLock.RUnlock()
return len(c.initSyncStateSummaries)
}
// GetAll retrieves all the beacon state summaries from the initial sync state summaries cache, the returned
// state summaries are unordered.
func (c *stateSummaryCache) getAll() []*statepb.StateSummary {
c.initSyncStateSummariesLock.RLock()
defer c.initSyncStateSummariesLock.RUnlock()
summaries := make([]*statepb.StateSummary, 0, len(c.initSyncStateSummaries))
for _, b := range c.initSyncStateSummaries {
summaries = append(summaries, b)
}
return summaries
}
// Clear clears out the initial sync state summaries cache.
func (c *stateSummaryCache) clear() {
c.initSyncStateSummariesLock.Lock()
defer c.initSyncStateSummariesLock.Unlock()
c.initSyncStateSummaries = make(map[[32]byte]*statepb.StateSummary)
}