Skip to content

Commit ec71de0

Browse files
committed
move reasons to consts
1 parent 5bb5ad1 commit ec71de0

File tree

3 files changed

+16
-6
lines changed

3 files changed

+16
-6
lines changed

baseapp/abci_utils.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -379,7 +379,7 @@ func (h *DefaultProposalHandler) PrepareProposalHandler() sdk.PrepareProposalHan
379379

380380
for _, invalidTx := range invalidTxs {
381381
reason := mempool.RemoveReason{
382-
Caller: "prepare_proposal.remove_invalid_tx",
382+
Caller: mempool.CallerPrepareProposalRemoveInvalid,
383383
Error: invalidTx.err,
384384
}
385385

baseapp/baseapp.go

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -882,7 +882,7 @@ func (app *BaseApp) RunTx(mode sdk.ExecMode, txBytes []byte, tx sdk.Tx, txIndex
882882
if mode == execModeReCheck {
883883
// if the ante handler fails on recheck, we want to remove the tx from the mempool
884884
errMempool := mempool.RemoveWithReason(ctx, app.mempool, tx, mempool.RemoveReason{
885-
Caller: "run_tx.recheck",
885+
Caller: mempool.CallerRunTxRecheck,
886886
Error: err,
887887
})
888888

@@ -905,7 +905,8 @@ func (app *BaseApp) RunTx(mode sdk.ExecMode, txBytes []byte, tx sdk.Tx, txIndex
905905
return gInfo, nil, anteEvents, err
906906
}
907907
case execModeFinalize:
908-
err = mempool.RemoveWithReason(ctx, app.mempool, tx, mempool.RemoveReason{Caller: "run_tx.finalize"})
908+
reason := mempool.RemoveReason{Caller: mempool.CallerRunTxFinalize}
909+
err = mempool.RemoveWithReason(ctx, app.mempool, tx, reason)
909910
if err != nil && !errors.Is(err, mempool.ErrTxNotFound) {
910911
return gInfo, nil, anteEvents, fmt.Errorf("failed to remove tx from mempool: %w", err)
911912
}

types/mempool/mempool.go

Lines changed: 12 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -38,12 +38,21 @@ type ExtMempool interface {
3838
RemoveWithReason(context.Context, sdk.Tx, RemoveReason) error
3939
}
4040

41+
// RemovalCaller is the origin of the removal
42+
type RemovalCaller string
43+
44+
// Various callers
45+
const (
46+
CallerRunTxRecheck RemovalCaller = "run_tx.recheck"
47+
CallerRunTxFinalize RemovalCaller = "run_tx.finalize"
48+
CallerPrepareProposalRemoveInvalid RemovalCaller = "prepare_proposal.remove_invalid"
49+
)
50+
4151
// RemoveReason is the reason for removing a transaction from the mempool.
4252
type RemoveReason struct {
43-
// Origin is the origin of the removal (eg. baseapp, cometbft, etc.)
44-
Caller string
53+
Caller RemovalCaller
4554

46-
// Error is an optional error that caused the removal.
55+
// Error is an optional error that caused the removal.
4756
Error error
4857
}
4958

0 commit comments

Comments
 (0)