feedback

x/bank/v2/depinject.go

@@ -98,7 +98,7 @@ func InvokeSetSendRestrictions(
 			return fmt.Errorf("can't find send restriction for module %s", module)
 		}
 
-		keeper.AppendSendRestriction(restriction)
+		keeper.AppendGlobalSendRestriction(restriction)
 	}
 
 	return nil

x/bank/v2/keeper/keeper.go

@@ -258,18 +258,3 @@ func newBalancesIndexes(sb *collections.SchemaBuilder) BalancesIndexes {
 type BalancesIndexes struct {
 	Denom *indexes.ReversePair[[]byte, string, math.Int]
 }
-
-// AppendSendRestriction adds the provided SendRestrictionFn to run after previously provided restrictions.
-func (k Keeper) AppendSendRestriction(restriction types.SendRestrictionFn) {
-	k.sendRestriction.append(restriction)
-}
-
-// PrependSendRestriction adds the provided SendRestrictionFn to run before previously provided restrictions.
-func (k Keeper) PrependSendRestriction(restriction types.SendRestrictionFn) {
-	k.sendRestriction.prepend(restriction)
-}
-
-// ClearSendRestriction removes the send restriction (if there is one).
-func (k Keeper) ClearSendRestriction() {
-	k.sendRestriction.clear()
-}

x/bank/v2/keeper/keeper_test.go

@@ -202,7 +202,7 @@ func (suite *KeeperTestSuite) TestSendCoins_WithRestriction() {
 		}
 		return to, nil
 	}
-	suite.bankKeeper.AppendSendRestriction(addrRestrictFunc)
+	suite.bankKeeper.AppendGlobalSendRestriction(addrRestrictFunc)
 
 	err := suite.bankKeeper.SendCoins(ctx, accAddrs[0], accAddrs[0], sendAmt)
 	require.Error(err)
@@ -215,7 +215,7 @@ func (suite *KeeperTestSuite) TestSendCoins_WithRestriction() {
 		}
 		return to, nil
 	}
-	suite.bankKeeper.AppendSendRestriction(amtRestrictFunc)
+	suite.bankKeeper.AppendGlobalSendRestriction(amtRestrictFunc)
 
 	// Pass the 1st but failt at the 2nd
 	err = suite.bankKeeper.SendCoins(ctx, accAddrs[0], accAddrs[1], sendAmt)

x/bank/v2/keeper/restriction.go

@@ -45,3 +45,18 @@ func (r *sendRestriction) apply(ctx context.Context, fromAddr, toAddr []byte, am
 	}
 	return r.fn(ctx, fromAddr, toAddr, amt)
 }
+
+// AppendSendRestriction adds the provided SendRestrictionFn to run after previously provided restrictions.
+func (k Keeper) AppendGlobalSendRestriction(restriction types.SendRestrictionFn) {
+	k.sendRestriction.append(restriction)
+}
+
+// PrependSendRestriction adds the provided SendRestrictionFn to run before previously provided restrictions.
+func (k Keeper) PrependGlobalSendRestriction(restriction types.SendRestrictionFn) {
+	k.sendRestriction.prepend(restriction)
+}
+
+// ClearSendRestriction removes the send restriction (if there is one).
+func (k Keeper) ClearGlobalSendRestriction() {
+	k.sendRestriction.clear()
+}