go-ethereum/rollup/da_syncer/block_queue.go

package da_syncer

import (
	"context"
	"fmt"

	"github.com/scroll-tech/go-ethereum/rollup/da_syncer/da"
)

// BlockQueue is a pipeline stage that reads batches from BatchQueue, extracts all da.PartialBlock from it and
// provides them to the next stage one-by-one.
type BlockQueue struct {
	batchQueue *BatchQueue
	blocks     []*da.PartialBlock
}

func NewBlockQueue(batchQueue *BatchQueue) *BlockQueue {
	return &BlockQueue{
		batchQueue: batchQueue,
		blocks:     make([]*da.PartialBlock, 0),
	}
}

func (bq *BlockQueue) NextBlock(ctx context.Context) (*da.PartialBlock, error) {
	for len(bq.blocks) == 0 {
		err := bq.getBlocksFromBatch(ctx)
		if err != nil {
			return nil, err
		}
	}
	block := bq.blocks[0]
	bq.blocks = bq.blocks[1:]
	return block, nil
}

func (bq *BlockQueue) getBlocksFromBatch(ctx context.Context) error {
	daEntry, err := bq.batchQueue.NextBatch(ctx)
	if err != nil {
		return err
	}

	entryWithBlocks, ok := daEntry.(da.EntryWithBlocks)
	// this should never happen because we only receive CommitBatch entries
	if !ok {
		return fmt.Errorf("unexpected type of daEntry: %T", daEntry)
	}

	bq.blocks = entryWithBlocks.Blocks()

	return nil
}

func (bq *BlockQueue) Reset(height uint64) {
	bq.blocks = make([]*da.PartialBlock, 0)
	bq.batchQueue.Reset(height)
}