Pricing revamp (#180)

* price handler helper * linear * price handlers * remove unused * fix * pass chain as param * more chain passing * remove unused * redo updates * remove unused * adding retrieval of historical prices * fix historical pricing --------- Co-authored-by: gmbronco <[email protected]>
balancer · Mar 11, 2024 · f64a491 · f64a491
1 parent 265da76
commit f64a491
Show file tree

Hide file tree

Showing 51 changed files with 817 additions and 1,231 deletions.
diff --git a/modules/actions/pool/update-on-chain-data.ts b/modules/actions/pool/update-on-chain-data.ts
@@ -205,8 +205,11 @@ export async function updateOnChainDataForPools(
                                 balanceUSD:
                                     poolToken.address === pool.address
                                         ? 0
-                                        : tokenService.getPriceForToken(tokenPricesForCurrentChain, poolToken.address) *
-                                          parseFloat(balance),
+                                        : tokenService.getPriceForToken(
+                                              tokenPricesForCurrentChain,
+                                              poolToken.address,
+                                              chain,
+                                          ) * parseFloat(balance),
                             },
                         }),
                     );

diff --git a/modules/beets/beets.service.ts b/modules/beets/beets.service.ts
@@ -15,7 +15,7 @@ export class BeetsService {
 
     public async getBeetsPrice(): Promise<string> {
         const tokenPrices = await tokenService.getTokenPrices();
-        return tokenService.getPriceForToken(tokenPrices, networkContext.data.beets!.address).toString();
+        return tokenService.getPriceForToken(tokenPrices, networkContext.data.beets!.address, 'FANTOM').toString();
     }
 }
 

diff --git a/modules/coingecko/coingecko-types.ts b/modules/coingecko/coingecko-types.ts
diff --git a/modules/coingecko/coingecko.service.ts b/modules/coingecko/coingecko.service.ts
diff --git a/modules/datastudio/datastudio.service.ts b/modules/datastudio/datastudio.service.ts
@@ -12,11 +12,12 @@ import { oneDayInSeconds, secondsPerDay } from '../common/time';
 import { isComposableStablePool, isWeightedPoolV2 } from '../pool/lib/pool-utils';
 import { networkContext } from '../network/network-context.service';
 import { DeploymentEnv } from '../network/network-config-types';
