mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 04:47:18 +00:00
06ee5695fb
* Save finalized checkpoint on save * gofmt and goimport * remove head.root arg * test * fixes * remove comment Co-authored-by: prylabs-bulldozer[bot] <58059840+prylabs-bulldozer[bot]@users.noreply.github.com>
42 lines
1.2 KiB
Go
42 lines
1.2 KiB
Go
package stategen
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/state"
|
|
"github.com/prysmaticlabs/prysm/shared/bytesutil"
|
|
"go.opencensus.io/trace"
|
|
)
|
|
|
|
// SaveState saves the state in the DB.
|
|
// It knows which cold and hot state section the input state should belong to.
|
|
func (s *State) SaveState(ctx context.Context, root [32]byte, state *state.BeaconState) error {
|
|
ctx, span := trace.StartSpan(ctx, "stateGen.SaveState")
|
|
defer span.End()
|
|
|
|
// The state belongs to the cold section if it's below the split slot threshold.
|
|
if state.Slot() < s.finalizedInfo.slot {
|
|
return s.saveColdState(ctx, root, state)
|
|
}
|
|
|
|
return s.saveHotState(ctx, root, state)
|
|
}
|
|
|
|
// ForceCheckpoint initiates a cold state save of the given state. This method does not update the
|
|
// "last archived state" but simply saves the specified state from the root argument into the DB.
|
|
func (s *State) ForceCheckpoint(ctx context.Context, root []byte) error {
|
|
ctx, span := trace.StartSpan(ctx, "stateGen.ForceCheckpoint")
|
|
defer span.End()
|
|
|
|
root32 := bytesutil.ToBytes32(root)
|
|
fs, err := s.loadHotStateByRoot(ctx, root32)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if err := s.beaconDB.SaveState(ctx, fs, root32); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|