feat: データベース操作部品を修正。パラメータ指定により、AUTOCOMMITで接続するかどうかを分岐するように修正

ecs/jskult-batch-daily/src/batch/laundering/sales_results_laundering.py

@@ -1,14 +1,14 @@
+from src.batch.batch_functions import logging_sql
 from src.db.database import Database
 from src.error.exceptions import BatchOperationException
 from src.logging.get_logger import get_logger
-from src.batch.batch_functions import logging_sql
 from src.system_var import environment
 
 logger = get_logger('卸卸販売洗替')
 
 
 def exec():
-    db = Database.get_instance()
+    db = Database.get_instance(autocommit=True)
     try:
         db.connect()
         logger.debug('処理開始')

ecs/jskult-batch-daily/src/db/database.py

@@ -13,15 +13,17 @@ logger = get_logger(__name__)
 class Database:
     """データベース操作クラス"""
     __connection: Connection = None
-    __engine: Engine = None
+    __transactional_engine: Engine = None
+    __autocommit_engine: Engine = None
     __host: str = None
     __port: str = None
     __username: str = None
     __password: str = None
     __schema: str = None
+    __autocommit: bool = None
     __connection_string: str = None
 
-    def __init__(self, username: str, password: str, host: str, port: int, schema: str) -> None:
+    def __init__(self, username: str, password: str, host: str, port: int, schema: str, autocommit: bool = False) -> None:
         """このクラスの新たなインスタンスを初期化します
 
         Args:
@@ -30,12 +32,14 @@ class Database:
             host (str): DBホスト名
             port (int): DBポート
             schema (str): DBスキーマ名
+            autocommit(bool): 自動コミットモードで接続するかどうか(Trueの場合、トランザクションの有無に限らず即座にコミットされる). Defaults to False.
         """
         self.__username = username
         self.__password = password
         self.__host = host
         self.__port = int(port)
         self.__schema = schema
+        self.__autocommit = autocommit
 
         self.__connection_string = URL.create(
             drivername='mysql+pymysql',
@@ -47,16 +51,20 @@ class Database:
             query={"charset": "utf8mb4"}
         )
 
-        self.__engine = create_engine(
+        self.__transactional_engine = create_engine(
             self.__connection_string,
             pool_timeout=5,
             poolclass=QueuePool
         )
 
+        self.__autocommit_engine = self.__transactional_engine.execution_options(isolation_level='AUTOCOMMIT')
+
     @classmethod
-    def get_instance(cls):
+    def get_instance(cls, autocommit=False):
         """インスタンスを取得します
 
+        Args:
+            autocommit (bool, optional): 自動コミットモードで接続するかどうか(Trueの場合、トランザクションの有無に限らず即座にコミットされる). Defaults to False.
         Returns:
             Database: DB操作クラスインスタンス
         """
@@ -65,7 +73,8 @@ class Database:
             password=environment.DB_PASSWORD,
             host=environment.DB_HOST,
             port=environment.DB_PORT,
-            schema=environment.DB_SCHEMA
+            schema=environment.DB_SCHEMA,
+            autocommit=autocommit
         )
 
     @retry(
@@ -77,12 +86,15 @@ class Database:
         stop=stop_after_attempt(environment.DB_CONNECTION_MAX_RETRY_ATTEMPT))
     def connect(self):
         """
-        DBに接続します。接続に失敗した場合、リトライします。
+        DBに接続します。接続に失敗した場合、リトライします。\n
+        インスタンスのautocommitがTrueの場合、自動コミットモードで接続する。（明示的なトランザクションも無視される）
         Raises:
             DBException: 接続失敗
         """
         try:
-            self.__connection = self.__engine.connect()
+            self.__connection = (
+                self.__autocommit_engine.connect() if self.__autocommit is True
+                else self.__transactional_engine.connect())
         except Exception as e:
             raise DBException(e)