mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-26 05:17:22 +00:00
7cc32c4dda
* remove unused code * remove defer use in loop * Remove unused methods and constants * gofmt and gaz * nilness check * remove unused args * Add TODO for refactoring subscribeWithBase to remove unused arg. It seems too involved to include in this sweeping PR. https://github.com/prysmaticlabs/prysm/issues/7437 * replace empty slice declaration * Remove unnecessary type conversions * remove redundant type declaration * rename receivers to be consistent * Remove bootnode query tool. It is now obsolete by discv5 * Remove relay node. It is no longer used or supported * Revert "Remove relay node. It is no longer used or supported" This reverts commit 4bd7717334dad85ef4766ed9bc4da711fb5fa810. * Delete unused test directory * Delete unsupported gcp startup script * Delete old k8s script * build fixes * fix build * go mod tidy * revert slasher/db/kv/block_header.go * fix build * remove redundant nil check * combine func args Co-authored-by: prylabs-bulldozer[bot] <58059840+prylabs-bulldozer[bot]@users.noreply.github.com> Co-authored-by: Victor Farazdagi <simple.square@gmail.com>
81 lines
2.1 KiB
Go
81 lines
2.1 KiB
Go
package kv
|
|
|
|
import (
|
|
"context"
|
|
|
|
pb "github.com/prysmaticlabs/prysm/proto/beacon/p2p/v1"
|
|
bolt "go.etcd.io/bbolt"
|
|
"go.opencensus.io/trace"
|
|
)
|
|
|
|
// SaveStateSummary saves a state summary object to the DB.
|
|
func (s *Store) SaveStateSummary(ctx context.Context, summary *pb.StateSummary) error {
|
|
ctx, span := trace.StartSpan(ctx, "BeaconDB.SaveStateSummary")
|
|
defer span.End()
|
|
|
|
return s.SaveStateSummaries(ctx, []*pb.StateSummary{summary})
|
|
}
|
|
|
|
// SaveStateSummaries saves state summary objects to the DB.
|
|
func (s *Store) SaveStateSummaries(ctx context.Context, summaries []*pb.StateSummary) error {
|
|
ctx, span := trace.StartSpan(ctx, "BeaconDB.SaveStateSummaries")
|
|
defer span.End()
|
|
|
|
return s.db.Update(func(tx *bolt.Tx) error {
|
|
bucket := tx.Bucket(stateSummaryBucket)
|
|
for _, summary := range summaries {
|
|
enc, err := encode(ctx, summary)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if err := bucket.Put(summary.Root, enc); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
})
|
|
}
|
|
|
|
// StateSummary returns the state summary object from the db using input block root.
|
|
func (s *Store) StateSummary(ctx context.Context, blockRoot [32]byte) (*pb.StateSummary, error) {
|
|
ctx, span := trace.StartSpan(ctx, "BeaconDB.StateSummary")
|
|
defer span.End()
|
|
enc, err := s.stateSummaryBytes(ctx, blockRoot)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if len(enc) == 0 {
|
|
return nil, nil
|
|
}
|
|
summary := &pb.StateSummary{}
|
|
if err := decode(ctx, enc, summary); err != nil {
|
|
return nil, err
|
|
}
|
|
return summary, nil
|
|
}
|
|
|
|
// HasStateSummary returns true if a state summary exists in DB.
|
|
func (s *Store) HasStateSummary(ctx context.Context, blockRoot [32]byte) bool {
|
|
ctx, span := trace.StartSpan(ctx, "BeaconDB.HasStateSummary")
|
|
defer span.End()
|
|
enc, err := s.stateSummaryBytes(ctx, blockRoot)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return len(enc) > 0
|
|
}
|
|
|
|
func (s *Store) stateSummaryBytes(ctx context.Context, blockRoot [32]byte) ([]byte, error) {
|
|
ctx, span := trace.StartSpan(ctx, "BeaconDB.stateSummaryBytes")
|
|
defer span.End()
|
|
|
|
var enc []byte
|
|
err := s.db.View(func(tx *bolt.Tx) error {
|
|
bucket := tx.Bucket(stateSummaryBucket)
|
|
enc = bucket.Get(blockRoot[:])
|
|
return nil
|
|
})
|
|
|
|
return enc, err
|
|
}
|