nik/newdwh2021
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

コードの修正

Browse Source
This commit is contained in:
mori.k 2025-05-22 19:51:33 +09:00
parent 800f79b825
commit fa408d2a83
1 changed files with 65 additions and 52 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
ecs/jskult-batch/src/manager/jskult_batch_status_manager.py
View File

@ -1,7 +1,16 @@
import os
from src.error.exceptions import MaxRunCountReachedException
from src.db.database import Database
DB_USERNAME = os.environ.get("DB_USERNAME")
DB_PASSWORD = os.environ.get("DB_PASSWORD")
DB_HOST = os.environ.get("DB_HOST")
DB_PORT = os.environ.get("DB_PORT")
DB_SCHEMA = os.environ.get("DB_SCHEMA")
# 実消化&アルトマーク_バッチステータス管理テーブルを管理するクラス
class JskultBatchStatusManager:
def __init__(self, process_name: str, process_type: str, max_run_count_flg: int, receive_file_count: int):
@ -18,13 +27,16 @@ class JskultBatchStatusManager:
self._receive_file_count: str = receive_file_count
# DB接続モジュール(バッチ)のget_instanceを呼び出し設定
self._db = Database().get_instance()
self._db = Database(DB_USERNAME, DB_PASSWORD, DB_HOST,
DB_PORT, DB_SCHEMA).get_instance()
# 処理ステータスの登録および更新を行う
def set_process_status(self, process_status: str):
try:
self._db.begin()
self._db.execute(f"CALL upsert_jskult_batch_status_manage({self._process_name} {self._process_type} {process_status} NULL NULL);")
self._db.execute(
f"CALL upsert_jskult_batch_status_manage({self._process_name}, {self._process_type}, {process_status}, NULL, NULL);"
)
self._db.commit()
except Exception as e:
@ -88,8 +100,8 @@ class JskultBatchStatusManager:
# 最大起動回数に到達した場合にメッセージをスロー
raise MaxRunCountReachedException("最大起動回数に到達しました")
# アルトマークデータ連携があったかを確認する
def is_done_ultmarc_import(self):
# SELECTの結果からレコード数を取得
@ -113,7 +125,8 @@ class JskultBatchStatusManager:
run_count += record[0]['run_count']
self._db.execute(f"CALL upsert_jskult_batch_status_manage({self._process_name} {self._process_type} NULL {run_count} NULL);")
self._db.execute(
f"CALL upsert_jskult_batch_status_manage({self._process_name}, {self._process_type}, NULL, {run_count}, NULL);")
self._db.commit()
@ -152,9 +165,9 @@ class JskultBatchStatusManager:
# 全ての後続処理が完了している場合Trueを返す
return True
# データ取込処理が完了しているかを判定する
def _is_done_process(self, process_name : str):
def _is_done_process(self):
# SELECTの結果からレコード数を取得
record_count = self._db.execute_select(
@ -166,8 +179,8 @@ class JskultBatchStatusManager:
return True
# 起動回数が最大回数に到達しているか判定する
def _is_max_run_count_reached(self):
# SELECTの結果からレコード数を取得
@ -183,13 +196,13 @@ class JskultBatchStatusManager:
return False
def _activate_max_run_count_flg(self):
try:
self._db.begin()
# 最大起動回数フラグにフラグを立てる
self._db.execute(f"CALL upsert_jskult_batch_status_manage({self._process_name} {self._process_type} NULL NULL 1);")
self._db.execute(
f"CALL upsert_jskult_batch_status_manage({self._process_name}, {self._process_type}, NULL, NULL, 1);")
self._db.commit()