New: Opportunity to provide custom connection management agent for th…

…e client
trustpilot · Mar 19, 2021 · 884004c · 884004c
1 parent c36bfe2
commit 884004c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/src/request-helper.ts b/src/request-helper.ts
@@ -1,9 +1,11 @@
 import * as rp from 'request-promise-native';
+import * as http from 'http';
+import * as https from 'https';
 
 import { AccessProvider } from './access-provider';
 
 export class RequestHelper {
-  constructor(private accessProvider: AccessProvider) {}
+  constructor(private accessProvider: AccessProvider, private agent?: http.Agent | https.Agent) {}
 
   get basicRequest() {
     const headers = this.accessProvider.trustpilotApiConfig.defaultHeaders || {};
@@ -12,6 +14,7 @@ export class RequestHelper {
       baseUrl: this.accessProvider.trustpilotApiConfig.baseUrl,
       headers,
       json: true,
+      agent: this.agent,
     });
   }
 

diff --git a/src/trustpilot-api.ts b/src/trustpilot-api.ts
@@ -1,13 +1,15 @@
 import { AccessProvider } from './access-provider';
 import { ITrustpilotApiConfig } from './models';
 import { RequestHelper } from './request-helper';
+import * as http from 'http';
+import * as https from 'https';
 
 export class TrustpilotApi {
   private requestHelper: RequestHelper;
 
-  constructor(config: Readonly<ITrustpilotApiConfig>) {
+  constructor(config: Readonly<ITrustpilotApiConfig>, agent?: http.Agent | https.Agent) {
     const accessProvider = new AccessProvider(Object.freeze(config));
-    this.requestHelper = new RequestHelper(accessProvider);
+    this.requestHelper = new RequestHelper(accessProvider, agent);
   }
 
   get apiRequest() {