mirror/prysm-pulse
1
0
Fork 0
mirror of https://gitlab.com/pulsechaincom/prysm-pulse.git synced 2025-01-03 16:37:39 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

native-state: Simplify MarshalSSZ (#10677)

Browse Source 
Co-authored-by: prylabs-bulldozer[bot] <58059840+prylabs-bulldozer[bot]@users.noreply.github.com>
This commit is contained in:
Preston Van Loon 2022-05-11 09:25:11 -05:00 committed by GitHub
parent abefe1e9d5
commit cc643ac4cc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 7 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
beacon-chain/state/state-native/BUILD.bazel
View File

@ -68,6 +68,7 @@ go_library(
"//encoding/ssz:go_default_library", "//encoding/ssz:go_default_library",
"//proto/prysm/v1alpha1:go_default_library", "//proto/prysm/v1alpha1:go_default_library",
"//runtime/version:go_default_library", "//runtime/version:go_default_library",
"@com_github_ferranbt_fastssz//:go_default_library",
"@com_github_pkg_errors//:go_default_library", "@com_github_pkg_errors//:go_default_library",
"@com_github_prysmaticlabs_go_bitfield//:go_default_library", "@com_github_prysmaticlabs_go_bitfield//:go_default_library",
"@io_opencensus_go//trace:go_default_library", "@io_opencensus_go//trace:go_default_library",

30
beacon-chain/state/state-native/ssz.go
View File

@ -1,36 +1,18 @@
package state_native package state_native
import ( import (
ssz "github.com/ferranbt/fastssz"
"github.com/pkg/errors" "github.com/pkg/errors"
ethpb "github.com/prysmaticlabs/prysm/proto/prysm/v1alpha1"
"github.com/prysmaticlabs/prysm/runtime/version"
) )
var errAssertionFailed = errors.New("failed to convert interface to proto state") var errAssertionFailed = errors.New("failed to convert interface to proto state")
var errUnsupportedVersion = errors.New("unsupported beacon state version")
func (b *BeaconState) MarshalSSZ() ([]byte, error) { func (b *BeaconState) MarshalSSZ() ([]byte, error) {
proto := b.ToProto() proto := b.ToProto()
switch b.Version() {
case version.Phase0: s, ok := proto.(ssz.Marshaler)
s, ok := proto.(*ethpb.BeaconState) if !ok {
if !ok { return nil, errAssertionFailed
return nil, errAssertionFailed
}
return s.MarshalSSZ()
case version.Altair:
s, ok := proto.(*ethpb.BeaconStateAltair)
if !ok {
return nil, errAssertionFailed
}
return s.MarshalSSZ()
case version.Bellatrix:
s, ok := proto.(*ethpb.BeaconStateBellatrix)
if !ok {
return nil, errAssertionFailed
}
return s.MarshalSSZ()
default:
return nil, errUnsupportedVersion
} }
return s.MarshalSSZ()
} }