feat: adds backoff to enterprise-catalog client calls

enterprise_access/apps/api_client/constants.py

@@ -0,0 +1,16 @@
+"""
+Constants for API client
+"""
+import backoff
+from requests.exceptions import ConnectionError as RequestsConnectionError
+from requests.exceptions import HTTPError
+from requests.exceptions import Timeout as RequestsTimeoutError
+
+# wait_gen references the field in the backoff decorator representing
+# the backoff strategy for retries
+wait_gen = backoff.expo
+
+exception = (
+    RequestsConnectionError,
+    RequestsTimeoutError,
+)

enterprise_access/apps/api_client/enterprise_catalog_client.py

@@ -1,10 +1,11 @@
 """
 API client for enterprise-catalog service.
 """
-
+import backoff
 from django.conf import settings
 
 from enterprise_access.apps.api_client.base_oauth import BaseOAuthClient
+from enterprise_access.apps.api_client.constants import exception, wait_gen
 
 
 class EnterpriseCatalogApiClient(BaseOAuthClient):
@@ -14,6 +15,7 @@ class EnterpriseCatalogApiClient(BaseOAuthClient):
     api_base_url = settings.ENTERPRISE_CATALOG_URL + '/api/v1/'
     enterprise_catalog_endpoint = api_base_url + 'enterprise-catalogs/'
 
+    @backoff.on_exception(wait_gen, exception)
     def contains_content_items(self, catalog_uuid, content_ids):
         """
         Check whether the specified enterprise catalog contains the given content.
@@ -31,8 +33,10 @@ def contains_content_items(self, catalog_uuid, content_ids):
         response = self.client.get(endpoint, params=query_params)
         response.raise_for_status()
         response_json = response.json()
+        print(response_json)
         return response_json.get('contains_content_items', False)
 
+    @backoff.on_exception(wait_gen, exception)
     def catalog_content_metadata(self, catalog_uuid, content_keys, traverse_pagination=True, **kwargs):
         """
         Returns a list of requested content metadata records for the given catalog_uuid.
@@ -68,6 +72,7 @@ def catalog_content_metadata(self, catalog_uuid, content_keys, traverse_paginati
         response.raise_for_status()
         return response.json()
 
+    @backoff.on_exception(wait_gen, exception)
     def get_content_metadata_count(self, catalog_uuid):
         """
         Returns the count of content metadata for a catalog.