feat: custom agent config

ganker-vito · Sep 2, 2022 · 86fc56b · 86fc56b
1 parent 4643268
commit 86fc56b
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/lib/config.js b/lib/config.js
@@ -595,8 +595,10 @@ exports.extend = function (newConf) {
     newConf.mode = process.env.WHISTLE_MODE + '|' + (newConf.mode || '');
   }
   var useAgent;
+  var newAgentConf;
   if (newConf) {
     config.uiMiddleware = newConf.uiMiddlewares || newConf.uiMiddleware;
+    newAgentConf = newConf.agentConfig;
     if (newConf.cmdName && CMD_RE.test(newConf.cmdName)) {
       config.cmdName = newConf.cmdName;
     }
@@ -1041,14 +1043,14 @@ exports.extend = function (newConf) {
   if (config.strict) {
     config.sockets = Math.max(config.sockets, 360);
   }
-  var agentConfig = {
+  var agentConfig = newAgentConf || {
     maxSockets: config.sockets,
     keepAlive: true,
     keepAliveMsecs: KEEP_ALIVE_MSECS,
     maxFreeSockets: 0
   };
   // node 11及以上版本缓存连接有问题，先禁掉
-  disableAgent = !useAgent && (disableAgent || config.debug);
+  disableAgent = !newAgentConf && !useAgent && (disableAgent || config.debug);
   config.httpAgent = disableAgent ? false : createAgent(agentConfig);
   config.httpsAgent = disableAgent ? false : createAgent(agentConfig, true);
   config.getSocksAgent = function (options) {