feat(api): pg_client reconnect if PG is not available

api/chalicelib/utils/pg_client.py

@@ -1,3 +1,4 @@
+import time
 from threading import Semaphore
 
 import psycopg2
@@ -37,9 +38,14 @@ class ORThreadedConnectionPool(psycopg2.pool.ThreadedConnectionPool):
 
 postgreSQL_pool: ORThreadedConnectionPool = None
 
+RETRY_MAX = config("PG_RETRY_MAX", cast=int, default=50)
+RETRY_INTERVAL = config("PG_RETRY_INTERVAL", cast=int, default=2)
+RETRY = 0
+
 
 def make_pool():
     global postgreSQL_pool
+    global RETRY
     if postgreSQL_pool is not None:
         try:
             postgreSQL_pool.closeall()
@@ -51,7 +57,13 @@ def make_pool():
             print("Connection pool created successfully")
     except (Exception, psycopg2.DatabaseError) as error:
         print("Error while connecting to PostgreSQL", error)
-        raise error
+        if RETRY < RETRY_MAX:
+            RETRY += 1
+            print(f"waiting for {RETRY_INTERVAL}s before retry n°{RETRY}")
+            time.sleep(RETRY_INTERVAL)
+            make_pool()
+        else:
+            raise error
 
 
 make_pool()