feat(uniswap-v2): Get positions for balances from DB (Zapper-fi#1903)

0xdapper · Feb 28, 2023 · c464db9 · c464db9
1 parent 1a0acbb
commit c464db9
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 0 deletions.
diff --git a/src/app-toolkit/app-toolkit.interface.ts b/src/app-toolkit/app-toolkit.interface.ts
@@ -46,6 +46,10 @@ export interface IAppToolkit {
     ...appTokenDefinition: AppGroupsDefinition[]
   ): Promise<AppTokenPosition<T>[]>;
 
+  getAppTokenPositionsFromDatabase<T = DefaultDataProps>(
+    ...appTokenDefinition: AppGroupsDefinition[]
+  ): Promise<AppTokenPosition<T>[]>;
+
   getAppContractPositions<T = DefaultDataProps>(
     ...appTokenDefinition: AppGroupsDefinition[]
   ): Promise<ContractPosition<T>[]>;

diff --git a/src/app-toolkit/app-toolkit.service.ts b/src/app-toolkit/app-toolkit.service.ts
@@ -80,6 +80,10 @@ export class AppToolkit implements IAppToolkit {
     return this.positionService.getAppTokenPositions<T>(...appTokenDefinitions);
   }
 
+  getAppTokenPositionsFromDatabase<T = DefaultDataProps>(...appTokenDefinitions: AppGroupsDefinition[]) {
+    return this.positionService.getAppTokenPositions<T>(...appTokenDefinitions);
+  }
+
   getAppContractPositions<T = DefaultDataProps>(...appTokenDefinitions: AppGroupsDefinition[]) {
     return this.positionService.getAppContractPositions<T>(...appTokenDefinitions);
   }

diff --git a/src/apps/uniswap-v2/ethereum/uniswap-v2.pool.token-fetcher.ts b/src/apps/uniswap-v2/ethereum/uniswap-v2.pool.token-fetcher.ts
@@ -41,6 +41,14 @@ export class EthereumUniswapV2PoolTokenFetcher extends UniswapV2DefaultPoolSubgr
   ignoredPools = ['0x3016a43b482d0480460f6625115bd372fe90c6bf'];
   first = 5000;
 
+  async getPositionsForBalances() {
+    return this.appToolkit.getAppTokenPositionsFromDatabase<UniswapV2TokenDataProps>({
+      appId: this.appId,
+      network: this.network,
+      groupIds: [this.groupId],
+    });
+  }
+
   async getBalances(_address: string): Promise<AppTokenPositionBalance<UniswapV2TokenDataProps>[]> {
     const multicall = this.appToolkit.getMulticall(this.network);
     const tokenLoader = this.appToolkit.getTokenDependencySelector();