Add event for swept funds. (#2307)

dydxprotocol · Sep 20, 2024 · e4c84f0 · e4c84f0
1 parent 5a02618
commit e4c84f0
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/protocol/x/vault/keeper/sweep_funds.go b/protocol/x/vault/keeper/sweep_funds.go
@@ -45,4 +45,10 @@ func (k Keeper) SweepMainVaultBankBalance(
 		)
 		return
 	}
+
+	ctx.EventManager().EmitEvent(
+		types.NewSweepToMegavaultEvent(
+			mainVaultBalance.Amount.BigInt().Uint64(),
+		),
+	)
 }
diff --git a/protocol/x/vault/types/events.go b/protocol/x/vault/types/events.go
@@ -19,6 +19,9 @@ const (
 	AttributeKeyTotalShares           = "total_shares"
 	AttributeKeyMegavaultEquity       = "megavault_equity"
 	AttributeKeyRedeemedQuoteQuantums = "redeemed_quote_quantums"
+
+	EventTypeSweepToMegavault      = "sweep_to_megavault"
+	AttributeKeySweptQuoteQuantums = "swept_quote_quantums"
 )
 
 // NewDepositToMegavaultEvent constructs a new deposit_to_megavault sdk.Event.
@@ -52,3 +55,12 @@ func NewWithdrawFromMegavaultEvent(
 		sdk.NewAttribute(AttributeKeyRedeemedQuoteQuantums, fmt.Sprintf("%d", redeemedQuoteQuantums)),
 	)
 }
+
+func NewSweepToMegavaultEvent(
+	quoteQuantums uint64,
+) sdk.Event {
+	return sdk.NewEvent(
+		EventTypeSweepToMegavault,
+		sdk.NewAttribute(AttributeKeySweptQuoteQuantums, fmt.Sprintf("%d", quoteQuantums)),
+	)
+}