Skip to content

Commit

Permalink
init
Browse files Browse the repository at this point in the history
  • Loading branch information
davidterpay committed Jun 24, 2024
1 parent bfdd584 commit c713a10
Show file tree
Hide file tree
Showing 5 changed files with 62 additions and 18 deletions.
50 changes: 39 additions & 11 deletions abci/abci.go
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@ import (
abci "github.com/cometbft/cometbft/abci/types"
sdk "github.com/cosmos/cosmos-sdk/types"

"github.com/cosmos/cosmos-sdk/baseapp"
"github.com/skip-mev/block-sdk/v2/block"
"github.com/skip-mev/block-sdk/v2/block/proposals"
"github.com/skip-mev/block-sdk/v2/block/utils"
Expand All @@ -16,26 +17,49 @@ type (
// ProposalHandler is a wrapper around the ABCI++ PrepareProposal and ProcessProposal
// handlers.
ProposalHandler struct {
logger log.Logger
txDecoder sdk.TxDecoder
txEncoder sdk.TxEncoder
mempool block.Mempool
logger log.Logger
txDecoder sdk.TxDecoder
txEncoder sdk.TxEncoder
mempool block.Mempool
useCustomProcessProposal bool
}
)

// NewProposalHandler returns a new ABCI++ proposal handler. This proposal handler will
// iteratively call each of the lanes in the chain to prepare and process the proposal.
func NewProposalHandler(
// NewDefaultProposalHandler returns a new ABCI++ proposal handler. This proposal handler will
// iteratively call each of the lanes in the chain to prepare and process the proposal. This
// will not use custom process proposal logic.
func NewDefaultProposalHandler(
logger log.Logger,
txDecoder sdk.TxDecoder,
txEncoder sdk.TxEncoder,
mempool block.Mempool,
) *ProposalHandler {
return &ProposalHandler{
logger: logger,
txDecoder: txDecoder,
txEncoder: txEncoder,
mempool: mempool,
logger: logger,
txDecoder: txDecoder,
txEncoder: txEncoder,
mempool: mempool,
useCustomProcessProposal: false,
}

Check warning on line 43 in abci/abci.go

View check run for this annotation

Codecov / codecov/patch

abci/abci.go#L38-L43

Added lines #L38 - L43 were not covered by tests
}

// New returns a new ABCI++ proposal handler with the ability to use custom process proposal logic.
//
// NOTE: It is highly recommended to use the default proposal handler unless you have a specific
// use case that requires custom process proposal logic.
func New(
logger log.Logger,
txDecoder sdk.TxDecoder,
txEncoder sdk.TxEncoder,
mempool block.Mempool,
useCustomProcessProposal bool,
) *ProposalHandler {
return &ProposalHandler{
logger: logger,
txDecoder: txDecoder,
txEncoder: txEncoder,
mempool: mempool,
useCustomProcessProposal: useCustomProcessProposal,
}
}

Expand Down Expand Up @@ -115,6 +139,10 @@ func (h *ProposalHandler) PrepareProposalHandler() sdk.PrepareProposalHandler {
// verify all transactions in the proposal that belong to the lane and pass any remaining transactions
// to the next lane in the chain.
func (h *ProposalHandler) ProcessProposalHandler() sdk.ProcessProposalHandler {
if !h.useCustomProcessProposal {
return baseapp.NoOpProcessProposal()
}

Check warning on line 144 in abci/abci.go

View check run for this annotation

Codecov / codecov/patch

abci/abci.go#L144

Added line #L144 was not covered by tests

return func(ctx sdk.Context, req *abci.RequestProcessProposal) (resp *abci.ResponseProcessProposal, err error) {
if req.Height <= 1 {
return &abci.ResponseProcessProposal{Status: abci.ResponseProcessProposal_ACCEPT}, nil
Expand Down
12 changes: 8 additions & 4 deletions abci/abci_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -546,11 +546,12 @@ func (s *ProposalsTestSuite) TestPrepareProposalEdgeCases() {
)
s.Require().NoError(err)

proposalHandler := abci.NewProposalHandler(
proposalHandler := abci.New(
log.NewNopLogger(),
s.encodingConfig.TxConfig.TxDecoder(),
s.encodingConfig.TxConfig.TxEncoder(),
mempool,
true,
).PrepareProposalHandler()

maxTxBytes := s.ctx.ConsensusParams().Block.MaxBytes
Expand Down Expand Up @@ -611,11 +612,12 @@ func (s *ProposalsTestSuite) TestPrepareProposalEdgeCases() {
)
s.Require().NoError(err)

proposalHandler := abci.NewProposalHandler(
proposalHandler := abci.New(
log.NewNopLogger(),
s.encodingConfig.TxConfig.TxDecoder(),
s.encodingConfig.TxConfig.TxEncoder(),
mempool,
true,
).PrepareProposalHandler()

maxTxBytes := s.ctx.ConsensusParams().Block.MaxBytes
Expand Down Expand Up @@ -683,11 +685,12 @@ func (s *ProposalsTestSuite) TestPrepareProposalEdgeCases() {
)
s.Require().NoError(err)

proposalHandler := abci.NewProposalHandler(
proposalHandler := abci.New(
log.NewNopLogger(),
s.encodingConfig.TxConfig.TxDecoder(),
s.encodingConfig.TxConfig.TxEncoder(),
mempool,
true,
).PrepareProposalHandler()

maxTxBytes := s.ctx.ConsensusParams().Block.MaxBytes
Expand Down Expand Up @@ -755,11 +758,12 @@ func (s *ProposalsTestSuite) TestPrepareProposalEdgeCases() {
)
s.Require().NoError(err)

proposalHandler := abci.NewProposalHandler(
proposalHandler := abci.New(
log.NewNopLogger(),
s.encodingConfig.TxConfig.TxDecoder(),
s.encodingConfig.TxConfig.TxEncoder(),
mempool,
true,
).PrepareProposalHandler()

maxTxBytes := s.ctx.ConsensusParams().Block.MaxBytes
Expand Down
3 changes: 2 additions & 1 deletion abci/utils_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -171,11 +171,12 @@ func (s *ProposalsTestSuite) setUpProposalHandlers(lanes []block.Lane) *abci.Pro
)
s.Require().NoError(err)

return abci.NewProposalHandler(
return abci.New(
log.NewNopLogger(),
s.encodingConfig.TxConfig.TxDecoder(),
s.encodingConfig.TxConfig.TxEncoder(),
mempool,
true,
)
}

Expand Down
4 changes: 3 additions & 1 deletion tests/app/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,8 @@ There are fix critical steps to building a test app that uses the Block SDK:
5. Setting the antehandlers - used for transaction validation - for each lane.
6. Setting the proposal handlers - used for block creation and verification - for the application to utilize the Block SDK's Prepare and Process Proposal handlers.

**IMPORTANT NOTE:** It is recommended that applications use the **`NewDefaultProposalHandler`** when constructing their Prepare and Process Proposal handlers. This is because the priority nonce mempool - the underlying storage device of transactions - has non-deterministic ordering of transactions on chains with multiple fees. The use of the `NewDefaultProposalHandler` ensures that the ordering of transactions on proposal construction follows the ordering logic of the lanes, which is deterministic, and that process proposal optimistically assumes that the ordering of transactions is correct.

### 1. Signer Extractor

The signer extractor is responsible for extracting signers and relevant information about who is signing the transaction. We recommend using the default implementation provided by the Block SDK.
Expand All @@ -29,7 +31,7 @@ signerAdapter := signerextraction.NewDefaultAdapter()

### 2. Lane Configurations

This controls how many transactions can be stored by each lane, how much block space is allocated to each lane, how to extract transacation information such as signers, fees, and more. Each lane should have a separate `LaneConfig` object.
This controls how many transactions can be stored by each lane, how much block space is allocated to each lane, how to extract transacation information such as signers, fees, and more. Each lane should have a separate `LaneConfig` object.`

For example, in [`lanes.go`](./lanes.go) we see the following:

Expand Down
11 changes: 10 additions & 1 deletion tests/app/app.go
Original file line number Diff line number Diff line change
Expand Up @@ -266,11 +266,20 @@ func New(

// Step 6: Create the proposal handler and set it on the app. Now the application
// will build and verify proposals using the Block SDK!
proposalHandler := abci.NewProposalHandler(
//
// NOTE: It is generally recommended to use the default proposal handler by constructing
// using the NewDefaultProposalHandler function. This will use the correct prepare logic
// for the lanes, but the process logic will be a no-op. To read more about the default
// proposal handler, see the documentation in readme.md in this directory.
//
// If you want to customize the prepare and process logic, you can construct the proposal
// handler using the New function and setting the useProcess flag to true.
proposalHandler := abci.New( // use NewDefaultProposalHandler instead for default behavior (RECOMMENDED)
app.Logger(),
app.TxConfig().TxDecoder(),
app.TxConfig().TxEncoder(),
mempool,
true,
)
app.App.SetPrepareProposal(proposalHandler.PrepareProposalHandler())
app.App.SetProcessProposal(proposalHandler.ProcessProposalHandler())
Expand Down

0 comments on commit c713a10

Please sign in to comment.