added retryhandler

Author: PraneethChandraThota

socketlabs/injectionapi/core/retryhandler.py

@@ -0,0 +1,25 @@
+from http import HTTPStatus
+from ..retrysettings import RetrySettings
+from .httprequest import HttpRequest
+class RetryHandler(object):
+
+    ErrorStatusCodes = [
+        HTTPStatus.INTERNAL_SERVER_ERROR,
+        HTTPStatus.BAD_GATEWAY,
+        HTTPStatus.SERVICE_UNAVAILABLE,
+        HTTPStatus.GATEWAY_TIMEOUT
+    ]
+
+    def __init__(self, http_client: HttpRequest, settings: RetrySettings):
+        
+        self.__http_client = http_client
+        self.__retry_settings = settings
+    
+    def send(self, body):
+
+        if self.__retry_settings.maximum_number_of_retries == 0:
+            return self.__http_client.send_request(body)
+
+        attempts = 0
+
+        return

socketlabs/injectionapi/retrysettings.py

@@ -10,14 +10,16 @@ class RetrySettings(object):
 
     def __init__(self, maximum_retries = None):
 
-        
-
         if maximum_retries:
+
             if maximum_retries < 0:
                 raise ValueError("maximumNumberOfRetries must be greater than 0")
+
             if maximum_retries > self.__maximum_allowed_number_of_retries:
                 raise ValueError("The maximum number of allowed retries is ", self.__maximum_allowed_number_of_retries)
+
             self.__maximum_number_of_retries = maximum_retries
+            
         else:
             self.__maximum_number_of_retries = self.__default_number_of_retries
 

socketlabs/injectionapi/socketlabsclient.py

@@ -6,6 +6,8 @@
 from .core.httprequest import HttpRequest
 from .core.injectionrequestfactory import InjectionRequestFactory
 from .core.sendvalidator import SendValidator
+from .core.retryhandler import RetryHandler
+from .retrysettings import RetrySettings
 from .message.basicmessage import BasicMessage
 from .message.bulkmessage import BulkMessage
 from .proxy import Proxy
@@ -32,6 +34,7 @@ def __init__(self, server_id: int, api_key: str, proxy: Proxy = None):
         self._api_key = api_key
         self._http_proxy = proxy
         self._request_timeout = 120
+        self._number_of_retries = 0
 
     @property
     def __endpoint(self):
@@ -69,6 +72,14 @@ def request_timeout(self, timeout: int):
         """
         self._request_timeout = timeout
 
+    @property
+    def number_of_retries(self):
+        return self._number_of_retries
+    
+    @number_of_retries.setter
+    def number_of_retries(self, retries: int):
+        self._number_of_retries = retries
+
     def __build_http_request(self):
         """
         Build the HttpRequest. Will add the proxy, if set
@@ -111,7 +122,8 @@ def __send_basic_message(self, message: BasicMessage):
         body = req_factory.generate_request(message)
 
         request = self.__build_http_request()
-        result = request.send_request(body)
+        retry_handler = RetryHandler(request, RetrySettings(self.number_of_retries))
+        result = retry_handler.send(body)
         return result
 
     def __send_bulk_message(self, message: BulkMessage):
@@ -130,7 +142,8 @@ def __send_bulk_message(self, message: BulkMessage):
         body = req_factory.generate_request(message)
 
         request = self.__build_http_request()
-        result = request.send_request(body)
+        retry_handler = RetryHandler(request, RetrySettings(self.number_of_retries))
+        result = retry_handler.send(body)
         return result
 
     def send_async(self, message: BasicMessage, on_success, on_error):