mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-12-30 18:25:19 +00:00
38 lines
1.0 KiB
Go
38 lines
1.0 KiB
Go
package ante
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/kava-labs/kava/x/circuit-breaker/keeper"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
)
|
|
|
|
// CircuitBreakerDecorator needs to be combined with other standard decorators (from auth) to create the app's AnteHandler.
|
|
|
|
// CircuitBreakerDecorator errors if a tx contains a disallowed msg type
|
|
// Call next AnteHandler if all msgs are allowed
|
|
type CircuitBreakerDecorator struct {
|
|
cbk keeper.Keeper
|
|
}
|
|
|
|
func NewCircuitBreakerDecorator(cbk keeper.Keeper) CircuitBreakerDecorator {
|
|
return CircuitBreakerDecorator{
|
|
cbk: cbk,
|
|
}
|
|
}
|
|
|
|
func (cbd CircuitBreakerDecorator) AnteHandle(ctx sdk.Context, tx sdk.Tx, simulate bool, next sdk.AnteHandler) (newCtx sdk.Context, err error) {
|
|
|
|
// get msg route, error if not allowed
|
|
disallowedRoutes := cbd.cbk.GetMsgRoutes(ctx)
|
|
for _, m := range tx.GetMsgs() {
|
|
for _, r := range disallowedRoutes {
|
|
if r.Route == m.Route() && r.Msg == m.Type() {
|
|
return ctx, fmt.Errorf("route %s has been circuit broken, tx rejected", r)
|
|
}
|
|
}
|
|
}
|
|
return next(ctx, tx, simulate)
|
|
}
|