prysm-pulse/beacon-chain/state/genesis/genesis.go
terence tsao 012d279663
Fix duplicated imports (#9304)
* Fix duplicated imports

* Fix metrics test
2021-07-29 16:45:17 -05:00

40 lines
927 B
Go

package genesis
import (
_ "embed"
"github.com/golang/snappy"
state "github.com/prysmaticlabs/prysm/beacon-chain/state/v1"
ethpb "github.com/prysmaticlabs/prysm/proto/prysm/v1alpha1"
"github.com/prysmaticlabs/prysm/shared/params"
)
var (
//go:embed mainnet.ssz.snappy
mainnetRawSSZCompressed []byte // 1.8Mb
)
// State returns a copy of the genesis state from a hardcoded value.
func State(name string) (*state.BeaconState, error) {
switch name {
case params.ConfigNames[params.Mainnet]:
return load(mainnetRawSSZCompressed)
default:
// No state found.
return nil, nil
}
}
// load a compressed ssz state file into a beacon state struct.
func load(b []byte) (*state.BeaconState, error) {
st := &ethpb.BeaconState{}
b, err := snappy.Decode(nil /*dst*/, b)
if err != nil {
return nil, err
}
if err := st.UnmarshalSSZ(b); err != nil {
return nil, err
}
return state.InitializeFromProtoUnsafe(st)
}