filter channels by port id

neutron-org · Dec 19, 2024 · 3c6f5d6 · 3c6f5d6
1 parent 66925ca
commit 3c6f5d6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/x/tokenfactory/keeper/keeper.go b/x/tokenfactory/keeper/keeper.go
@@ -104,7 +104,7 @@ func (k Keeper) CreateModuleAccount(ctx sdk.Context) {
 func (k Keeper) GetAllIBCEscrowAccounts(ctx sdk.Context) map[string]bool {
 	escrowAddresses := make(map[string]bool)
 
-	transferChannels := k.channelKeeper.GetAllChannels(ctx)
+	transferChannels := k.channelKeeper.GetAllChannelsWithPortPrefix(ctx, transfertypes.PortID)
 	for _, channel := range transferChannels {
 		escrowAddress := transfertypes.GetEscrowAddress(channel.PortId, channel.ChannelId)
 		escrowAddresses[escrowAddress.String()] = true

diff --git a/x/tokenfactory/types/expected_keepers.go b/x/tokenfactory/types/expected_keepers.go
@@ -43,5 +43,5 @@ type ContractKeeper interface {
 
 // ChannelKeeper defines the expected IBC channel keeper
 type ChannelKeeper interface {
-	GetAllChannels(ctx sdk.Context) (channels []channeltypes.IdentifiedChannel)
+	GetAllChannelsWithPortPrefix(ctx sdk.Context, portPrefix string) (channels []channeltypes.IdentifiedChannel)
 }