set pool size and max overflow (#26)

rciam · Oct 19, 2023 · f4d34c6 · f4d34c6
1 parent 815a684
commit f4d34c6
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/app/database.py b/app/database.py
@@ -5,9 +5,12 @@
 def get_session():
     # Initialize
     config_file = 'config.global.py'
+    db_params = configParser.getConfig('database_parameters', config_file)
 
-    url = configParser.getConfig('database_parameters', config_file)['database_url']
-    engine = create_engine(url)
+    url = db_params['database_url']
+    pool_size = int(db_params.get('pool_size', 25))
+    max_overflow = int(db_params.get('max_overflow', 5))
+    engine = create_engine(url, pool_size=pool_size, max_overflow=max_overflow)
 
     with Session(engine) as session:
         yield session