add frozen when icq timeout

osmosis-labs · Oct 31, 2023 · d13a4b0 · d13a4b0
1 parent 3ef7b78
commit d13a4b0
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 8 deletions.
diff --git a/x/feeabs/ibc_module.go b/x/feeabs/ibc_module.go
@@ -209,6 +209,8 @@ func (am IBCModule) OnTimeoutPacket(
 	if err := types.ModuleCdc.UnmarshalJSON(packet.GetData(), &icqPacketData); err != nil {
 		return sdkerrors.Wrapf(errorstypes.ErrUnknownRequest, "cannot unmarshal packet data: %v", err)
 	}
+	// Frozen all hostzone
+	am.keeper.FrozenAllHostZone(ctx)
 	// Resend request if timeout
 	err := am.keeper.OnTimeoutPacket(ctx)
 	if err != nil {

diff --git a/x/feeabs/keeper/ibc.go b/x/feeabs/keeper/ibc.go
@@ -167,14 +167,7 @@ func (k Keeper) OnAcknowledgementPacket(ctx sdk.Context, ack channeltypes.Acknow
 			),
 		)
 	case *channeltypes.Acknowledgement_Error:
-		k.IterateHostZone(ctx, func(hostZoneConfig types.HostChainFeeAbsConfig) (stop bool) {
-			err := k.FrozenHostZoneByIBCDenom(ctx, hostZoneConfig.IbcDenom)
-			if err != nil {
-				k.Logger(ctx).Error(fmt.Sprintf("Failed to frozen host zone %s", err.Error()))
-			}
-
-			return false
-		})
+		k.FrozenAllHostZone(ctx)
 		k.Logger(ctx).Error(fmt.Sprintf("failed to send packet ICQ request %v", resp.Error))
 
 		ctx.EventManager().EmitEvent(
@@ -187,6 +180,18 @@ func (k Keeper) OnAcknowledgementPacket(ctx sdk.Context, ack channeltypes.Acknow
 	return nil
 }
 
+// FrozenAllHostZone resend packet when timeout
+func (k Keeper) FrozenAllHostZone(ctx sdk.Context) {
+	k.IterateHostZone(ctx, func(hostZoneConfig types.HostChainFeeAbsConfig) (stop bool) {
+		err := k.FrozenHostZoneByIBCDenom(ctx, hostZoneConfig.IbcDenom)
+		if err != nil {
+			k.Logger(ctx).Error(fmt.Sprintf("Failed to frozen host zone %s", err.Error()))
+		}
+
+		return false
+	})
+}
+
 // OnTimeoutPacket resend packet when timeout
 func (k Keeper) OnTimeoutPacket(ctx sdk.Context) error {
 	return k.handleOsmosisIbcQuery(ctx)