expose cache size env var (#858)

* expose cache size env var

* fix typo

Author: d4mr

src/shared/db/wallet-credentials/get-wallet-credential.ts

@@ -29,7 +29,7 @@ export type ParsedWalletCredential = z.infer<typeof walletCredentialsSchema>;
 export const walletCredentialsCache = new LRUMap<
   string,
   ParsedWalletCredential
->(2048);
+>(env.ACCOUNT_CACHE_SIZE);
 
 interface GetWalletCredentialParams {
   id: string;

src/shared/db/wallets/get-wallet-details.ts

@@ -151,7 +151,7 @@ export type SmartBackendWalletType = (typeof SmartBackendWalletTypes)[number];
 export type BackendWalletType = (typeof BackendWalletTypes)[number];
 export type ParsedWalletDetails = z.infer<typeof walletDetailsSchema>;
 
-export const walletDetailsCache = new LRUMap<string, ParsedWalletDetails>(2048);
+export const walletDetailsCache = new LRUMap<string, ParsedWalletDetails>(env.ACCOUNT_CACHE_SIZE);
 /**
  * Return the wallet details for the given address.
  *

src/shared/utils/account.ts

@@ -21,8 +21,9 @@ import { thirdwebClient } from "./sdk";
 import { getWalletCredential } from "../db/wallet-credentials/get-wallet-credential";
 import { getCircleAccount } from "../../server/utils/wallets/circle";
 import { getConfig } from "./cache/get-config";
+import { env } from "./env";
 
-export const _accountsCache = new LRUMap<string, Account>(2048);
+export const _accountsCache = new LRUMap<string, Account>(env.ACCOUNT_CACHE_SIZE);
 
 export const getAccount = async (args: {
   chainId: number;
@@ -212,7 +213,7 @@ export const walletDetailsToAccount = async ({
   }
 };
 
-export const _adminAccountsCache = new LRUMap<string, Account>(2048);
+export const _adminAccountsCache = new LRUMap<string, Account>(env.ACCOUNT_CACHE_SIZE);
 
 /**
  * Get the admin account for a smart backend wallet (cached)

src/shared/utils/cache/access-token.ts

@@ -1,9 +1,10 @@
 import type { Tokens } from "@prisma/client";
 import LRUMap from "mnemonist/lru-map";
 import { getToken } from "../../db/tokens/get-token";
+import { env } from "../env";
 
 // Cache an access token JWT to the token object, or null if not found.
-export const accessTokenCache = new LRUMap<string, Tokens | null>(2048);
+export const accessTokenCache = new LRUMap<string, Tokens | null>(env.ACCOUNT_CACHE_SIZE);
 
 interface GetAccessTokenParams {
   jwt: string;

src/shared/utils/cache/get-sdk.ts

@@ -7,7 +7,7 @@ import { getChain } from "../chain";
 import { env } from "../env";
 import { getWallet } from "./get-wallet";
 
-export const sdkCache = new LRUMap<string, ThirdwebSDK>(2048);
+export const sdkCache = new LRUMap<string, ThirdwebSDK>(env.ACCOUNT_CACHE_SIZE);
 
 export const networkResponseSchema = Type.Object({
   name: Type.String({

src/shared/utils/cache/get-smart-wallet-v5.ts

@@ -3,8 +3,9 @@ import { getContract, readContract, type Address, type Chain } from "thirdweb";
 import { smartWallet, type Account } from "thirdweb/wallets";
 import { getAccount } from "../account";
 import { thirdwebClient } from "../sdk";
+import { env } from "../env";
 
-export const smartWalletsCache = new LRUMap<string, Account>(2048);
+export const smartWalletsCache = new LRUMap<string, Account>(env.ACCOUNT_CACHE_SIZE);
 
 interface SmartWalletParams {
   chain: Chain;

src/shared/utils/cache/get-wallet.ts

@@ -14,8 +14,9 @@ import { splitAwsKmsArn } from "../../../server/utils/wallets/aws-kms-arn";
 import { splitGcpKmsResourcePath } from "../../../server/utils/wallets/gcp-kms-resource-path";
 import { getLocalWallet } from "../../../server/utils/wallets/get-local-wallet";
 import { getSmartWallet } from "../../../server/utils/wallets/get-smart-wallet";
+import { env } from "../env";
 
-export const walletsCache = new LRUMap<string, EVMWallet>(2048);
+export const walletsCache = new LRUMap<string, EVMWallet>(env.ACCOUNT_CACHE_SIZE);
 
 interface GetWalletParams {
   pgtx?: PrismaTransaction;

src/shared/utils/cache/get-webhook.ts

@@ -2,8 +2,9 @@ import type { Webhooks } from "@prisma/client";
 import LRUMap from "mnemonist/lru-map";
 import { getAllWebhooks } from "../../db/webhooks/get-all-webhooks";
 import type { WebhooksEventTypes } from "../../schemas/webhooks";
+import { env } from "../env";
 
-export const webhookCache = new LRUMap<string, Webhooks[]>(2048);
+export const webhookCache = new LRUMap<string, Webhooks[]>(env.ACCOUNT_CACHE_SIZE);
 
 export const getWebhooksByEventType = async (
   eventType: WebhooksEventTypes,

src/shared/utils/cache/keypair.ts

@@ -1,9 +1,10 @@
 import type { Keypairs } from "@prisma/client";
 import LRUMap from "mnemonist/lru-map";
 import { getKeypairByHash, getKeypairByPublicKey } from "../../db/keypair/get";
+import { env } from "../env";
 
 // Cache a public key to the Keypair object, or null if not found.
-export const keypairCache = new LRUMap<string, Keypairs | null>(2048);
+export const keypairCache = new LRUMap<string, Keypairs | null>(env.ACCOUNT_CACHE_SIZE);
 
 /**
  * Get a keypair by public key or hash.

src/shared/utils/env.ts

@@ -63,6 +63,7 @@ export const env = createEnv({
       .enum(["default", "sandbox", "server_only", "worker_only"])
       .default("default"),
     GLOBAL_RATE_LIMIT_PER_MIN: z.coerce.number().default(400 * 60),
+    ACCOUNT_CACHE_SIZE: z.coerce.number().default(2048),
     DD_TRACER_ACTIVATED: boolEnvSchema(false),
 
     // Prometheus
@@ -145,6 +146,7 @@ export const env = createEnv({
     ENABLE_CUSTOM_HMAC_AUTH: process.env.ENABLE_CUSTOM_HMAC_AUTH,
     CUSTOM_HMAC_AUTH_CLIENT_ID: process.env.CUSTOM_HMAC_AUTH_CLIENT_ID,
     CUSTOM_HMAC_AUTH_CLIENT_SECRET: process.env.CUSTOM_HMAC_AUTH_CLIENT_SECRET,
+    ACCOUNT_CACHE_SIZE: process.env.ACCOUNT_CAHCE_SIZE,
   },
   onValidationError: (error: ZodError) => {
     console.error(