Merge pull request #25 from DanteInc/issue-allow-max-sockets-configur…

…ation

Allow max sockets configuration in sdk client.

package.json

@@ -1,6 +1,6 @@
 {
   "name": "aws-kms-ee",
-  "version": "1.0.3",
+  "version": "1.0.4",
   "description": "AWS KMS Envelope Encryption",
   "main": "./lib/index.js",
   "files": [

src/connector.js

@@ -1,6 +1,7 @@
 /* eslint import/no-extraneous-dependencies: ["error", {"devDependencies": true}] */
 import Promise from 'bluebird';
 import memoryCache from 'memory-cache';
+import { Agent } from 'https';
 import { DecryptCommand, EncryptCommand, GenerateDataKeyCommand, KMSClient } from '@aws-sdk/client-kms';
 import { NodeHttpHandler } from '@smithy/node-http-handler';
 import { getClientLogger } from './utils';
@@ -14,13 +15,17 @@ class Connector {
     timeout = Number(process.env.KMS_TIMEOUT || process.env.TIMEOUT || 1000),
     connectTimeout = Number(process.env.KMS_CONNECT_TIMEOUT || process.env.CONNECT_TIMEOUT || 1000),
     maxAge = Number(process.env.DATA_KEY_MAX_AGE) || undefined, // default to life of lambda function
+    maxSockets = Number(process.env.MAX_SOCKETS) || 50,
   ) {
     this.maxAge = maxAge;
     this.masterKeyAlias = masterKeyAlias;
     this.kms = new KMSClient({
       requestHandler: new NodeHttpHandler({
         requestTimeout: timeout,
         connectionTimeout: connectTimeout,
+        httpsAgent: new Agent({
+          maxSockets,
+        }),
       }),
       logger: getClientLogger(),
       region,