eth-tracker/internal/syncer/syncer.go

package syncer

import (
	"context"
	"log/slog"

	"github.com/celo-org/celo-blockchain"
	"github.com/celo-org/celo-blockchain/core/types"
	"github.com/celo-org/celo-blockchain/ethclient"
	"github.com/ef-ds/deque/v2"
	"github.com/grassrootseconomics/celo-tracker/internal/chain"
	"github.com/grassrootseconomics/celo-tracker/internal/db"
	"github.com/grassrootseconomics/celo-tracker/internal/stats"
)

type (
	SyncerOpts struct {
		WebSocketEndpoint string
		EnableHistorical  bool
		StartBlock        uint64
		BatchQueue        *deque.Deque[uint64]
		BlocksQueue       *deque.Deque[types.Block]
		BatchSize         int
		Chain             *chain.Chain
		Logg              *slog.Logger
		Stats             *stats.Stats
		DB                *db.DB
	}

	Syncer struct {
		batchQueue        *deque.Deque[uint64]
		blocksQueue       *deque.Deque[types.Block]
		chain             *chain.Chain
		logg              *slog.Logger
		stats             *stats.Stats
		ethClient         *ethclient.Client
		batchSize         int
		db                *db.DB
		quit              chan struct{}
		startBlock        uint64
		realtimeSub       celo.Subscription
		historicalEnabled bool
	}
)

func New(o SyncerOpts) (*Syncer, error) {
	if o.EnableHistorical {
		latestBlock, err := o.Chain.GetLatestBlock(context.Background())
		if err != nil {
			return nil, err
		}

		if o.StartBlock == 0 {
			o.StartBlock = latestBlock
		}

		if err := o.DB.SetLowerBound(o.StartBlock); err != nil {
			return nil, err
		}

		if err := o.DB.SetUpperBound(latestBlock); err != nil {
			return nil, err
		}
	}

	ethClient, err := ethclient.Dial(o.WebSocketEndpoint)
	if err != nil {
		return nil, err
	}

	return &Syncer{
		batchQueue:        o.BatchQueue,
		blocksQueue:       o.BlocksQueue,
		chain:             o.Chain,
		logg:              o.Logg,
		stats:             o.Stats,
		ethClient:         ethClient,
		db:                o.DB,
		batchSize:         o.BatchSize,
		quit:              make(chan struct{}),
		startBlock:        o.StartBlock,
		historicalEnabled: o.EnableHistorical,
	}, nil
}