add max_connection_pool parameter

Author: ramonpetgrave64

kmsauth/__init__.py

@@ -57,6 +57,7 @@ def __init__(
             endpoint_url=None,
             token_cache_size=4096,
             stats=None,
+            max_pool_connections=100
             ):
         """Create a KMSTokenValidator object.
 
@@ -107,13 +108,15 @@ def __init__(
                 aws_access_key_id=self.aws_creds['AccessKeyId'],
                 aws_secret_access_key=self.aws_creds['SecretAccessKey'],
                 aws_session_token=self.aws_creds['SessionToken'],
-                endpoint_url=endpoint_url
+                endpoint_url=endpoint_url,
+                max_pool_connections=max_pool_connections
             )
         else:
             self.kms_client = kmsauth.services.get_boto_client(
                 'kms',
                 region=self.region,
-                endpoint_url=endpoint_url
+                endpoint_url=endpoint_url,
+                max_pool_connections=max_pool_connections
             )
         if extra_context is None:
             self.extra_context = {}

kmsauth/services.py

@@ -1,6 +1,7 @@
 """Module for accessing boto3 clients, resources and sessions."""
 
 import boto3
+import botocore
 import logging
 
 CLIENT_CACHE = {}
@@ -13,7 +14,8 @@ def get_boto_client(
         aws_access_key_id=None,
         aws_secret_access_key=None,
         aws_session_token=None,
-        endpoint_url=None
+        endpoint_url=None,
+        max_pool_connections=None
         ):
     """Get a boto3 client connection."""
     cache_key = '{0}:{1}:{2}:{3}'.format(
@@ -37,7 +39,10 @@ def get_boto_client(
 
     CLIENT_CACHE[cache_key] = session.client(
         client,
-        endpoint_url=endpoint_url
+        endpoint_url=endpoint_url,
+        config=botocore.config.Config(
+            max_pool_connections=max_pool_connections
+        )
     )
     return CLIENT_CACHE[cache_key]
 

setup.py

@@ -13,7 +13,7 @@
 
 from setuptools import setup, find_packages
 
-VERSION = "0.6.0"
+VERSION = "0.6.1.dev1"
 
 requirements = [
     # Boto3 is the Amazon Web Services (AWS) Software Development Kit (SDK)