Added cache for functions that uses Coin Gecko

Team-Kujira · Nov 28, 2023 · 45dab94 · 45dab94
1 parent a56a5eb
commit 45dab94
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 10 deletions.
diff --git a/src/connectors/kujira/kujira.config.ts b/src/connectors/kujira/kujira.config.ts
@@ -117,7 +117,14 @@ export namespace KujiraConfig {
       marketsData: configManager.get('kujira.cache.marketsData') || 3600, // in seconds
       markets: configManager.get('kujira.cache.markets') || 3600, // in seconds
       tokens: configManager.get('kujira.cache.markets') || 3600, // in seconds
-      coinGeckoCoins: configManager.get('kujira.cache.coinGeckoCoins') || 3600, // in seconds
+      fetchCoinGecko: configManager.get('kujira.cache.fetchCoinGecko') || 3600, // in seconds
+      getTicker: configManager.get('kujira.cache.getTicker') || 60 * 5, // in seconds
+      getAllTokensQuotationsInUSD:
+        configManager.get('kujira.cache.getAllTokensQuotationsInUSD') || 3600, // in seconds
+      getKujiraTokenSymbolsToCoinGeckoIdsMap:
+        configManager.get(
+          'kujira.cache.getKujiraTokenSymbolsToCoinGeckoIdsMap'
+        ) || 3600, // in seconds
     },
     coinGecko: {
       coinsUrl:

diff --git a/src/connectors/kujira/kujira.ts b/src/connectors/kujira/kujira.ts
@@ -175,7 +175,12 @@ const caches = {
   instances: new CacheContainer(new MemoryStorage()),
   tokens: new CacheContainer(new MemoryStorage()),
   markets: new CacheContainer(new MemoryStorage()),
-  coinGeckoCoins: new CacheContainer(new MemoryStorage()),
+  fetchCoinGecko: new CacheContainer(new MemoryStorage()),
+  getTicker: new CacheContainer(new MemoryStorage()),
+  getAllTokensQuotationsInUSD: new CacheContainer(new MemoryStorage()),
+  getKujiraTokenSymbolsToCoinGeckoIdsMap: new CacheContainer(
+    new MemoryStorage()
+  ),
 };
 
 const config = KujiraConfig.config;
@@ -778,20 +783,15 @@ export class Kujira {
     return output;
   }
 
-  @Cache(caches.coinGeckoCoins, { ttl: config.cache.coinGeckoCoins })
-  async getKujiraTokenSymbolsToCoinGeckoIdsMap(
-    _options?: any,
-    _network?: string
-  ): Promise<GetKujiraTokenSymbolsToCoinGeckoTokenIdsMapResponse> {
-    const output = IMap<TokenSymbol, CoinGeckoId | undefined>().asMutable();
-
+  @Cache(caches.fetchCoinGecko, { ttl: config.cache.fetchCoinGecko })
+  async fetchCoinGecko(): Promise<any> {
     const apiKeys = config.coinGecko.apiKeys;
     const randomIndex = Math.floor(Math.random() * apiKeys.length);
     const apiKey = apiKeys[randomIndex];
 
     const finalUrl = config.coinGecko.coinsUrl.replace('{apiKey}', apiKey);
 
-    const result: any = (
+    return (
       await runWithRetryAndTimeout(
         axios,
         axios.get,
@@ -800,6 +800,18 @@ export class Kujira {
         0
       )
     ).data;
+  }
+
+  @Cache(caches.getKujiraTokenSymbolsToCoinGeckoIdsMap, {
+    ttl: config.cache.getKujiraTokenSymbolsToCoinGeckoIdsMap,
+  })
+  async getKujiraTokenSymbolsToCoinGeckoIdsMap(
+    _options?: any,
+    _network?: string
+  ): Promise<GetKujiraTokenSymbolsToCoinGeckoTokenIdsMapResponse> {
+    const output = IMap<TokenSymbol, CoinGeckoId | undefined>().asMutable();
+
+    const result: any = await this.fetchCoinGecko();
 
     const coinGeckoSymbolsToIdsMap = IMap<
       CoinGeckoSymbol,
@@ -982,6 +994,7 @@ export class Kujira {
    *
    * @param options
    */
+  @Cache(caches.getTicker, { ttl: config.cache.getTicker })
   async getTicker(options: GetTickerRequest): Promise<GetTickerResponse> {
     const market = await this.getMarket(
       options.marketId ? { id: options.marketId } : { name: options.marketName }
@@ -1126,6 +1139,7 @@ export class Kujira {
     return await this.getTickers({ marketIds });
   }
 
+  @Cache(caches.getAllTokensQuotationsInUSD, { ttl: config.cache.getTicker })
   async getAllTokensQuotationsInUSD(
     _options: any
   ): Promise<IMap<TokenId, Price>> {