simplified block request calls

vms/platformvm/block/builder/builder.go

@@ -13,7 +13,6 @@ import (
 
 	"github.com/ava-labs/avalanchego/ids"
 	"github.com/ava-labs/avalanchego/snow/consensus/snowman"
-	"github.com/ava-labs/avalanchego/snow/engine/common"
 	"github.com/ava-labs/avalanchego/utils/timer"
 	"github.com/ava-labs/avalanchego/utils/timer/mockable"
 	"github.com/ava-labs/avalanchego/utils/units"
@@ -40,7 +39,11 @@ var (
 
 type Builder interface {
 	mempool.Mempool
-	mempool.BlockTimer
+
+	// ResetBlockTimer schedules a timer to notify the consensus engine once
+	// there is a block ready to be built. If a block is ready to be built when
+	// this function is called, the engine will be notified directly.
+	ResetBlockTimer()
 
 	// BuildBlock is called on timer clock to attempt to create
 	// next block
@@ -58,9 +61,6 @@ type builder struct {
 	txExecutorBackend *txexecutor.Backend
 	blkManager        blockexecutor.Manager
 
-	// channel to send messages to the consensus engine
-	toEngine chan<- common.Message
-
 	// This timer goes off when it is time for the next validator to add/leave
 	// the validator set. When it goes off ResetTimer() is called, potentially
 	// triggering creation of a new block.
@@ -72,14 +72,12 @@ func New(
 	txBuilder txbuilder.Builder,
 	txExecutorBackend *txexecutor.Backend,
 	blkManager blockexecutor.Manager,
-	toEngine chan<- common.Message,
 ) Builder {
 	builder := &builder{
 		Mempool:           mempool,
 		txBuilder:         txBuilder,
 		txExecutorBackend: txExecutorBackend,
 		blkManager:        blkManager,
-		toEngine:          toEngine,
 	}
 
 	builder.timer = timer.NewTimer(builder.setNextBuildBlockTime)
@@ -192,7 +190,7 @@ func (b *builder) setNextBuildBlockTime() {
 
 	if _, err := b.buildBlock(); err == nil {
 		// We can build a block now
-		b.notifyBlockReady()
+		b.Mempool.RequestBuildBlock()
 		return
 	}
 
@@ -229,16 +227,6 @@ func (b *builder) setNextBuildBlockTime() {
 	b.timer.SetTimeoutIn(waitTime)
 }
 
-// notifyBlockReady tells the consensus engine that a new block is ready to be
-// created
-func (b *builder) notifyBlockReady() {
-	select {
-	case b.toEngine <- common.PendingTxs:
-	default:
-		b.txExecutorBackend.Ctx.Log.Debug("dropping message to consensus engine")
-	}
-}
-
 // [timestamp] is min(max(now, parent timestamp), next staker change time)
 func buildBlock(
 	builder *builder,

vms/platformvm/block/builder/builder_test.go

@@ -80,7 +80,7 @@ func TestPreviouslyDroppedTxsCanBeReAddedToMempool(t *testing.T) {
 	txID := tx.ID()
 
 	// A tx simply added to mempool is obviously not marked as dropped
-	require.NoError(env.mempool.Add(tx))
+	require.NoError(env.mempool.Add([]*txs.Tx{tx}))
 	require.True(env.mempool.Has(txID))
 	reason := env.mempool.GetDropReason(txID)
 	require.NoError(reason)
@@ -94,7 +94,7 @@ func TestPreviouslyDroppedTxsCanBeReAddedToMempool(t *testing.T) {
 	// A previously dropped tx, popped then re-added to mempool,
 	// is not dropped anymore
 	env.mempool.Remove([]*txs.Tx{tx})
-	require.NoError(env.mempool.Add(tx))
+	require.NoError(env.mempool.Add([]*txs.Tx{tx}))
 
 	require.True(env.mempool.Has(txID))
 	reason = env.mempool.GetDropReason(txID)

vms/platformvm/block/builder/helpers_test.go

@@ -169,7 +169,7 @@ func newEnvironment(t *testing.T) *environment {
 	metrics, err := metrics.New("", registerer)
 	require.NoError(err)
 
-	res.mempool, err = mempool.New("mempool", registerer, res)
+	res.mempool, err = mempool.New(res.ctx, "mempool", registerer, nil)
 	require.NoError(err)
 
 	res.blkManager = blockexecutor.NewManager(
@@ -193,7 +193,6 @@ func newEnvironment(t *testing.T) *environment {
 		res.txBuilder,
 		&res.backend,
 		res.blkManager,
-		nil, // toEngine,
 	)
 
 	res.blkManager.SetPreference(genesisID)

vms/platformvm/block/builder/standard_block_test.go

@@ -72,7 +72,7 @@ func TestAtomicTxImports(t *testing.T) {
 	)
 	require.NoError(err)
 
-	require.NoError(env.Builder.Add(tx))
+	require.NoError(env.Builder.Add([]*txs.Tx{tx}))
 	b, err := env.Builder.BuildBlock(context.Background())
 	require.NoError(err)
 	// Test multiple verify calls work

vms/platformvm/block/executor/helpers_test.go

@@ -63,8 +63,6 @@ const (
 )
 
 var (
-	_ mempool.BlockTimer = (*environment)(nil)
-
 	defaultMinStakingDuration = 24 * time.Hour
 	defaultMaxStakingDuration = 365 * 24 * time.Hour
 	defaultGenesisTime        = time.Date(1997, 1, 1, 0, 0, 0, 0, time.UTC)
@@ -131,10 +129,6 @@ type environment struct {
 	backend        *executor.Backend
 }
 
-func (*environment) ResetBlockTimer() {
-	// dummy call, do nothing for now
-}
-
 func newEnvironment(t *testing.T, ctrl *gomock.Controller) *environment {
 	res := &environment{
 		isBootstrapped: &utils.Atomic[bool]{},
@@ -199,7 +193,7 @@ func newEnvironment(t *testing.T, ctrl *gomock.Controller) *environment {
 	metrics := metrics.Noop
 
 	var err error
-	res.mempool, err = mempool.New("mempool", registerer, res)
+	res.mempool, err = mempool.New(res.ctx, "mempool", registerer, nil)
 	if err != nil {
 		panic(fmt.Errorf("failed to create mempool: %w", err))
 	}

vms/platformvm/block/executor/rejector.go

@@ -71,15 +71,11 @@ func (r *rejector) rejectBlock(b block.Block, blockType string) error {
 		return nil
 	}
 
-	for _, tx := range b.Txs() {
-		if err := r.Mempool.Add(tx); err != nil {
-			r.ctx.Log.Debug(
-				"failed to reissue tx",
-				zap.Stringer("txID", tx.ID()),
-				zap.Stringer("blkID", blkID),
-				zap.Error(err),
-			)
-		}
+	if err := r.Mempool.Add(b.Txs()); err != nil {
+		r.ctx.Log.Debug(
+			"failed to reissue txs from rejected block",
+			zap.Stringer("blkID", blkID),
+		)
 	}
 
 	return nil

vms/platformvm/block/executor/rejector_test.go

@@ -138,9 +138,7 @@ func TestRejectBlock(t *testing.T) {
 			}
 
 			// Set expected calls on dependencies.
-			for _, tx := range blk.Txs() {
-				mempool.EXPECT().Add(tx).Return(nil).Times(1)
-			}
+			mempool.EXPECT().Add(blk.Txs()).Return(nil).Times(1)
 
 			require.NoError(tt.rejectFunc(rejector, blk))
 			// Make sure block and its parent are removed from the state map.

vms/platformvm/network/network.go

@@ -171,12 +171,7 @@ func (n *network) issueTx(tx *txs.Tx) error {
 		return nil
 	}
 
-	if err := n.mempool.Add(tx); err != nil {
-		n.ctx.Log.Debug("tx failed to be added to the mempool",
-			zap.Stringer("txID", txID),
-			zap.Error(err),
-		)
-
+	if err := n.mempool.Add([]*txs.Tx{tx}); err != nil {
 		n.mempool.MarkDropped(txID, err)
 		return err
 	}