Increase initial Hasura connection retry count (#857)

Co-authored-by: Vladimir Bobrikov <[email protected]>
Co-authored-by: Lev Gorodetskiy <[email protected]>

Author: 3 people

CHANGELOG.md

@@ -12,11 +12,11 @@ The format is based on [Keep a Changelog], and this project adheres to [Semantic
 
 ### Fixed
 
-
-- cli: Added warning for timescaledb-ha users
+- cli: Use correct data path with timescaledb-ha Docker image.
 - demos: Fixed decimal overflow in `demo_uniswap` project.
 - evm.node: Fixed incorrect log request parameters.
 - evm.subsquid.events: Fixed issue with determining the last level when syncing with node.
+- hasura: Increated retry count for initial connection (healthcheck).
 
 ## [7.0.0] - 2023-09-25
 

src/dipdup/hasura.py

@@ -210,9 +210,11 @@ def _get_source(self, metadata: dict[str, Any], name: str) -> dict[str, Any] | N
         else:
             return None
 
-    async def _hasura_request(self, endpoint: str, json: dict[str, Any] | None = None) -> dict[str, Any]:
+    async def _hasura_request(self, endpoint: str, json: dict[str, Any] | None = None, retry_count: int | None = None) -> dict[str, Any]:
         self._logger.debug('Sending `%s` request: %s', endpoint, orjson.dumps(json))
         try:
+            if retry_count is not None:
+                self._http_config.retry_count, retry_count = retry_count, self._http_config.retry_count
             result = await self.request(
                 method='get' if json is None else 'post',
                 url=f'v1/{endpoint}',
@@ -221,6 +223,9 @@ async def _hasura_request(self, endpoint: str, json: dict[str, Any] | None = Non
             )
         except ClientResponseError as e:
             raise HasuraError(f'{e.status} {e.message}') from e
+        finally:
+            if retry_count is not None:
+                self._http_config.retry_count, retry_count = retry_count, self._http_config.retry_count
 
         self._logger.debug('Response: %s', result)
         if errors := result.get('error') or result.get('errors'):
@@ -230,7 +235,7 @@ async def _hasura_request(self, endpoint: str, json: dict[str, Any] | None = Non
 
     async def _healthcheck(self) -> None:
         self._logger.info('Connecting to Hasura instance')
-        version_json = await self._hasura_request('version')
+        version_json = await self._hasura_request('version', retry_count=20)
         version = version_json['version']
         if version.startswith('v1'):
             raise UnsupportedAPIError(