Keep-alive (reuse) connections running in NodeJS environment to reduc…

…e the number of new connections Ref: near/nearcore#1316
near · Oct 4, 2019 · 2cf6dca · 2cf6dca
1 parent 2485e64
commit 2cf6dca
Show file tree

Hide file tree

Showing 2 changed files with 45 additions and 2 deletions.
diff --git a/lib/utils/web.js b/lib/utils/web.js
diff --git a/src.ts/utils/web.ts b/src.ts/utils/web.ts
@@ -11,7 +11,29 @@ export interface ConnectionInfo {
     headers?: { [key: string]: string | number };
 }
 
-const fetch = (typeof window === 'undefined' || window.name === 'nodejs') ? require('node-fetch') : window.fetch;
+let fetch;
+if (typeof window === 'undefined' || window.name === 'nodejs') {
+  const nodeFetch = require('node-fetch');
+  const http = require('http');
+  const https = require('https');
+
+  const httpAgent = new http.Agent({ keepAlive: true });
+  const httpsAgent = new https.Agent({ keepAlive: true });
+
+  function agent(_parsedURL) {
+    if (_parsedURL.protocol === 'http:') {
+      return httpAgent;
+    } else {
+      return httpsAgent;
+    }
+  }
+
+  fetch = function(resource, init) {
+    return nodeFetch(resource, { agent, ...init });
+  };
+} else {
+  fetch = window.fetch;
+}
 
 export async function fetchJson(connection: string | ConnectionInfo, json?: string): Promise<any> {
     let url: string = null;