From e87da068bb9986f1bac482cde2cd0ecd6013fc87 Mon Sep 17 00:00:00 2001
From: "shimoda.m@nds-tyo.co.jp" <shimoda.m@nds-tyo.co.jp>
Date: Tue, 11 Jul 2023 13:03:54 +0900
Subject: [PATCH] =?UTF-8?q?feat:=20DB=E5=86=8D=E8=B5=B7=E5=8B=95=E3=80=81?=
 =?UTF-8?q?=E6=99=82=E9=96=93=E7=B5=8C=E9=81=8E=E5=BE=8C=E3=81=AE=E3=82=B3?=
 =?UTF-8?q?=E3=83=8D=E3=82=AF=E3=82=B7=E3=83=A7=E3=83=B3=E5=86=8D=E7=A2=BA?=
 =?UTF-8?q?=E7=AB=8B=E3=81=AE=E4=BB=95=E8=BE=BC=E3=81=BF?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ecs/jskult-webapp/src/db/database.py | 20 +++++++++++++++++++-
 1 file changed, 19 insertions(+), 1 deletion(-)

diff --git a/ecs/jskult-webapp/src/db/database.py b/ecs/jskult-webapp/src/db/database.py
index 85759f96..f700c016 100644
--- a/ecs/jskult-webapp/src/db/database.py
+++ b/ecs/jskult-webapp/src/db/database.py
@@ -1,10 +1,28 @@
 from sqlalchemy import (Connection, CursorResult, Engine, QueuePool,
-                        create_engine, text)
+                        create_engine, event, exc, text)
 from sqlalchemy.engine.url import URL
+from sqlalchemy.pool import Pool
 
 from src.error.exceptions import DBException
+from src.logging.get_logger import get_logger
 from src.system_var import environment
 
+logger = get_logger('DB接続')
+
+
+@event.listens_for(Pool, 'checkout')
+def ping_connection(dbapi_connection, connection_record, connection_proxy):
+    """コネクションが切れた場合、再接続"""
+    cursor = dbapi_connection.cursor()
+    try:
+        cursor.execute("SELECT 1")
+    except Exception as e:
+        logger.info(f'DB接続に失敗したため、リトライします: {e}')
+        # raise DisconnectionError - pool will try
+        # connecting again up to three times before raising.
+        raise exc.DisconnectionError()
+    cursor.close()
+
 
 class Database:
     """データベース操作クラス"""