Added health check with check connection to db adapter

Author: ivan

core/db_adapter/aioredis_adapter.py

@@ -70,3 +70,9 @@ async def _path_exists(self, path):
 
     async def _on_prepare(self):
         pass
+
+    async def is_alive(self) -> bool | None:
+        try:
+            return self._redis.ping()
+        except Exception as _:
+            log("AIORedisAdapter is not alive", level="INFO")

core/db_adapter/db_adapter.py

@@ -91,6 +91,9 @@ def _on_all_tries_fail(self):
 class AsyncDBAdapter(DBAdapter):
     IS_ASYNC = True
 
+    async def is_alive(self) -> bool | None:
+        raise NotImplementedError
+
     async def _on_all_tries_fail(self):
         raise
 

core/db_adapter/ignite_adapter.py

@@ -88,3 +88,10 @@ def _on_prepare(self):
 
     async def _get_counter_name(self):
         return "ignite_async_adapter"
+
+    async def is_alive(self) -> bool | None:
+        try:
+            await self.connect()
+            return True
+        except Exception as _:
+            log("IgniteAdapter is not alive", level="INFO")

core/db_adapter/memory_adapter.py

@@ -49,3 +49,6 @@ async def _get(self, id):
 
     async def _list_dir(self, path):
         pass
+
+    async def is_alive(self) -> bool | None:
+        return True

smart_kit/start_points/main_loop_async_http.py

@@ -21,6 +21,7 @@ def __init__(self, *args, **kwargs):
         self.app = aiohttp.web.Application()
         super().__init__(*args, **kwargs)
         self.app.add_routes([aiohttp.web.route('*', '/health', self.get_health_check)])
+        self.app.add_routes([aiohttp.web.route('*', '/health_db_adapter', self.get_health_db_adapter_check)])
         self.app.add_routes([aiohttp.web.route('*', '/{tail:.*}', self.iterate)])
 
     async def async_init(self):
@@ -174,6 +175,21 @@ async def get_health_check(self, request: aiohttp.web.Request):
             status=status, reason=reason, data=answer,
         )
 
+    async def get_health_db_adapter_check(self, request: aiohttp.web.Request):
+        status, reason, answer = 200, "OK", "ok"
+        try:
+            is_alive = await self.db_adapter.is_alive()
+            if not is_alive:
+                status, reason, answer = 500, "ERROR", "db adapter connection error"
+        except NotImplementedError as _:
+            status, reason, answer = 500, "ERROR", f"Method is not implemented in {type(self.db_adapter)}"
+        except Exception as e:
+            status, reason, answer = 500, "ERROR", str(e)
+
+        return aiohttp.web.json_response(
+            status=status, reason=reason, data=answer,
+        )
+
     async def iterate(self, request: aiohttp.web.Request):
         headers = self._get_headers(request.headers)
         body = await request.text()