refactor(x/circuit)!: Use collections, KVStoreService, and context.Context

baseapp/circuit.go

@@ -1,10 +1,8 @@
 package baseapp
 
-import (
-	sdk "github.com/cosmos/cosmos-sdk/types"
-)
+import "context"
 
 // CircuitBreaker is an interface that defines the methods for a circuit breaker.
 type CircuitBreaker interface {
-	IsAllowed(ctx sdk.Context, typeURL string) bool
+	IsAllowed(ctx context.Context, typeURL string) (bool, error)
 }

baseapp/msg_service_router.go

@@ -135,7 +135,12 @@ func (msr *MsgServiceRouter) RegisterService(sd *grpc.ServiceDesc, handler inter
 
 			if msr.circuitBreaker != nil {
 				msgURL := sdk.MsgTypeURL(msg)
-				if !msr.circuitBreaker.IsAllowed(ctx, msgURL) {
+				isAllowed, err := msr.circuitBreaker.IsAllowed(ctx, msgURL)
+				if err != nil {
+					return nil, err
+				}
+
+				if !isAllowed {
 					return nil, fmt.Errorf("circuit breaker disables execution of this message: %s", msgURL)
 				}
 			}

simapp/app.go

@@ -343,7 +343,7 @@ func NewSimApp(
 		stakingtypes.NewMultiStakingHooks(app.DistrKeeper.Hooks(), app.SlashingKeeper.Hooks()),
 	)
 
-	app.CircuitKeeper = circuitkeeper.NewKeeper(keys[circuittypes.StoreKey], authtypes.NewModuleAddress(govtypes.ModuleName).String(), app.AccountKeeper.AddressCodec())
+	app.CircuitKeeper = circuitkeeper.NewKeeper(appCodec, runtime.NewKVStoreService(keys[circuittypes.StoreKey]), authtypes.NewModuleAddress(govtypes.ModuleName).String(), app.AccountKeeper.AddressCodec())
 	app.BaseApp.SetCircuitBreaker(&app.CircuitKeeper)
 
 	app.AuthzKeeper = authzkeeper.NewKeeper(runtime.NewKVStoreService(keys[authzkeeper.StoreKey]), appCodec, app.MsgServiceRouter(), app.AccountKeeper)

x/circuit/CHANGELOG.md

@@ -29,4 +29,4 @@ Ref: https://keepachangelog.com/en/1.0.0/
 
 # Changelog
 
-## [Unreleased]
+## [Unreleased]

x/circuit/ante/circuit.go

@@ -1,13 +1,16 @@
 package ante
 
 import (
+	"context"
+
 	"github.com/cockroachdb/errors"
+
 	sdk "github.com/cosmos/cosmos-sdk/types"
 )
 
 // CircuitBreaker is an interface that defines the methods for a circuit breaker.
 type CircuitBreaker interface {
-	IsAllowed(ctx sdk.Context, typeURL string) bool
+	IsAllowed(ctx context.Context, typeURL string) (bool, error)
 }
 
 // CircuitBreakerDecorator is an AnteDecorator that checks if the transaction type is allowed to enter the mempool or be executed
@@ -24,7 +27,12 @@ func NewCircuitBreakerDecorator(ck CircuitBreaker) CircuitBreakerDecorator {
 func (cbd CircuitBreakerDecorator) AnteHandle(ctx sdk.Context, tx sdk.Tx, simulate bool, next sdk.AnteHandler) (sdk.Context, error) {
 	// loop through all the messages and check if the message type is allowed
 	for _, msg := range tx.GetMsgs() {
-		if !cbd.circuitKeeper.IsAllowed(ctx, sdk.MsgTypeURL(msg)) {
+		isAllowed, err := cbd.circuitKeeper.IsAllowed(ctx, sdk.MsgTypeURL(msg))
+		if err != nil {
+			return ctx, err
+		}
+
+		if !isAllowed {
 			return ctx, errors.New("tx type not allowed")
 		}
 	}

x/circuit/ante/circuit_test.go

@@ -1,12 +1,14 @@
 package ante_test
 
 import (
+	"context"
 	"testing"
 
 	storetypes "cosmossdk.io/store/types"
 	cbtypes "cosmossdk.io/x/circuit/types"
 	abci "github.com/cometbft/cometbft/abci/types"
 	cmproto "github.com/cometbft/cometbft/proto/tendermint/types"
+
 	"github.com/cosmos/cosmos-sdk/client"
 	"github.com/cosmos/cosmos-sdk/testutil"
 	clitestutil "github.com/cosmos/cosmos-sdk/testutil/cli"
@@ -16,12 +18,13 @@ import (
 	"github.com/cosmos/cosmos-sdk/x/bank"
 
 	"cosmossdk.io/x/circuit/ante"
-	sdk "github.com/cosmos/cosmos-sdk/types"
 	"github.com/stretchr/testify/require"
+
+	sdk "github.com/cosmos/cosmos-sdk/types"
 )
 
 type fixture struct {
-	ctx           sdk.Context
+	ctx           context.Context
 	mockStoreKey  storetypes.StoreKey
 	mockMsgURL    string
 	mockclientCtx client.Context
@@ -32,8 +35,8 @@ type MockCircuitBreaker struct {
 	isAllowed bool
 }
 
-func (m MockCircuitBreaker) IsAllowed(ctx sdk.Context, typeURL string) bool {
-	return typeURL == "/cosmos.circuit.v1.MsgAuthorizeCircuitBreaker"
+func (m MockCircuitBreaker) IsAllowed(ctx context.Context, typeURL string) (bool, error) {
+	return typeURL == "/cosmos.circuit.v1.MsgAuthorizeCircuitBreaker", nil
 }
 
 func initFixture(t *testing.T) *fixture {
@@ -78,7 +81,8 @@ func TestCircuitBreakerDecorator(t *testing.T) {
 		f.txBuilder.SetMsgs(tc.msg)
 		tx := f.txBuilder.GetTx()
 
-		_, err := decorator.AnteHandle(f.ctx, tx, false, func(ctx sdk.Context, tx sdk.Tx, simulate bool) (newCtx sdk.Context, err error) {
+		sdkCtx := sdk.UnwrapSDKContext(f.ctx)
+		_, err := decorator.AnteHandle(sdkCtx, tx, false, func(ctx sdk.Context, tx sdk.Tx, simulate bool) (newCtx sdk.Context, err error) {
 			return ctx, nil
 		})
 

x/circuit/keeper/genesis.go

@@ -1,34 +1,43 @@
 package keeper
 
 import (
+	context "context"
+
+	"cosmossdk.io/collections"
+	errorsmod "cosmossdk.io/errors"
 	"cosmossdk.io/x/circuit/types"
-	sdk "github.com/cosmos/cosmos-sdk/types"
 )
 
-func (k *Keeper) ExportGenesis(ctx sdk.Context) (data *types.GenesisState) {
+func (k *Keeper) ExportGenesis(ctx context.Context) (data *types.GenesisState) {
 	var (
 		permissions  []*types.GenesisAccountPermissions
 		disabledMsgs []string
 	)
 
-	k.IteratePermissions(ctx, func(address []byte, perm types.Permissions) (stop bool) {
+	err := k.Permissions.Walk(ctx, nil, func(address []byte, perm types.Permissions) (stop bool, err error) {
 		add, err := k.addressCodec.BytesToString(address)
 		if err != nil {
-			panic(err)
+			return true, err
 		}
 		// Convert the Permissions struct to a GenesisAccountPermissions struct
 		// and add it to the permissions slice
 		permissions = append(permissions, &types.GenesisAccountPermissions{
 			Address:     add,
 			Permissions: &perm,
 		})
-		return false
+		return false, nil
 	})
+	if err != nil && !errorsmod.IsOf(err, collections.ErrInvalidIterator) {
+		panic(err)
+	}
 
-	k.IterateDisableLists(ctx, func(address []byte, perm types.Permissions) (stop bool) {
-		disabledMsgs = append(disabledMsgs, perm.LimitTypeUrls...)
-		return false
+	err = k.DisableList.Walk(ctx, nil, func(msgUrl string) (stop bool, err error) {
+		disabledMsgs = append(disabledMsgs, msgUrl)
+		return false, nil
 	})
+	if err != nil && !errorsmod.IsOf(err, collections.ErrInvalidIterator) {
+		panic(err)
+	}
 
 	return &types.GenesisState{
 		AccountPermissions: permissions,
@@ -37,20 +46,22 @@ func (k *Keeper) ExportGenesis(ctx sdk.Context) (data *types.GenesisState) {
 }
 
 // InitGenesis initializes the circuit module's state from a given genesis state.
-func (k *Keeper) InitGenesis(ctx sdk.Context, genState *types.GenesisState) {
+func (k *Keeper) InitGenesis(ctx context.Context, genState *types.GenesisState) {
 	for _, accounts := range genState.AccountPermissions {
 		add, err := k.addressCodec.StringToBytes(accounts.Address)
 		if err != nil {
 			panic(err)
 		}
 
 		// Set the permissions for the account
-		if err := k.SetPermissions(ctx, add, accounts.Permissions); err != nil {
+		if err := k.Permissions.Set(ctx, add, *accounts.Permissions); err != nil {
 			panic(err)
 		}
 	}
 	for _, url := range genState.DisabledTypeUrls {
 		// Set the disabled type urls
-		k.DisableMsg(ctx, url)
+		if err := k.DisableList.Set(ctx, url); err != nil {
+			panic(err)
+		}
 	}
 }