From 06f0314024e02801f091916500946af94bd6e11c Mon Sep 17 00:00:00 2001
From: Georg Friedrich <georg@codezero.io>
Date: Thu, 19 Jun 2025 09:00:56 +1000
Subject: [PATCH 1/3] set keepAlive in order to reuse connections to services
 by default

see https://github.com/nodejs/node/blob/v20.x/lib/_http_agent.js#L167
---
 src/index.mts | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/index.mts b/src/index.mts
index 63a0a53..d756be7 100644
--- a/src/index.mts
+++ b/src/index.mts
@@ -37,6 +37,7 @@ export class CodezeroAgent extends HttpsProxyAgent<string> {
     this.orgID = orgID;
     this.orgApiKey = orgApiKey;
     this.spaceID = spaceID;
+    this.keepAlive = true;
 
     if (process.env.CZ_HUB_SERVER_BASE_URL) {
       this._hubServerBaseUrl = process.env.CZ_HUB_SERVER_BASE_URL!;

From 25735d75705f675ba2b22e541cfed73b98a09dee Mon Sep 17 00:00:00 2001
From: Georg Friedrich <georg@codezero.io>
Date: Thu, 19 Jun 2025 10:51:37 +1000
Subject: [PATCH 2/3] fix test

---
 test/test.mts | 1 +
 1 file changed, 1 insertion(+)

diff --git a/test/test.mts b/test/test.mts
index 92dd9ff..89526b9 100644
--- a/test/test.mts
+++ b/test/test.mts
@@ -92,6 +92,7 @@ describe('CodezeroAgent', () => {
 
     it('should refetch credentials if token is expired', async () => {
       const agent = new CodezeroAgent({orgID: "orgId", orgApiKey: "orgApiKey", spaceID: "spaceId"});
+      agent.keepAlive = false; // avoid reusing the socket
 
       await fetch(targetServerUrl.href, { agent });
       expect(hubRequestCount).toBe(1);

From 678f993f94882da8ff26ca30c9c795682dcbff2d Mon Sep 17 00:00:00 2001
From: Georg Friedrich <georg@codezero.io>
Date: Thu, 19 Jun 2025 10:59:09 +1000
Subject: [PATCH 3/3] add test for connection reuse

---
 test/test.mts | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)

diff --git a/test/test.mts b/test/test.mts
index 89526b9..c2c9ef3 100644
--- a/test/test.mts
+++ b/test/test.mts
@@ -102,5 +102,35 @@ describe('CodezeroAgent', () => {
 
       expect(hubRequestCount).toBe(2);
     });
+    
+    it('should reuse the sockets for subsequent requests to the same host', async () => {
+      const targetServer0 = createServer((req, res) => {
+        res.end('targetServer0');
+      });
+      const targetServer1 = createServer((req, res) => {
+        res.end('targetServer1');
+      });
+      const targetServer0Url = await listen(targetServer0);
+      const targetServer1Url = await listen(targetServer1);
+      
+      const agent = new CodezeroAgent({orgID: "orgId", orgApiKey: "orgApiKey", spaceID: "spaceId"});
+
+      let res = await fetch(targetServer0Url.href, { agent });
+      expect(hubRequestCount).toBe(1);
+      expect(await res.text()).toBe('targetServer0');
+
+      agent['_credentials']!['token'] = createToken(-3*60);
+      res = await fetch(targetServer0Url.href, { agent });
+      expect(hubRequestCount).toBe(1); // should not refetch credentials because it's reusing the socket
+      expect(await res.text()).toBe('targetServer0');
+      
+      
+      res = await fetch(targetServer1Url.href, { agent });
+      expect(hubRequestCount).toBe(2); // should refetch credentials because it's a different target server
+      expect(await res.text()).toBe('targetServer1');
+      
+      targetServer0.close();
+      targetServer1.close();
+    });
   });
 });