check limit status when migrate bucket

bnb-chain · May 23, 2024 · fe94788 · fe94788
1 parent 3ad0d51
commit fe94788
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/x/storage/keeper/payment.go b/x/storage/keeper/payment.go
@@ -602,6 +602,11 @@ func (k Keeper) ChargeBucketReadStoreFee(ctx sdk.Context, bucketInfo *storagetyp
 	}
 
 	if ctx.IsUpgraded(upgradetypes.Erdos) {
+		isRateLimited := k.IsBucketRateLimited(ctx, bucketInfo.BucketName)
+		if isRateLimited {
+			return fmt.Errorf("bucket is rate limited: %s", bucketInfo.BucketName)
+		}
+
 		err := k.isBucketFlowRateUnderLimit(ctx, sdk.MustAccAddressFromHex(bucketInfo.PaymentAddress), sdk.MustAccAddressFromHex(bucketInfo.Owner), bucketInfo.BucketName, bill)
 		if err != nil {
 			return err