+import { Chain } from '@prisma/client';
 
 export class DatastudioService {
     constructor(private readonly secretsManager: SecretsManager, private readonly jwtClientHelper: GoogleJwtClient) {}
 
-    public async feedPoolData() {
+    public async feedPoolData(chain: Chain) {
         const privateKey = await this.secretsManager.getSecret('backend-v3-datafeed-privatekey');
         const jwtClient = await this.jwtClientHelper.getAuthorizedSheetsClient(privateKey);
 
@@ -89,7 +90,7 @@ export class DatastudioService {
                 dynamicData: {
                     totalLiquidity: { gte: 5000 },
                 },
-                chain: networkContext.chain,
+                chain: chain,
             },
             include: {
                 dynamicData: true,
@@ -265,7 +266,8 @@ export class DatastudioService {
                             }
                             const rewardsPerDay = parseFloat(rewarder.rewardPerSecond) * secondsPerDay;
                             const rewardsValuePerDay =
-                                tokenService.getPriceForToken(tokenPrices, rewarder.tokenAddress) * rewardsPerDay;
+                                tokenService.getPriceForToken(tokenPrices, rewarder.tokenAddress, chain) *
+                                rewardsPerDay;
                             if (rewardsPerDay > 0) {
                                 allEmissionDataRows.push([
                                     endOfYesterday.format('DD MMM YYYY'),
@@ -309,7 +311,7 @@ export class DatastudioService {
                         }
                         const rewardsPerDay = parseFloat(reward.rewardPerSecond) * secondsPerDay;
                         const rewardsValuePerDay =
-                            tokenService.getPriceForToken(tokenPrices, reward.tokenAddress) * rewardsPerDay;
+                            tokenService.getPriceForToken(tokenPrices, reward.tokenAddress, chain) * rewardsPerDay;
                         if (rewardsPerDay > 0) {
                             allEmissionDataRows.push([
                                 endOfYesterday.format('DD MMM YYYY'),

diff --git a/modules/network/arbitrum.ts b/modules/network/arbitrum.ts
@@ -6,15 +6,10 @@ import { BoostedPoolAprService } from '../pool/lib/apr-data-sources/boosted-pool
 import { SwapFeeAprService } from '../pool/lib/apr-data-sources/swap-fee-apr.service';
 import { GaugeAprService } from '../pool/lib/apr-data-sources/ve-bal-gauge-apr.service';
 import { GaugeStakingService } from '../pool/lib/staking/gauge-staking.service';
-import { BptPriceHandlerService } from '../token/lib/token-price-handlers/bpt-price-handler.service';
-import { LinearWrappedTokenPriceHandlerService } from '../token/lib/token-price-handlers/linear-wrapped-token-price-handler.service';
-import { SwapsPriceHandlerService } from '../token/lib/token-price-handlers/swaps-price-handler.service';
 import { UserSyncGaugeBalanceService } from '../user/lib/user-sync-gauge-balance.service';
 import { every } from '../../worker/intervals';
 import { GithubContentService } from '../content/github-content.service';
 import { gaugeSubgraphService } from '../subgraphs/gauge-subgraph/gauge-subgraph.service';
-import { CoingeckoPriceHandlerService } from '../token/lib/token-price-handlers/coingecko-price-handler.service';
-import { coingeckoService } from '../coingecko/coingecko.service';
 import { YbTokensAprService } from '../pool/lib/apr-data-sources/yb-tokens-apr.service';
 import { env } from '../../app/env';
 import { BalancerSubgraphService } from '../subgraphs/balancer-subgraph/balancer-subgraph.service';
@@ -227,12 +222,6 @@ export const arbitrumNetworkConfig: NetworkConfig = {
         new GaugeAprService(tokenService, [arbitrumNetworkData.bal!.address]),
     ],
     poolStakingServices: [new GaugeStakingService(gaugeSubgraphService, arbitrumNetworkData.bal!.address)],
-    tokenPriceHandlers: [
-        new CoingeckoPriceHandlerService(coingeckoService),
-        new BptPriceHandlerService(),
-        new LinearWrappedTokenPriceHandlerService(),
-        new SwapsPriceHandlerService(),
-    ],
     userStakedBalanceServices: [new UserSyncGaugeBalanceService()],
     services: {
         balancerSubgraphService: new BalancerSubgraphService(

diff --git a/modules/network/avalanche.ts b/modules/network/avalanche.ts
@@ -6,15 +6,10 @@ import { BoostedPoolAprService } from '../pool/lib/apr-data-sources/boosted-pool
 import { SwapFeeAprService } from '../pool/lib/apr-data-sources/swap-fee-apr.service';
 import { GaugeAprService } from '../pool/lib/apr-data-sources/ve-bal-gauge-apr.service';
 import { GaugeStakingService } from '../pool/lib/staking/gauge-staking.service';
-import { BptPriceHandlerService } from '../token/lib/token-price-handlers/bpt-price-handler.service';
-import { LinearWrappedTokenPriceHandlerService } from '../token/lib/token-price-handlers/linear-wrapped-token-price-handler.service';
-import { SwapsPriceHandlerService } from '../token/lib/token-price-handlers/swaps-price-handler.service';
 import { UserSyncGaugeBalanceService } from '../user/lib/user-sync-gauge-balance.service';
 import { every } from '../../worker/intervals';
 import { GithubContentService } from '../content/github-content.service';
 import { gaugeSubgraphService } from '../subgraphs/gauge-subgraph/gauge-subgraph.service';
-import { coingeckoService } from '../coingecko/coingecko.service';
-import { CoingeckoPriceHandlerService } from '../token/lib/token-price-handlers/coingecko-price-handler.service';
 import { env } from '../../app/env';
 import { YbTokensAprService } from '../pool/lib/apr-data-sources/yb-tokens-apr.service';
 import { BalancerSubgraphService } from '../subgraphs/balancer-subgraph/balancer-subgraph.service';
@@ -202,12 +197,6 @@ export const avalancheNetworkConfig: NetworkConfig = {
         new GaugeAprService(tokenService, [avalancheNetworkData.bal!.address]),
     ],
     poolStakingServices: [new GaugeStakingService(gaugeSubgraphService, avalancheNetworkData.bal!.address)],
-    tokenPriceHandlers: [
-        new CoingeckoPriceHandlerService(coingeckoService),
-        new BptPriceHandlerService(),
-        new LinearWrappedTokenPriceHandlerService(),
-        new SwapsPriceHandlerService(),
-    ],
     userStakedBalanceServices: [new UserSyncGaugeBalanceService()],
     services: {
         balancerSubgraphService: new BalancerSubgraphService(