erigon-pulse/consensus/bor/heimdallgrpc/state_sync.go

package heimdallgrpc

import (
	"context"
	"errors"
	"io"

	libcommon "github.com/ledgerwatch/erigon-lib/common"
	"github.com/ledgerwatch/erigon/common"
	"github.com/ledgerwatch/erigon/consensus/bor/clerk"
	proto "github.com/maticnetwork/polyproto/heimdall"
)

func (h *HeimdallGRPCClient) StateSyncEvents(ctx context.Context, fromID uint64, to int64) ([]*clerk.EventRecordWithTime, error) {
	eventRecords := make([]*clerk.EventRecordWithTime, 0)

	req := &proto.StateSyncEventsRequest{
		FromID: fromID,
		ToTime: uint64(to),
		Limit:  uint64(stateFetchLimit),
	}

	var (
		res    proto.Heimdall_StateSyncEventsClient
		events *proto.StateSyncEventsResponse
		err    error
	)

	res, err = h.client.StateSyncEvents(ctx, req)
	if err != nil {
		return nil, err
	}

	for {
		events, err = res.Recv()
		if errors.Is(err, io.EOF) {
			return eventRecords, nil
		}

		if err != nil {
			return nil, err
		}

		for _, event := range events.Result {
			eventRecord := &clerk.EventRecordWithTime{
				EventRecord: clerk.EventRecord{
					ID:       event.ID,
					Contract: libcommon.HexToAddress(event.Contract),
					Data:     common.Hex2Bytes(event.Data[2:]),
					TxHash:   libcommon.HexToHash(event.TxHash),
					LogIndex: event.LogIndex,
					ChainID:  event.ChainID,
				},
				Time: event.Time.AsTime(),
			}
			eventRecords = append(eventRecords, eventRecord)
		}
	}
}