mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 04:47:18 +00:00
29513c804c
* bytesutil * gaz Co-authored-by: prylabs-bulldozer[bot] <58059840+prylabs-bulldozer[bot]@users.noreply.github.com>
58 lines
1.7 KiB
Go
58 lines
1.7 KiB
Go
package debug
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/prysmaticlabs/prysm/encoding/bytesutil"
|
|
pbrpc "github.com/prysmaticlabs/prysm/proto/prysm/v1alpha1"
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
)
|
|
|
|
// GetBeaconState retrieves an ssz-encoded beacon state
|
|
// from the beacon node by either a slot or block root.
|
|
func (ds *Server) GetBeaconState(
|
|
ctx context.Context,
|
|
req *pbrpc.BeaconStateRequest,
|
|
) (*pbrpc.SSZResponse, error) {
|
|
switch q := req.QueryFilter.(type) {
|
|
case *pbrpc.BeaconStateRequest_Slot:
|
|
currentSlot := ds.GenesisTimeFetcher.CurrentSlot()
|
|
requestedSlot := q.Slot
|
|
if requestedSlot > currentSlot {
|
|
return nil, status.Errorf(
|
|
codes.InvalidArgument,
|
|
"Cannot retrieve information about a slot in the future, current slot %d, requested slot %d",
|
|
currentSlot,
|
|
requestedSlot,
|
|
)
|
|
}
|
|
|
|
st, err := ds.StateGen.StateBySlot(ctx, q.Slot)
|
|
if err != nil {
|
|
return nil, status.Errorf(codes.Internal, "Could not compute state by slot: %v", err)
|
|
}
|
|
encoded, err := st.MarshalSSZ()
|
|
if err != nil {
|
|
return nil, status.Errorf(codes.Internal, "Could not ssz encode beacon state: %v", err)
|
|
}
|
|
return &pbrpc.SSZResponse{
|
|
Encoded: encoded,
|
|
}, nil
|
|
case *pbrpc.BeaconStateRequest_BlockRoot:
|
|
st, err := ds.StateGen.StateByRoot(ctx, bytesutil.ToBytes32(q.BlockRoot))
|
|
if err != nil {
|
|
return nil, status.Errorf(codes.Internal, "Could not compute state by block root: %v", err)
|
|
}
|
|
encoded, err := st.MarshalSSZ()
|
|
if err != nil {
|
|
return nil, status.Errorf(codes.Internal, "Could not ssz encode beacon state: %v", err)
|
|
}
|
|
return &pbrpc.SSZResponse{
|
|
Encoded: encoded,
|
|
}, nil
|
|
default:
|
|
return nil, status.Error(codes.InvalidArgument, "Need to specify either a block root or slot to request state")
|
|
}
|
|
}
|