mirror of
https://github.com/scroll-tech/scroll.git
synced 2026-01-14 00:18:03 -05:00
314 lines
12 KiB
Go
314 lines
12 KiB
Go
package watcher
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"math/big"
|
|
"time"
|
|
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
"github.com/prometheus/client_golang/prometheus/promauto"
|
|
"github.com/scroll-tech/go-ethereum/common"
|
|
"github.com/scroll-tech/go-ethereum/log"
|
|
"github.com/scroll-tech/go-ethereum/params"
|
|
"gorm.io/gorm"
|
|
|
|
"scroll-tech/common/forks"
|
|
"scroll-tech/common/types"
|
|
"scroll-tech/common/types/encoding"
|
|
|
|
"scroll-tech/rollup/internal/config"
|
|
"scroll-tech/rollup/internal/orm"
|
|
"scroll-tech/rollup/internal/utils"
|
|
)
|
|
|
|
// BatchProposer proposes batches based on available unbatched chunks.
|
|
type BatchProposer struct {
|
|
ctx context.Context
|
|
db *gorm.DB
|
|
|
|
batchOrm *orm.Batch
|
|
chunkOrm *orm.Chunk
|
|
l2BlockOrm *orm.L2Block
|
|
|
|
maxChunkNumPerBatch uint64
|
|
maxL1CommitGasPerBatch uint64
|
|
maxL1CommitCalldataSizePerBatch uint64
|
|
batchTimeoutSec uint64
|
|
gasCostIncreaseMultiplier float64
|
|
forkMap map[uint64]bool
|
|
|
|
cfg *config.BatchProposerConfig
|
|
chainCfg *params.ChainConfig
|
|
|
|
batchProposerCircleTotal prometheus.Counter
|
|
proposeBatchFailureTotal prometheus.Counter
|
|
proposeBatchUpdateInfoTotal prometheus.Counter
|
|
proposeBatchUpdateInfoFailureTotal prometheus.Counter
|
|
totalL1CommitGas prometheus.Gauge
|
|
totalL1CommitCalldataSize prometheus.Gauge
|
|
totalL1CommitBlobSize prometheus.Gauge
|
|
batchChunksNum prometheus.Gauge
|
|
batchFirstBlockTimeoutReached prometheus.Counter
|
|
batchChunksProposeNotEnoughTotal prometheus.Counter
|
|
}
|
|
|
|
// NewBatchProposer creates a new BatchProposer instance.
|
|
func NewBatchProposer(ctx context.Context, cfg *config.BatchProposerConfig, chainCfg *params.ChainConfig, db *gorm.DB, reg prometheus.Registerer) *BatchProposer {
|
|
forkHeights, forkMap, _ := forks.CollectSortedForkHeights(chainCfg)
|
|
log.Debug("new batch proposer",
|
|
"maxChunkNumPerBatch", cfg.MaxChunkNumPerBatch,
|
|
"maxL1CommitGasPerBatch", cfg.MaxL1CommitGasPerBatch,
|
|
"maxL1CommitCalldataSizePerBatch", cfg.MaxL1CommitCalldataSizePerBatch,
|
|
"batchTimeoutSec", cfg.BatchTimeoutSec,
|
|
"gasCostIncreaseMultiplier", cfg.GasCostIncreaseMultiplier,
|
|
"forkHeights", forkHeights)
|
|
|
|
p := &BatchProposer{
|
|
ctx: ctx,
|
|
db: db,
|
|
batchOrm: orm.NewBatch(db),
|
|
chunkOrm: orm.NewChunk(db),
|
|
l2BlockOrm: orm.NewL2Block(db),
|
|
maxChunkNumPerBatch: cfg.MaxChunkNumPerBatch,
|
|
maxL1CommitGasPerBatch: cfg.MaxL1CommitGasPerBatch,
|
|
maxL1CommitCalldataSizePerBatch: cfg.MaxL1CommitCalldataSizePerBatch,
|
|
batchTimeoutSec: cfg.BatchTimeoutSec,
|
|
gasCostIncreaseMultiplier: cfg.GasCostIncreaseMultiplier,
|
|
forkMap: forkMap,
|
|
cfg: cfg,
|
|
chainCfg: chainCfg,
|
|
|
|
batchProposerCircleTotal: promauto.With(reg).NewCounter(prometheus.CounterOpts{
|
|
Name: "rollup_propose_batch_circle_total",
|
|
Help: "Total number of propose batch total.",
|
|
}),
|
|
proposeBatchFailureTotal: promauto.With(reg).NewCounter(prometheus.CounterOpts{
|
|
Name: "rollup_propose_batch_failure_circle_total",
|
|
Help: "Total number of propose batch total.",
|
|
}),
|
|
proposeBatchUpdateInfoTotal: promauto.With(reg).NewCounter(prometheus.CounterOpts{
|
|
Name: "rollup_propose_batch_update_info_total",
|
|
Help: "Total number of propose batch update info total.",
|
|
}),
|
|
proposeBatchUpdateInfoFailureTotal: promauto.With(reg).NewCounter(prometheus.CounterOpts{
|
|
Name: "rollup_propose_batch_update_info_failure_total",
|
|
Help: "Total number of propose batch update info failure total.",
|
|
}),
|
|
totalL1CommitGas: promauto.With(reg).NewGauge(prometheus.GaugeOpts{
|
|
Name: "rollup_propose_batch_total_l1_commit_gas",
|
|
Help: "The total l1 commit gas",
|
|
}),
|
|
totalL1CommitCalldataSize: promauto.With(reg).NewGauge(prometheus.GaugeOpts{
|
|
Name: "rollup_propose_batch_total_l1_call_data_size",
|
|
Help: "The total l1 call data size",
|
|
}),
|
|
totalL1CommitBlobSize: promauto.With(reg).NewGauge(prometheus.GaugeOpts{
|
|
Name: "rollup_propose_batch_total_l1_commit_blob_size",
|
|
Help: "The total l1 commit blob size",
|
|
}),
|
|
batchChunksNum: promauto.With(reg).NewGauge(prometheus.GaugeOpts{
|
|
Name: "rollup_propose_batch_chunks_number",
|
|
Help: "The number of chunks in the batch",
|
|
}),
|
|
batchFirstBlockTimeoutReached: promauto.With(reg).NewCounter(prometheus.CounterOpts{
|
|
Name: "rollup_propose_batch_first_block_timeout_reached_total",
|
|
Help: "Total times of batch's first block timeout reached",
|
|
}),
|
|
batchChunksProposeNotEnoughTotal: promauto.With(reg).NewCounter(prometheus.CounterOpts{
|
|
Name: "rollup_propose_batch_chunks_propose_not_enough_total",
|
|
Help: "Total number of batch chunk propose not enough",
|
|
}),
|
|
}
|
|
|
|
return p
|
|
}
|
|
|
|
// TryProposeBatch tries to propose a new batches.
|
|
func (p *BatchProposer) TryProposeBatch() {
|
|
p.batchProposerCircleTotal.Inc()
|
|
if err := p.proposeBatch(); err != nil {
|
|
p.proposeBatchFailureTotal.Inc()
|
|
log.Error("proposeBatchChunks failed", "err", err)
|
|
return
|
|
}
|
|
}
|
|
|
|
func (p *BatchProposer) updateDBBatchInfo(batch *encoding.Batch, codecVersion encoding.CodecVersion) error {
|
|
err := p.db.Transaction(func(dbTX *gorm.DB) error {
|
|
batch, dbErr := p.batchOrm.InsertBatch(p.ctx, batch, codecVersion, dbTX)
|
|
if dbErr != nil {
|
|
log.Warn("BatchProposer.updateBatchInfoInDB insert batch failure",
|
|
"start chunk index", batch.StartChunkIndex, "end chunk index", batch.EndChunkIndex, "error", dbErr)
|
|
return dbErr
|
|
}
|
|
dbErr = p.chunkOrm.UpdateBatchHashInRange(p.ctx, batch.StartChunkIndex, batch.EndChunkIndex, batch.Hash, dbTX)
|
|
if dbErr != nil {
|
|
log.Warn("BatchProposer.UpdateBatchHashInRange update the chunk's batch hash failure", "hash", batch.Hash, "error", dbErr)
|
|
return dbErr
|
|
}
|
|
|
|
skipProof := false
|
|
if p.cfg.EnableTestEnvSamplingFeature && ((batch.Index % 100) >= p.cfg.SamplingPercentage) {
|
|
skipProof = true
|
|
}
|
|
if skipProof {
|
|
dbErr = p.batchOrm.UpdateProvingStatus(p.ctx, batch.Hash, types.ProvingTaskVerified, dbTX)
|
|
if dbErr != nil {
|
|
log.Warn("BatchProposer.updateBatchInfoInDB update batch proving_status failure",
|
|
"batch hash", batch.Hash, "error", dbErr)
|
|
return dbErr
|
|
}
|
|
dbErr = p.chunkOrm.UpdateProvingStatusInRange(p.ctx, batch.StartChunkIndex, batch.EndChunkIndex, types.ProvingTaskVerified, dbTX)
|
|
if dbErr != nil {
|
|
log.Warn("BatchProposer.updateBatchInfoInDB update chunk proving_status failure",
|
|
"start chunk index", batch.StartChunkIndex, "end chunk index", batch.EndChunkIndex,
|
|
"batch hash", batch.Hash, "error", dbErr)
|
|
return dbErr
|
|
}
|
|
}
|
|
|
|
return nil
|
|
})
|
|
if err != nil {
|
|
p.proposeBatchUpdateInfoFailureTotal.Inc()
|
|
log.Error("update batch info in db failed", "err", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (p *BatchProposer) proposeBatch() error {
|
|
unbatchedChunkIndex, err := p.batchOrm.GetFirstUnbatchedChunkIndex(p.ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// select at most p.maxChunkNumPerBatch chunks
|
|
dbChunks, err := p.chunkOrm.GetChunksGEIndex(p.ctx, unbatchedChunkIndex, int(p.maxChunkNumPerBatch))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if len(dbChunks) == 0 {
|
|
return nil
|
|
}
|
|
|
|
maxChunksThisBatch := p.maxChunkNumPerBatch
|
|
for i, chunk := range dbChunks {
|
|
// if a chunk is starting at a fork boundary, only consider earlier chunks
|
|
if i != 0 && p.forkMap[chunk.StartBlockNumber] {
|
|
dbChunks = dbChunks[:i]
|
|
if uint64(len(dbChunks)) < maxChunksThisBatch {
|
|
maxChunksThisBatch = uint64(len(dbChunks))
|
|
}
|
|
break
|
|
}
|
|
}
|
|
|
|
codecVersion := encoding.CodecV0
|
|
if p.chainCfg.IsBernoulli(new(big.Int).SetUint64(dbChunks[0].StartBlockNumber)) {
|
|
codecVersion = encoding.CodecV1
|
|
}
|
|
|
|
daChunks, err := p.getDAChunks(dbChunks)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
dbParentBatch, err := p.batchOrm.GetLatestBatch(p.ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var batch encoding.Batch
|
|
batch.Index = dbParentBatch.Index + 1
|
|
batch.ParentBatchHash = common.HexToHash(dbParentBatch.Hash)
|
|
parentBatchEndBlockNumber := daChunks[0].Blocks[0].Header.Number.Uint64() - 1
|
|
parentBatchCodecVersion := encoding.CodecV0
|
|
// Genesis batch uses codecv0 encoding, otherwise using bernoulli fork to choose codec version.
|
|
if dbParentBatch.Index > 0 && p.chainCfg.IsBernoulli(new(big.Int).SetUint64(parentBatchEndBlockNumber)) {
|
|
parentBatchCodecVersion = encoding.CodecV1
|
|
}
|
|
batch.TotalL1MessagePoppedBefore, err = utils.GetTotalL1MessagePoppedBeforeBatch(dbParentBatch.BatchHeader, parentBatchCodecVersion)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for i, chunk := range daChunks {
|
|
batch.Chunks = append(batch.Chunks, chunk)
|
|
metrics, calcErr := utils.CalculateBatchMetrics(&batch, codecVersion)
|
|
if calcErr != nil {
|
|
return fmt.Errorf("failed to calculate batch metrics: %w", calcErr)
|
|
}
|
|
totalOverEstimateL1CommitGas := uint64(p.gasCostIncreaseMultiplier * float64(metrics.L1CommitGas))
|
|
if metrics.L1CommitCalldataSize > p.maxL1CommitCalldataSizePerBatch ||
|
|
totalOverEstimateL1CommitGas > p.maxL1CommitGasPerBatch ||
|
|
metrics.L1CommitBlobSize > maxBlobSize {
|
|
if i == 0 {
|
|
// The first chunk exceeds hard limits, which indicates a bug in the chunk-proposer, manual fix is needed.
|
|
return fmt.Errorf("the first chunk exceeds limits; start block number: %v, end block number: %v, limits: %+v, maxChunkNum: %v, maxL1CommitCalldataSize: %v, maxL1CommitGas: %v, maxBlobSize: %v",
|
|
dbChunks[0].StartBlockNumber, dbChunks[0].EndBlockNumber, metrics, p.maxChunkNumPerBatch, p.maxL1CommitCalldataSizePerBatch, p.maxL1CommitGasPerBatch, maxBlobSize)
|
|
}
|
|
|
|
log.Debug("breaking limit condition in batching",
|
|
"currentL1CommitCalldataSize", metrics.L1CommitCalldataSize,
|
|
"maxL1CommitCalldataSizePerBatch", p.maxL1CommitCalldataSizePerBatch,
|
|
"currentOverEstimateL1CommitGas", totalOverEstimateL1CommitGas,
|
|
"maxL1CommitGasPerBatch", p.maxL1CommitGasPerBatch)
|
|
|
|
batch.Chunks = batch.Chunks[:len(batch.Chunks)-1]
|
|
|
|
metrics, err := utils.CalculateBatchMetrics(&batch, codecVersion)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to calculate batch metrics: %w", err)
|
|
}
|
|
|
|
p.recordBatchMetrics(metrics)
|
|
return p.updateDBBatchInfo(&batch, codecVersion)
|
|
}
|
|
}
|
|
|
|
metrics, calcErr := utils.CalculateBatchMetrics(&batch, codecVersion)
|
|
if calcErr != nil {
|
|
return fmt.Errorf("failed to calculate batch metrics: %w", calcErr)
|
|
}
|
|
currentTimeSec := uint64(time.Now().Unix())
|
|
if metrics.FirstBlockTimestamp+p.batchTimeoutSec < currentTimeSec || metrics.NumChunks == maxChunksThisBatch {
|
|
log.Info("reached maximum number of chunks in batch or first block timeout",
|
|
"chunk count", metrics.NumChunks,
|
|
"start block number", dbChunks[0].StartBlockNumber,
|
|
"start block timestamp", dbChunks[0].StartBlockTime,
|
|
"current time", currentTimeSec)
|
|
|
|
p.batchFirstBlockTimeoutReached.Inc()
|
|
p.recordBatchMetrics(metrics)
|
|
return p.updateDBBatchInfo(&batch, codecVersion)
|
|
}
|
|
|
|
log.Debug("pending chunks do not reach one of the constraints or contain a timeout block")
|
|
p.batchChunksProposeNotEnoughTotal.Inc()
|
|
return nil
|
|
}
|
|
|
|
func (p *BatchProposer) getDAChunks(dbChunks []*orm.Chunk) ([]*encoding.Chunk, error) {
|
|
chunks := make([]*encoding.Chunk, len(dbChunks))
|
|
for i, c := range dbChunks {
|
|
blocks, err := p.l2BlockOrm.GetL2BlocksInRange(p.ctx, c.StartBlockNumber, c.EndBlockNumber)
|
|
if err != nil {
|
|
log.Error("Failed to fetch blocks", "start number", c.StartBlockNumber, "end number", c.EndBlockNumber, "error", err)
|
|
return nil, err
|
|
}
|
|
chunks[i] = &encoding.Chunk{
|
|
Blocks: blocks,
|
|
}
|
|
}
|
|
return chunks, nil
|
|
}
|
|
|
|
func (p *BatchProposer) recordBatchMetrics(metrics *utils.BatchMetrics) {
|
|
p.totalL1CommitGas.Set(float64(metrics.L1CommitGas))
|
|
p.totalL1CommitCalldataSize.Set(float64(metrics.L1CommitCalldataSize))
|
|
p.batchChunksNum.Set(float64(metrics.NumChunks))
|
|
p.totalL1CommitBlobSize.Set(float64(metrics.L1CommitBlobSize))
|
|
}
|