Merge pull request #99 from Carifio24/cache-hashed-keys

Cache valid API keys
cosmicds · Dec 12, 2023 · 308e29d · 308e29d
2 parents cce1794 + ee2b033
commit 308e29d
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/authorization.ts b/src/authorization.ts
@@ -4,11 +4,20 @@ import { APIKey } from "./models/api_key";
 
 const HASHER = new SHA3(256);
 
+const validKeys = new Map<string, APIKey>();
+
 export async function getAPIKey(key: string): Promise<APIKey | null> {
+  const cachedKey = validKeys.get(key);
+  if (cachedKey !== undefined) {
+    return cachedKey;
+  }
   HASHER.update(key);
   const hashedKey = HASHER.digest("hex");
   const apiKey = await APIKey.findOne({ where: { hashed_key: hashedKey } });
   HASHER.reset();
+  if (apiKey !== null) {
+    validKeys.set(key, apiKey);
+  }
   return apiKey;
 }