From 664a6a1121af386f64a80427534f5894f20d0a04 Mon Sep 17 00:00:00 2001 From: yutianwu Date: Fri, 15 Mar 2024 16:20:26 +0800 Subject: [PATCH] fix: add upgrade check when deleting rate limit status --- x/storage/keeper/keeper.go | 21 ++++++++++++--------- 1 file changed, 12 insertions(+), 9 deletions(-) diff --git a/x/storage/keeper/keeper.go b/x/storage/keeper/keeper.go index 9fc41affa..b1dd90f79 100644 --- a/x/storage/keeper/keeper.go +++ b/x/storage/keeper/keeper.go @@ -6,6 +6,14 @@ import ( "cosmossdk.io/errors" sdkmath "cosmossdk.io/math" + "github.com/cometbft/cometbft/libs/log" + "github.com/cosmos/cosmos-sdk/codec" + "github.com/cosmos/cosmos-sdk/store/prefix" + storetypes "github.com/cosmos/cosmos-sdk/store/types" + sdk "github.com/cosmos/cosmos-sdk/types" + upgradetypes "github.com/cosmos/cosmos-sdk/x/upgrade/types" + "github.com/cosmos/gogoproto/proto" + "github.com/bnb-chain/greenfield/internal/sequence" gnfdtypes "github.com/bnb-chain/greenfield/types" types2 "github.com/bnb-chain/greenfield/types" @@ -18,13 +26,6 @@ import ( sptypes "github.com/bnb-chain/greenfield/x/sp/types" "github.com/bnb-chain/greenfield/x/storage/types" virtualgroupmoduletypes "github.com/bnb-chain/greenfield/x/virtualgroup/types" - "github.com/cometbft/cometbft/libs/log" - "github.com/cosmos/cosmos-sdk/codec" - "github.com/cosmos/cosmos-sdk/store/prefix" - storetypes "github.com/cosmos/cosmos-sdk/store/types" - sdk "github.com/cosmos/cosmos-sdk/types" - upgradetypes "github.com/cosmos/cosmos-sdk/x/upgrade/types" - "github.com/cosmos/gogoproto/proto" ) type ( @@ -269,8 +270,10 @@ func (k Keeper) doDeleteBucket(ctx sdk.Context, operator sdk.AccAddress, bucketI } } - // delete bucket flow rate limit status - k.deleteBucketFlowRateLimitStatus(ctx, bucketInfo.BucketName) + if ctx.IsUpgraded(upgradetypes.Serengeti) { + // delete bucket flow rate limit status + k.deleteBucketFlowRateLimitStatus(ctx, bucketInfo.BucketName) + } return nil }