From f4d34c64c7bf4af68222846c9e9934e2614dd901 Mon Sep 17 00:00:00 2001
From: lionick <44701697+lionick@users.noreply.github.com>
Date: Thu, 19 Oct 2023 10:11:02 +0300
Subject: [PATCH] set pool size and max overflow (#26)

---
 app/database.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/app/database.py b/app/database.py
index b67e276..c547c36 100644
--- 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