From 6b6f07a95fa250e796b0eb8cdd2af004f343625e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fran=C3=A7ois?= <francois@schmidts.fr>
Date: Sun, 16 Jun 2024 21:17:23 +0200
Subject: [PATCH] Fine tuning db connection (#260)

---
 jarr/bootstrap.py | 10 +++++++++-
 jarr/metaconf.yml |  6 ++++++
 2 files changed, 15 insertions(+), 1 deletion(-)

diff --git a/jarr/bootstrap.py b/jarr/bootstrap.py
index c5910f56..5314dd5d 100644
--- a/jarr/bootstrap.py
+++ b/jarr/bootstrap.py
@@ -39,7 +39,15 @@ def init_logging(log_path=None, log_level=logging.INFO, modules=(),
 
 def init_db(echo=False):
     mapper_registry = registry()
-    new_engine = create_engine(conf.db.pg_uri, echo=echo)
+    new_engine = create_engine(
+        conf.db.pg_uri,
+        echo=echo,
+        pool_size=conf.db.postgres.pool_size,
+        max_overflow=conf.db.postgres.max_overflow,
+        pool_recycle=conf.db.postgres.pool_recycle,
+        pool_pre_ping=conf.db.postgres.pool_pre_ping,
+        pool_use_lifo=conf.db.postgres.pool_use_lifo,
+    )
     NewBase = mapper_registry.generate_base()
     new_session = scoped_session(sessionmaker(bind=new_engine))
     return mapper_registry, new_engine, new_session, NewBase
diff --git a/jarr/metaconf.yml b/jarr/metaconf.yml
index a7e5acf0..191bae0a 100644
--- a/jarr/metaconf.yml
+++ b/jarr/metaconf.yml
@@ -116,6 +116,12 @@ parameters:
   - server_name: {'default': '', 'type': 'str'}
 - db:
   - pg_uri: {'default': 'postgresql://postgresql/jarr'}
+  - postgres:
+    - pool_size: {'default': 10}
+    - max_overflow: {'default': 2}
+    - pool_recycle: {'default': 300}
+    - pool_pre_ping: {'default': true}
+    - pool_use_lifo: {'default': true}
   - redis:
     - host: {'default': 'redis'}
     - db: {'default': 1}