Merge pull request #769 from snyk/feat/re-use-sockets

feat: re-use sockets when sending data to kubernetes upstream

Author: dkontorovskyy

src/transmitter/index.ts

@@ -1,7 +1,11 @@
+import { parse } from 'url';
 import { queue } from 'async';
 import * as needle from 'needle';
-import { NeedleResponse, NeedleHttpVerbs, NeedleOptions } from 'needle';
 import * as sleep from 'sleep-promise';
+import { Agent as HttpAgent } from 'http';
+import { Agent as HttpsAgent } from 'https';
+import { NeedleResponse, NeedleHttpVerbs, NeedleOptions } from 'needle';
+
 import { logger } from '../common/logger';
 import { config } from '../common/config';
 import {
@@ -28,6 +32,16 @@ interface HomebaseRequest {
 const upstreamUrl =
   config.INTEGRATION_API || config.DEFAULT_KUBERNETES_UPSTREAM_URL;
 
+let agent = new HttpAgent({
+  keepAlive: true,
+});
+
+if (parse(upstreamUrl).protocol?.startsWith('https')) {
+  agent = new HttpsAgent({
+    keepAlive: true,
+  });
+}
+
 // Async queue wraps around the call to retryRequest in order to limit
 // the number of requests in flight to Homebase at any one time.
 const reqQueue = queue(async function (req: HomebaseRequest) {
@@ -227,7 +241,9 @@ async function retryRequest(
   const options: NeedleOptions = {
     json: true,
     compressed: true,
+    agent,
   };
+
   if (config.HTTP_PROXY || config.HTTPS_PROXY) {
     options.agent = getProxyAgent(config, url);
   }