fix: Positions update in position manager

pancakeswap · Jun 19, 2023 · a863001 · a863001
1 parent 85e148c
commit a863001
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 7 deletions.
diff --git a/subgraphs/user-position-v3/template/mappings/pool.ts b/subgraphs/user-position-v3/template/mappings/pool.ts
@@ -39,9 +39,11 @@ export function handleMint(event: MintEvent): void {
   transaction.tickLower = BigInt.fromI32(event.params.tickLower);
   transaction.tickUpper = BigInt.fromI32(event.params.tickUpper);
   transaction.increaseLiquidityAmount = event.params.amount;
+  transaction.pool = pool.id;
+
   transaction.save();
 
-  updateUserPosition(event, transaction, pool.id);
+  updateUserPosition(event, transaction);
 }
 
 export function handleBurn(event: BurnEvent): void {
@@ -76,7 +78,9 @@ export function handleBurn(event: BurnEvent): void {
   transaction.tickLower = BigInt.fromI32(event.params.tickLower);
   transaction.tickUpper = BigInt.fromI32(event.params.tickUpper);
   transaction.decreaseLiquidityAmount = event.params.amount;
+  transaction.pool = pool.id;
+
   transaction.save();
 
-  updateUserPosition(event, transaction, pool.id);
+  updateUserPosition(event, transaction);
 }
diff --git a/subgraphs/user-position-v3/template/mappings/position-manager.ts b/subgraphs/user-position-v3/template/mappings/position-manager.ts
@@ -5,20 +5,22 @@ import {
   IncreaseLiquidity,
   Transfer,
 } from "../generated/NonfungiblePositionManager/NonfungiblePositionManager";
-import { loadTransaction } from "../utils/schema";
+import { loadTransaction, updateUserPosition } from "../utils/schema";
 import { ADDRESS_ZERO } from "../utils/constants";
 import { UserPosition } from "../generated/schema";
 
 export function handleIncreaseLiquidity(event: IncreaseLiquidity): void {
   let transaction = loadTransaction(event);
   transaction.tokenId = event.params.tokenId;
   transaction.save();
+  updateUserPosition(event, transaction);
 }
 
 export function handleDecreaseLiquidity(event: DecreaseLiquidity): void {
   let transaction = loadTransaction(event);
   transaction.tokenId = event.params.tokenId;
   transaction.save();
+  updateUserPosition(event, transaction);
 }
 
 export function handleTransfer(event: Transfer): void {

diff --git a/subgraphs/user-position-v3/template/schema.graphql b/subgraphs/user-position-v3/template/schema.graphql
@@ -30,6 +30,7 @@ type Pool @entity {
 type Transaction @entity {
   # txn hash
   id: ID!
+  pool: Pool
   # block txn was included in
   blockNumber: BigInt!
   # timestamp txn was confirmed
@@ -122,7 +123,7 @@ type UserPosition @entity {
   id: ID!
   owner: Bytes!
   # pool position is within
-  pool: Pool!
+  pool: Pool
 
   liquidity: BigInt!
   tickLower: BigInt!

diff --git a/subgraphs/user-position-v3/template/utils/schema.ts b/subgraphs/user-position-v3/template/utils/schema.ts
@@ -16,7 +16,7 @@ export function loadTransaction(event: ethereum.Event): Transaction {
   return transaction as Transaction;
 }
 
-export function updateUserPosition(event: ethereum.Event, tx: Transaction, poolId: string): void {
+export function updateUserPosition(event: ethereum.Event, tx: Transaction): void {
   if (
     tx.isPositionUpdated === false &&
     tx.tickLower !== null &&
@@ -27,9 +27,9 @@ export function updateUserPosition(event: ethereum.Event, tx: Transaction, poolI
   ) {
     let userPosition = UserPosition.load(tx.tokenId.toString());
     if (userPosition === null) {
-      userPosition = new UserPosition(tx.tokenId.toString());
+      userPosition = new UserPosition(tx.tokenId.toString() + "-" + tx.pool);
       userPosition.liquidity = ZERO_BI;
-      userPosition.pool = poolId;
+      userPosition.pool = tx.pool;
       userPosition.createdAtBlockNumber = event.block.number;
       userPosition.createdAtTimestamp = event.block.timestamp;
     }