fix: added db retry mechanism to mcpgateway/db.py (#226)

* fix: added db retry mechanism to mcpgateway/db.py

Signed-off-by: reevebarreto <[email protected]>

* cleaned code to a one liner

Signed-off-by: reevebarreto <[email protected]>

* Flake8 fix

Signed-off-by: Mihai Criveti <[email protected]>

---------

Signed-off-by: reevebarreto <[email protected]>
Signed-off-by: Mihai Criveti <[email protected]>
Co-authored-by: Mihai Criveti <[email protected]>

Author: reevebarreto

mcpgateway/db.py

@@ -25,6 +25,7 @@
 from mcpgateway.config import settings
 from mcpgateway.types import ResourceContent
 from mcpgateway.utils.create_slug import slugify
+from mcpgateway.utils.db_isready import wait_for_db_ready
 
 # Third-Party
 import jsonschema
@@ -1220,4 +1221,7 @@ def init_db():
 
 
 if __name__ == "__main__":
+    # Wait for database to be ready before initializing
+    wait_for_db_ready(max_tries=int(settings.db_max_retries), interval=int(settings.db_retry_interval_ms) / 1000, sync=True)  # Converting ms to s
+
     init_db()