From 14e8dde249ae42233fad6bc9e0fea9e33262d3f5 Mon Sep 17 00:00:00 2001
From: Taha Yassine Kraiem <tahayk2@gmail.com>
Date: Thu, 7 Apr 2022 13:44:43 +0200
Subject: [PATCH] feat(api): pg_client reconnect if PG is not available

---
 api/chalicelib/utils/pg_client.py | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/api/chalicelib/utils/pg_client.py b/api/chalicelib/utils/pg_client.py
index 6379bad1e..3d60dda5c 100644
--- a/api/chalicelib/utils/pg_client.py
+++ b/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()