From c9da0f2f512817769b9fe79684dcb14ae2eb98d7 Mon Sep 17 00:00:00 2001
From: Kyle Peacock <kylpeacock@gmail.com>
Date: Wed, 8 Nov 2023 11:54:40 -0800
Subject: [PATCH] feat: uses expirable map for subnet keys

---
 docs/generated/changelog.html          | 1 +
 packages/agent/src/agent/http/index.ts | 5 ++++-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/docs/generated/changelog.html b/docs/generated/changelog.html
index d67a34196..8f06d5273 100644
--- a/docs/generated/changelog.html
+++ b/docs/generated/changelog.html
@@ -12,6 +12,7 @@ <h1>Agent-JS Changelog</h1>
     <section>
       <h2>Version x.x.x</h2>
       <ul>
+        <li>feat: uses expirable map for subnet keys in agent-js, with a timeout of 1 hour</li>
         <ul>
           <strong>feat!: node signature verification</strong
           ><br />
diff --git a/packages/agent/src/agent/http/index.ts b/packages/agent/src/agent/http/index.ts
index b8b9e043c..db0dc78b3 100644
--- a/packages/agent/src/agent/http/index.ts
+++ b/packages/agent/src/agent/http/index.ts
@@ -30,6 +30,7 @@ import { AgentHTTPResponseError } from './errors';
 import { SubnetStatus, request } from '../../canisterStatus';
 import { CertificateVerificationError } from '../../certificate';
 import { ed25519 } from '@noble/curves/ed25519';
+import { ExpirableMap } from '../../utils/expirableMap';
 import { Ed25519PublicKey } from '../../public_key';
 
 export * from './transforms';
@@ -186,7 +187,9 @@ export class HttpAgent implements Agent {
   #queryPipeline: HttpAgentRequestTransformFn[] = [];
   #updatePipeline: HttpAgentRequestTransformFn[] = [];
 
-  #subnetKeys: Map<string, SubnetStatus> = new Map();
+  #subnetKeys: ExpirableMap<string, SubnetStatus> = new ExpirableMap({
+    expirationTime: 60 * 60 * 1000, // 1 hour
+  });
   #verifyQuerySignatures = true;
 
   constructor(options: HttpAgentOptions = {}) {