mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 04:47:18 +00:00
45 lines
1.3 KiB
Go
45 lines
1.3 KiB
Go
package debug
|
|
|
|
import (
|
|
"context"
|
|
"encoding/hex"
|
|
|
|
"github.com/golang/protobuf/ptypes/empty"
|
|
pbrpc "github.com/prysmaticlabs/prysm/proto/prysm/v1alpha1"
|
|
)
|
|
|
|
// GetProtoArrayForkChoice returns proto array fork choice store.
|
|
func (ds *Server) GetProtoArrayForkChoice(_ context.Context, _ *empty.Empty) (*pbrpc.ProtoArrayForkChoiceResponse, error) {
|
|
store := ds.HeadFetcher.ProtoArrayStore()
|
|
|
|
nodes := store.Nodes()
|
|
returnedNodes := make([]*pbrpc.ProtoArrayNode, len(nodes))
|
|
|
|
for i := 0; i < len(returnedNodes); i++ {
|
|
r := nodes[i].Root()
|
|
returnedNodes[i] = &pbrpc.ProtoArrayNode{
|
|
Slot: nodes[i].Slot(),
|
|
Root: r[:],
|
|
Parent: nodes[i].Parent(),
|
|
JustifiedEpoch: nodes[i].JustifiedEpoch(),
|
|
FinalizedEpoch: nodes[i].FinalizedEpoch(),
|
|
Weight: nodes[i].Weight(),
|
|
BestChild: nodes[i].BestChild(),
|
|
BestDescendant: nodes[i].BestDescendant(),
|
|
}
|
|
}
|
|
|
|
indices := make(map[string]uint64, len(store.NodesIndices()))
|
|
for k, v := range store.NodesIndices() {
|
|
indices[hex.EncodeToString(k[:])] = v
|
|
}
|
|
|
|
return &pbrpc.ProtoArrayForkChoiceResponse{
|
|
PruneThreshold: store.PruneThreshold(),
|
|
JustifiedEpoch: store.JustifiedEpoch(),
|
|
FinalizedEpoch: store.FinalizedEpoch(),
|
|
ProtoArrayNodes: returnedNodes,
|
|
Indices: indices,
|
|
}, nil
|
|
}
|