/
historical_data_retrieval.go
53 lines (50 loc) · 1.5 KB
/
historical_data_retrieval.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
45
46
47
48
49
50
51
52
53
package beaconclient
import (
"context"
types "github.com/prysmaticlabs/eth2-types"
ethpb "github.com/prysmaticlabs/prysm/proto/eth/v1alpha1"
"github.com/prysmaticlabs/prysm/shared/cmd"
"go.opencensus.io/trace"
)
// RequestHistoricalAttestations requests all indexed attestations for a
// given epoch from a beacon node via gRPC.
func (s *Service) RequestHistoricalAttestations(
ctx context.Context,
epoch types.Epoch,
) ([]*ethpb.IndexedAttestation, error) {
ctx, span := trace.StartSpan(ctx, "beaconclient.RequestHistoricalAttestations")
defer span.End()
indexedAtts := make([]*ethpb.IndexedAttestation, 0)
res := ðpb.ListIndexedAttestationsResponse{}
var err error
for {
if ctx.Err() != nil {
return nil, ctx.Err()
}
if res == nil {
res = ðpb.ListIndexedAttestationsResponse{}
}
res, err = s.cfg.BeaconClient.ListIndexedAttestations(ctx, ðpb.ListIndexedAttestationsRequest{
QueryFilter: ðpb.ListIndexedAttestationsRequest_Epoch{
Epoch: epoch,
},
PageSize: int32(cmd.Get().MaxRPCPageSize),
PageToken: res.NextPageToken,
})
if err != nil {
log.WithError(err).Errorf("could not request indexed attestations for epoch: %d", epoch)
break
}
indexedAtts = append(indexedAtts, res.IndexedAttestations...)
log.Infof(
"Retrieved %d/%d indexed attestations for epoch %d",
len(indexedAtts),
res.TotalSize,
epoch,
)
if res.NextPageToken == "" || res.TotalSize == 0 || len(indexedAtts) == int(res.TotalSize) {
break
}
}
return indexedAtts, nil
}