erigon-pulse/cmd/state/exec22/txNums.go

package exec22

import (
	"context"
	"fmt"
	"sort"

	"github.com/ledgerwatch/erigon-lib/common/cmp"
	"github.com/ledgerwatch/erigon-lib/kv"
	"github.com/ledgerwatch/erigon/cmd/hack/tool"
	"github.com/ledgerwatch/erigon/eth/stagedsync/stages"
	"github.com/ledgerwatch/erigon/turbo/snapshotsync"
)

type TxNums struct {
	nums []uint64
}

func (s *TxNums) LastBlockNum() uint64                  { return uint64(len(s.nums) - 1) }
func (s *TxNums) MaxOf(blockNum uint64) (txnNum uint64) { return s.nums[blockNum] }
func (s *TxNums) MinOf(blockNum uint64) (txnNum uint64) {
	if blockNum == 0 {
		return 0
	}
	return s.nums[blockNum-1] + 1
}
func (s *TxNums) Append(blockNum, maxTxnNum uint64) {
	if len(s.nums) > int(blockNum) {
		err := fmt.Errorf("trying append blockNum=%d, but already have=%d\n", blockNum, len(s.nums))
		panic(err)
	}
	s.nums = append(s.nums, maxTxnNum)
	//fmt.Printf("append: %d, %d, %d\n", blockNum, maxTxnNum, len(s.nums))
}
func (s *TxNums) Unwind(unwindTo uint64) {
	s.nums = s.nums[:unwindTo]
	//fmt.Printf("unwind: %d, %d\n", unwindTo, s.nums)
}
func (s *TxNums) Find(endTxNumMinimax uint64) (ok bool, blockNum uint64) {
	blockNum = uint64(sort.Search(len(s.nums), func(i int) bool {
		return s.nums[i] > endTxNumMinimax
	}))
	return int(blockNum) != len(s.nums), blockNum
}

func TxNumsFromDB(s *snapshotsync.RoSnapshots, db kv.RoDB) *TxNums {
	historyV2 := tool.HistoryV2FromDB(db)
	if !historyV2 {
		return nil
	}

	toBlock := s.BlocksAvailable()
	if err := db.View(context.Background(), func(tx kv.Tx) error {
		p, err := stages.GetStageProgress(tx, stages.Bodies)
		if err != nil {
			return err
		}
		toBlock = cmp.Max(toBlock, p)
		return nil
	}); err != nil {
		panic(err)
	}

	txNums := &TxNums{nums: make([]uint64, toBlock+1)}
	if err := db.View(context.Background(), func(tx kv.Tx) error {
		if err := (snapshotsync.BodiesIterator{}).ForEach(tx, s, 0, func(blockNum, baseTxNum, txAmount uint64) error {
			if blockNum > toBlock {
				return nil
			}
			txNums.nums[blockNum] = baseTxNum + txAmount
			return nil
		}); err != nil {
			return fmt.Errorf("build txNum => blockNum mapping: %w", err)
		}
		return nil
	}); err != nil {
		panic(err)
	}
	return txNums
}