adding connection params for to prevent failed connections

Police-Data-Accessibility-Project · Feb 27, 2024 · 5fc1510 · 5fc1510
1 parent 7cc573f
commit 5fc1510
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/middleware/initialize_psycopg2_connection.py b/middleware/initialize_psycopg2_connection.py
@@ -6,7 +6,13 @@ def initialize_psycopg2_connection():
     try:
         DO_DATABASE_URL = os.getenv("DO_DATABASE_URL")
 
-        return psycopg2.connect(DO_DATABASE_URL)
+        return psycopg2.connect(
+            DO_DATABASE_URL,
+            keepalives=1,
+            keepalives_idle=30,
+            keepalives_interval=10,
+            keepalives_count=5,
+        )
 
     except:
         print("Error while initializing the DigitalOcean client with psycopg2.")