Merge pull request #27 from balancer/add-pool-states

mendesfabio · web-flow · commit d36335c9de67 · 2025-03-10T08:57:10.000-03:00
fix dynamic swap fee rounding, add pool pause and recovery mode
diff --git a/subgraphs/v3-vault/schema.graphql b/subgraphs/v3-vault/schema.graphql
@@ -48,6 +48,11 @@ type Pool @entity {
   transactionHash: Bytes!
   "Indicates whether the Pool has been initialized"
   isInitialized: Boolean!
+  "Indicates whether the Pool is currently paused"
+  isPaused: Boolean!
+  "Indicates whether the Pool is in recovery mode"
+  isInRecoveryMode: Boolean!
+
 
   "Account empowered to pause/unpause the pool"
   pauseManager: Bytes!
diff --git a/subgraphs/v3-vault/src/helpers/math.ts b/subgraphs/v3-vault/src/helpers/math.ts
@@ -9,7 +9,7 @@ export function mulDown(a: BigInt, b: BigInt): BigInt {
   return product.div(ONE);
 }
 
-export function computeAggregateSwapFee(
+export function mulDownSwapFee(
   swapFeeAmountRaw: BigInt,
   swapFeePercentage: BigDecimal
 ): BigInt {
diff --git a/subgraphs/v3-vault/src/mappings/vault.ts b/subgraphs/v3-vault/src/mappings/vault.ts
@@ -6,8 +6,11 @@ import {
   LiquidityAddedToBuffer,
   LiquidityRemoved,
   LiquidityRemovedFromBuffer,
+  PoolPausedStateChanged,
+  PoolRecoveryModeStateChanged,
   PoolRegistered,
   Swap as SwapEvent,
+  SwapFeePercentageChanged,
   Unwrap,
   Wrap,
 } from "../types/Vault/Vault";
@@ -37,7 +40,7 @@ import { BPT } from "../types/templates";
 import { ERC20 } from "../types/Vault/ERC20";
 import { VaultExtension } from "../types/Vault/VaultExtension";
 import { ERC4626 } from "../types/Vault/ERC4626";
-import { computeAggregateSwapFee } from "../helpers/math";
+import { mulDownSwapFee } from "../helpers/math";
 
 /************************************
  ******* POOLS REGISTRATIONS ********
@@ -54,6 +57,8 @@ export function handlePoolRegistered(event: PoolRegistered): void {
   pool.pauseWindowEndTime = event.params.pauseWindowEndTime;
   pool.totalShares = ZERO_BD;
   pool.isInitialized = false;
+  pool.isInRecoveryMode = false;
+  pool.isPaused = false;
   pool.swapsCount = ZERO_BI;
   pool.holdersCount = ZERO_BI;
   pool.protocolSwapFee = vault.protocolSwapFee;
@@ -189,10 +194,7 @@ export function handleLiquidityAdded(event: LiquidityAdded): void {
     );
 
     let aggregateSwapFeeAmount = scaleDown(
-      computeAggregateSwapFee(
-        event.params.swapFeeAmountsRaw[i],
-        pool.protocolSwapFee
-      ),
+      mulDownSwapFee(event.params.swapFeeAmountsRaw[i], pool.protocolSwapFee),
       poolToken.decimals
     );
 
@@ -259,10 +261,7 @@ export function handleLiquidityRemoved(event: LiquidityRemoved): void {
     );
 
     let aggregateSwapFeeAmount = scaleDown(
-      computeAggregateSwapFee(
-        event.params.swapFeeAmountsRaw[i],
-        pool.protocolSwapFee
-      ),
+      mulDownSwapFee(event.params.swapFeeAmountsRaw[i], pool.protocolSwapFee),
       poolToken.decimals
     );
 
@@ -331,8 +330,14 @@ export function handleSwap(event: SwapEvent): void {
   let swapFeeDeltaAmount = ZERO_BD;
   if (hasDynamicSwapFee) {
     let swapFeeDelta = swapFeePercentage.minus(pool.swapFee);
-    swapFeeBaseAmount = tokenAmountIn.times(pool.swapFee);
-    swapFeeDeltaAmount = tokenAmountIn.times(swapFeeDelta);
+    swapFeeBaseAmount = scaleDown(
+      mulDownSwapFee(event.params.amountIn, pool.swapFee),
+      tokenIn.decimals
+    );
+    swapFeeDeltaAmount = scaleDown(
+      mulDownSwapFee(event.params.amountIn, swapFeeDelta),
+      tokenIn.decimals
+    );
   }
 
   swap.pool = event.params.pool;
@@ -369,7 +374,7 @@ export function handleSwap(event: SwapEvent): void {
   }
 
   let aggregateSwapFeeAmount = scaleDown(
-    computeAggregateSwapFee(event.params.swapFeeAmount, pool.protocolSwapFee),
+    mulDownSwapFee(event.params.swapFeeAmount, pool.protocolSwapFee),
     poolTokenIn.decimals
   );
 
@@ -594,3 +599,31 @@ export function handleWrap(event: Wrap): void {
   );
   buffer.save();
 }
+
+/************************************
+ ********** POOLS STATE *************
+ ************************************/
+
+export function handleSwapFeePercentageChanged(
+  event: SwapFeePercentageChanged
+): void {
+  let pool = Pool.load(event.params.pool) as Pool;
+  pool.swapFee = scaleDown(event.params.swapFeePercentage, 18);
+  pool.save();
+}
+
+export function handlePoolRecoveryModeStateChanged(
+  event: PoolRecoveryModeStateChanged
+): void {
+  let pool = Pool.load(event.params.pool) as Pool;
+  pool.isInRecoveryMode = event.params.recoveryMode;
+  pool.save();
+}
+
+export function handlePoolPausedStateChanged(
+  event: PoolPausedStateChanged
+): void {
+  let pool = Pool.load(event.params.pool) as Pool;
+  pool.isPaused = event.params.paused;
+  pool.save();
+}
diff --git a/subgraphs/v3-vault/subgraph.arbitrum-one.yaml b/subgraphs/v3-vault/subgraph.arbitrum-one.yaml
@@ -44,6 +44,12 @@ dataSources:
           handler: handleWrap
         - event: Unwrap(indexed address,uint256,uint256,bytes32)
           handler: handleUnwrap
+        - event: SwapFeePercentageChanged(indexed address,uint256)
+          handler: handleSwapFeePercentageChanged
+        - event: PoolRecoveryModeStateChanged(indexed address,bool)
+          handler: handlePoolRecoveryModeStateChanged
+        - event: PoolPausedStateChanged(indexed address,bool)
+          handler: handlePoolPausedStateChanged
       file: ./src/mappings/vault.ts
   - kind: ethereum
     name: ProtocolFeeController
diff --git a/subgraphs/v3-vault/subgraph.base.yaml b/subgraphs/v3-vault/subgraph.base.yaml
@@ -44,6 +44,12 @@ dataSources:
           handler: handleWrap
         - event: Unwrap(indexed address,uint256,uint256,bytes32)
           handler: handleUnwrap
+        - event: SwapFeePercentageChanged(indexed address,uint256)
+          handler: handleSwapFeePercentageChanged
+        - event: PoolRecoveryModeStateChanged(indexed address,bool)
+          handler: handlePoolRecoveryModeStateChanged
+        - event: PoolPausedStateChanged(indexed address,bool)
+          handler: handlePoolPausedStateChanged
       file: ./src/mappings/vault.ts
   - kind: ethereum
     name: ProtocolFeeController
diff --git a/subgraphs/v3-vault/subgraph.gnosis.yaml b/subgraphs/v3-vault/subgraph.gnosis.yaml
@@ -44,6 +44,12 @@ dataSources:
           handler: handleWrap
         - event: Unwrap(indexed address,uint256,uint256,bytes32)
           handler: handleUnwrap
+        - event: SwapFeePercentageChanged(indexed address,uint256)
+          handler: handleSwapFeePercentageChanged
+        - event: PoolRecoveryModeStateChanged(indexed address,bool)
+          handler: handlePoolRecoveryModeStateChanged
+        - event: PoolPausedStateChanged(indexed address,bool)
+          handler: handlePoolPausedStateChanged
       file: ./src/mappings/vault.ts
   - kind: ethereum
     name: ProtocolFeeController
diff --git a/subgraphs/v3-vault/subgraph.sepolia.yaml b/subgraphs/v3-vault/subgraph.sepolia.yaml
@@ -44,6 +44,12 @@ dataSources:
           handler: handleWrap
         - event: Unwrap(indexed address,uint256,uint256,bytes32)
           handler: handleUnwrap
+        - event: SwapFeePercentageChanged(indexed address,uint256)
+          handler: handleSwapFeePercentageChanged
+        - event: PoolRecoveryModeStateChanged(indexed address,bool)
+          handler: handlePoolRecoveryModeStateChanged
+        - event: PoolPausedStateChanged(indexed address,bool)
+          handler: handlePoolPausedStateChanged
       file: ./src/mappings/vault.ts
   - kind: ethereum
     name: ProtocolFeeController
diff --git a/subgraphs/v3-vault/subgraph.sonic.yaml b/subgraphs/v3-vault/subgraph.sonic.yaml
@@ -44,6 +44,12 @@ dataSources:
           handler: handleWrap
         - event: Unwrap(indexed address,uint256,uint256,bytes32)
           handler: handleUnwrap
+        - event: SwapFeePercentageChanged(indexed address,uint256)
+          handler: handleSwapFeePercentageChanged
+        - event: PoolRecoveryModeStateChanged(indexed address,bool)
+          handler: handlePoolRecoveryModeStateChanged
+        - event: PoolPausedStateChanged(indexed address,bool)
+          handler: handlePoolPausedStateChanged
       file: ./src/mappings/vault.ts
   - kind: ethereum
     name: ProtocolFeeController
diff --git a/subgraphs/v3-vault/subgraph.yaml b/subgraphs/v3-vault/subgraph.yaml
@@ -44,6 +44,12 @@ dataSources:
           handler: handleWrap
         - event: Unwrap(indexed address,uint256,uint256,bytes32)
           handler: handleUnwrap
+        - event: SwapFeePercentageChanged(indexed address,uint256)
+          handler: handleSwapFeePercentageChanged
+        - event: PoolRecoveryModeStateChanged(indexed address,bool)
+          handler: handlePoolRecoveryModeStateChanged
+        - event: PoolPausedStateChanged(indexed address,bool)
+          handler: handlePoolPausedStateChanged
       file: ./src/mappings/vault.ts
   - kind: ethereum
     name: ProtocolFeeController
diff --git a/subgraphs/v3-vault/template.yaml b/subgraphs/v3-vault/template.yaml
@@ -44,6 +44,12 @@ dataSources:
           handler: handleWrap
         - event: Unwrap(indexed address,uint256,uint256,bytes32)
           handler: handleUnwrap
+        - event: SwapFeePercentageChanged(indexed address,uint256)
+          handler: handleSwapFeePercentageChanged
+        - event: PoolRecoveryModeStateChanged(indexed address,bool)
+          handler: handlePoolRecoveryModeStateChanged
+        - event: PoolPausedStateChanged(indexed address,bool)
+          handler: handlePoolPausedStateChanged
       file: ./src/mappings/vault.ts
 
   - kind: ethereum

Original file line number	Diff line number	Diff line change
`@@ -9,7 +9,7 @@ export function mulDown(a: BigInt, b: BigInt): BigInt {`
`9`	`9`	`return product.div(ONE);`
`10`	`10`	`}`
`11`	`11`
`12`		`-export function computeAggregateSwapFee(`
	`12`	`+export function mulDownSwapFee(`
`13`	`13`	`swapFeeAmountRaw: BigInt,`
`14`	`14`	`swapFeePercentage: BigDecimal`
`15`	`15`	`): BigInt {`