64 lines
1.9 KiB
Go
64 lines
1.9 KiB
Go
package middleware
|
|
|
|
import (
|
|
"context"
|
|
|
|
abci "github.com/tendermint/tendermint/abci/types"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/cosmos/cosmos-sdk/types/tx"
|
|
)
|
|
|
|
type indexEventsTxHandler struct {
|
|
// indexEvents defines the set of events in the form {eventType}.{attributeKey},
|
|
// which informs Tendermint what to index. If empty, all events will be indexed.
|
|
indexEvents map[string]struct{}
|
|
inner tx.Handler
|
|
}
|
|
|
|
// NewIndexEventsTxMiddleware defines a middleware to optionally only index a
|
|
// subset of the emitted events inside the Tendermint events indexer.
|
|
func NewIndexEventsTxMiddleware(indexEvents map[string]struct{}) tx.Middleware {
|
|
return func(txHandler tx.Handler) tx.Handler {
|
|
return indexEventsTxHandler{
|
|
indexEvents: indexEvents,
|
|
inner: txHandler,
|
|
}
|
|
}
|
|
}
|
|
|
|
var _ tx.Handler = indexEventsTxHandler{}
|
|
|
|
// CheckTx implements tx.Handler.CheckTx method.
|
|
func (txh indexEventsTxHandler) CheckTx(ctx context.Context, tx sdk.Tx, req abci.RequestCheckTx) (abci.ResponseCheckTx, error) {
|
|
res, err := txh.inner.CheckTx(ctx, tx, req)
|
|
if err != nil {
|
|
return res, err
|
|
}
|
|
|
|
res.Events = sdk.MarkEventsToIndex(res.Events, txh.indexEvents)
|
|
return res, nil
|
|
}
|
|
|
|
// DeliverTx implements tx.Handler.DeliverTx method.
|
|
func (txh indexEventsTxHandler) DeliverTx(ctx context.Context, tx sdk.Tx, req abci.RequestDeliverTx) (abci.ResponseDeliverTx, error) {
|
|
res, err := txh.inner.DeliverTx(ctx, tx, req)
|
|
if err != nil {
|
|
return res, err
|
|
}
|
|
|
|
res.Events = sdk.MarkEventsToIndex(res.Events, txh.indexEvents)
|
|
return res, nil
|
|
}
|
|
|
|
// SimulateTx implements tx.Handler.SimulateTx method.
|
|
func (txh indexEventsTxHandler) SimulateTx(ctx context.Context, tx sdk.Tx, req tx.RequestSimulateTx) (tx.ResponseSimulateTx, error) {
|
|
res, err := txh.inner.SimulateTx(ctx, tx, req)
|
|
if err != nil {
|
|
return res, err
|
|
}
|
|
|
|
res.Result.Events = sdk.MarkEventsToIndex(res.Result.Events, txh.indexEvents)
|
|
return res, nil
|
|
}
|