/
data.go
44 lines (37 loc) · 1.39 KB
/
data.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
package evaluators
import (
"context"
"errors"
"github.com/prysmaticlabs/prysm/v5/consensus-types/primitives"
eth "github.com/prysmaticlabs/prysm/v5/proto/prysm/v1alpha1"
e2etypes "github.com/prysmaticlabs/prysm/v5/testing/endtoend/types"
"google.golang.org/grpc"
)
const epochToCheck = 50 // must be more than 46 (32 hot states + 16 chkpt interval)
// ColdStateCheckpoint checks data from the database using cold state storage.
var ColdStateCheckpoint = e2etypes.Evaluator{
Name: "cold_state_assignments_from_epoch_%d",
Policy: func(currentEpoch primitives.Epoch) bool {
return currentEpoch == epochToCheck
},
Evaluation: checkColdStateCheckpoint,
}
// Checks the first node for an old checkpoint using cold state storage.
func checkColdStateCheckpoint(_ *e2etypes.EvaluationContext, conns ...*grpc.ClientConn) error {
ctx := context.Background()
client := eth.NewBeaconChainClient(conns[0])
for i := primitives.Epoch(0); i < epochToCheck; i++ {
res, err := client.ListValidatorAssignments(ctx, ð.ListValidatorAssignmentsRequest{
QueryFilter: ð.ListValidatorAssignmentsRequest_Epoch{Epoch: i},
})
if err != nil {
return err
}
// A simple check to ensure we received some data.
if res == nil || res.Epoch != i {
return errors.New("failed to return a validator assignments response for an old epoch " +
"using cold state storage from the database")
}
}
return nil
}