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

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

@ -1,10 +1,19 @@
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):
def __init__(self, process_name: str, process_type: str, max_run_count_flg: int, receive_file_count: int):
# 処理名
self._process_name: str = process_name
@ -18,43 +27,46 @@ 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):
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:
# Exceptionをcatchした場合はrollback
self._db.rollback()
raise e
raise e
# 後続処理を実行してよいか判定する
def can_run_post_process(self):
# SELECTの結果からレコード数を取得
record_count = self._db.execute_select(
f"SELECT * FROM internal07.jskult_batch_status_manage WHERE process_name = {self._process_name} AND process_date = src07.get_syor_date();"
f"SELECT * FROM internal07.jskult_batch_status_manage WHERE process_name = {self._process_name} AND process_date = src07.get_syor_date();"
).count()
if record_count == 0:
raise ValueError("レコードの取得が出来ませんでした。")
# 起動回数のインクリメント
self._increment_run_count()
# データ取込が完了していた場合
if self._is_done_data_import():
return True
#最大起動回数に到達していない場合
# 最大起動回数に到達していない場合
if not self._is_max_run_count_reached:
return False
# 最大起動回数フラグを立てる
self._activate_max_run_count_flg()
@ -65,41 +77,41 @@ class JskultBatchStatusManager:
# SELECTの結果からレコード数を取得
record_count = self._db.execute_select(
f"SELECT * FROM internal07.jskult_batch_status_manage WHERE process_name = {self._process_name} AND process_date = src07.get_syor_date();"
f"SELECT * FROM internal07.jskult_batch_status_manage WHERE process_name = {self._process_name} AND process_date = src07.get_syor_date();"
).count()
if record_count == 0:
raise ValueError("レコードの取得が出来ませんでした。")
# 起動回数のインクリメント
self._increment_run_count()
# 後続処理が完了していた場合
if self._is_done_post_process():
return True
#最大起動回数に到達していない場合
# 最大起動回数に到達していない場合
if not self._is_max_run_count_reached():
return False
# 最大起動回数フラグを立てる
self._activate_max_run_count_flg()
# 最大起動回数に到達した場合にメッセージをスロー
raise MaxRunCountReachedException("最大起動回数に到達しました")
# アルトマークデータ連携があったかを確認する
def is_done_ultmarc_import(self):
# SELECTの結果からレコード数を取得
record_count = self._db.execute_select(
f"SELECT * FROM internal07.jskult_batch_status_manage WHERE process_name = 'jskult-batch-ultmarc-io' AND process_date = src07.get_syor_date();"
f"SELECT * FROM internal07.jskult_batch_status_manage WHERE process_name = 'jskult-batch-ultmarc-io' AND process_date = src07.get_syor_date();"
).count()
# アルトマークデータ連携が無かった場合
if record_count == 0:
return False
return True
# 起動回数をインクリメントする
@ -108,12 +120,13 @@ class JskultBatchStatusManager:
self._db.begin()
# SELECTの結果からレコードを取得
record = self._db.execute_select(
f"SELECT * FROM internal07.jskult_batch_status_manage WHERE process_name = {self._process_name} AND process_date = src07.get_syor_date();"
f"SELECT * FROM internal07.jskult_batch_status_manage WHERE process_name = {self._process_name} AND process_date = src07.get_syor_date();"
)
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()
@ -122,78 +135,78 @@ class JskultBatchStatusManager:
# Exceptionをcatchした場合はrollbakc
self._db.rollback()
raise e
# データ取込処理が完了しているかを判定する
def _is_done_data_import(self):
# SELECTの結果からレコード数を取得
# SELECTの結果からレコード数を取得
record_count = self._db.execute_select(
f"SELECT * FROM internal07.jskult_batch_status_manage WHERE process_type = {self._process_type} AND process_status = 'done' AND process_date = src07.get_syor_date();"
f"SELECT * FROM internal07.jskult_batch_status_manage WHERE process_type = {self._process_type} AND process_status = 'done' AND process_date = src07.get_syor_date();"
).count()
# データ取込数が一致している場合
if(self._receive_file_count == record_count):
if (self._receive_file_count == record_count):
return True
return False
# 後続処理のすべての処理が完了しているかを判定する
def _is_done_post_process(self):
if not self._is_done_process("jskult-batch-trn-result-data-bio-lot"):
return False
if not self._is_done_process("jskult-batch-mst-inst"):
return False
if not self._is_done_process("jskult-batch-dcf-inst-merge-io"):
return False
# 全ての後続処理が完了している場合Trueを返す
return True
if not self._is_done_process("jskult-batch-trn-result-data-bio-lot"):
return False
if not self._is_done_process("jskult-batch-mst-inst"):
return False
if not self._is_done_process("jskult-batch-dcf-inst-merge-io"):
return False
# 全ての後続処理が完了している場合Trueを返す
return True
# データ取込処理が完了しているかを判定する
def _is_done_process(self, process_name : str):
def _is_done_process(self):
# SELECTの結果からレコード数を取得
record_count = self._db.execute_select(
f"SELECT * FROM internal07.jskult_batch_status_manage WHERE process_name = {self._process_name} AND process_status = 'done' AND process_date = src07.get_syor_date();"
f"SELECT * FROM internal07.jskult_batch_status_manage WHERE process_name = {self._process_name} AND process_status = 'done' AND process_date = src07.get_syor_date();"
).count()
if(record_count == 0):
if (record_count == 0):
return False
return True
# 起動回数が最大回数に到達しているか判定する
def _is_max_run_count_reached(self):
# SELECTの結果からレコード数を取得
record = self._db.execute_select(
f"SELECT * FROM internal07.jskult_batch_status_manage WHERE process_name = {self._process_name} AND process_date = src07.get_syor_date();"
)
)
run_count = record[0]['run_count']
# 取得した起動回数とフィールド変数の最大起動回数が一致するか確認
if run_count == self._max_run_count_flg:
return True
return False
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()
except Exception as e:
self._db.rollback()
raise e
raise e