add keepAlive for proxy agent

Last-Order · Jan 10, 2024 · 7249b25 · 7249b25
1 parent b0c0cdf
commit 7249b25
Show file tree

Hide file tree

Showing 3 changed files with 69 additions and 63 deletions.
diff --git a/npm-shrinkwrap.json b/npm-shrinkwrap.json
diff --git a/package.json b/package.json
@@ -30,14 +30,14 @@
     "license": "GPL-3.0-or-later",
     "dependencies": {
         "@eridanussora/reconnecting-websocket": "^4.4.2",
-        "agent-base": "^6.0.2",
+        "agent-base": "^7.1.0",
         "axios": "^1.6.0",
         "chalk": "^2.4.2",
         "cli-progress": "^3.10.0",
         "crypto-js": "^3.3.0",
         "erii": "^2.0.6",
-        "https-proxy-agent": "^5.0.0",
-        "socks-proxy-agent": "^6.1.0",
+        "https-proxy-agent": "^7.0.2",
+        "socks-proxy-agent": "^8.0.2",
         "ws": "^7.4.6"
     },
     "devDependencies": {

diff --git a/src/utils/agent.ts b/src/utils/agent.ts
@@ -22,7 +22,9 @@ class ProxyAgentHelper {
         }
         if (proxy.startsWith("http://") || proxy.startsWith("https://")) {
             // HTTP Proxy
-            this.proxyAgentInstance = new HttpsProxyAgent(proxy);
+            this.proxyAgentInstance = new HttpsProxyAgent(proxy, {
+                keepAlive: true,
+            });
             logger.debug(`HTTP/HTTPS Proxy set: ${proxy}`);
         } else if (proxy.startsWith("socks")) {
             if (proxy.startsWith("socks4")) {
@@ -31,7 +33,9 @@ class ProxyAgentHelper {
             // Socks5 Proxy
             try {
                 const [_, host, port] = proxy.match(/socks5?(?:(?<=5)h)?[:：]\/\/(.+)[:：](\d+)/);
-                this.proxyAgentInstance = new SocksProxyAgent(`socks5h://${host}:${port}`);
+                this.proxyAgentInstance = new SocksProxyAgent(`socks5h://${host}:${port}`, {
+                    keepAlive: true,
+                });
                 logger.debug(`Socks5 Proxy set: socks5h://${host}:${port}`);
             } catch (e) {
                 throw new InvalidProxyServerError("Proxy server invalid.");
@@ -40,7 +44,9 @@ class ProxyAgentHelper {
             // For compatibility, use proxy without protocol as socks5 proxy
             try {
                 const [_, host, port] = proxy.match(/(.+)[:：](\d+)/);
-                this.proxyAgentInstance = new SocksProxyAgent(`socks5h://${host}:${port}`);
+                this.proxyAgentInstance = new SocksProxyAgent(`socks5h://${host}:${port}`, {
+                    keepAlive: true,
+                });
                 logger.debug(`Socks5 Proxy set: socks5h://${host}:${port}`);
             } catch (e) {
                 throw new InvalidProxyServerError("Proxy server invalid.");