feat(redis/idempotency): adding base class

Author: leandrodamascena

aws_lambda_powertools/utilities/connections/__init__.py

@@ -0,0 +1,6 @@
+from aws_lambda_powertools.utilities.connections.redis import (
+    RedisCluster,
+    RedisStandalone,
+)
+
+__all__ = (RedisStandalone, RedisCluster)

aws_lambda_powertools/utilities/connections/base_sync.py

@@ -0,0 +1,7 @@
+from abc import ABC, abstractmethod
+
+
+class BaseConnectionSync(ABC):
+    @abstractmethod
+    def init_connection(self):
+        raise NotImplementedError()  # pragma: no cover

aws_lambda_powertools/utilities/database/exceptions.py renamed to aws_lambda_powertools/utilities/connections/exceptions.py

@@ -3,12 +3,13 @@
 
 import redis
 
-from aws_lambda_powertools.utilities.database.exceptions import RedisConnectionError
+from .base_sync import BaseConnectionSync
+from .exceptions import RedisConnectionError
 
 logger = logging.getLogger(__name__)
 
 
-class RedisStandalone:
+class RedisStandalone(BaseConnectionSync):
     def __init__(
         self,
         host: Optional[str] = None,
@@ -30,7 +31,7 @@ def __init__(
             Name of the username to connect to Redis instance/cluster in case of using ACL
             See: https://redis.io/docs/management/security/acl/
         password: str
-            Password to connect to Redis instance/cluster
+            Passwod to connect to Redis instance/cluster
         db_index: int
             Index of Redis database
             See: https://redis.io/commands/select/
@@ -47,7 +48,7 @@ def __init__(
         self.db_index = db_index
         self._connection = None
 
-    def get_redis_connection(self):
+    def init_connection(self):
         """
         Connection is cached, so returning this
         """
@@ -72,7 +73,7 @@ def get_redis_connection(self):
         return self._connection
 
 
-class RedisCluster:
+class RedisCluster(BaseConnectionSync):
     def __init__(
         self,
         host: Optional[str] = None,
@@ -107,7 +108,7 @@ def __init__(
         self.read_from_replicas = read_from_replicas
         self._connection = None
 
-    def get_redis_connection(self):
+    def init_connection(self):
         """
         Connection is cached, so returning this
         """

aws_lambda_powertools/utilities/database/redis.py renamed to aws_lambda_powertools/utilities/connections/redis.py

@@ -41,7 +41,7 @@ def __init__(
         """
 
         # Initialize connection with Redis
-        self._connection = connection._init_connection()
+        self._connection = connection.init_connection()
 
         if static_pk_value is None:
             static_pk_value = f"idempotency#{os.getenv(constants.LAMBDA_FUNCTION_NAME_ENV, '')}"