63 lines
2.8 KiB
Python
63 lines
2.8 KiB
Python
"""実消化＆アルトマークダンプ復元スクリプト"""
|
|
|
|
import os
|
|
import subprocess
|
|
import textwrap
|
|
|
|
from src.logging.get_logger import get_logger
|
|
from src.system_var import constants, environment
|
|
|
|
logger = get_logger('実消化＆アルトマークダンプ復元スクリプト')
|
|
|
|
|
|
def exec():
|
|
try:
|
|
logger.info('実消化＆アルトマークダンプ復元スクリプト：開始')
|
|
|
|
# MySQL接続情報を作成する
|
|
my_cnf_file_content = f"""
|
|
[client]
|
|
user={environment.DB_USERNAME}
|
|
password={environment.DB_PASSWORD}
|
|
host={environment.DB_HOST}
|
|
"""
|
|
# my.cnfファイルのパス
|
|
my_cnf_path = os.path.join('my.cnf')
|
|
|
|
# my.cnfファイルを生成する
|
|
with open(my_cnf_path, 'w') as f:
|
|
f.write(textwrap.dedent(my_cnf_file_content)[1:-1])
|
|
|
|
# 復元対象のダンプファイルを特定
|
|
s3_file_path = environment.JSKULT_BACKUP_DUMP_S3_FILE_PATH
|
|
|
|
# aws s3 cpコマンドを実行してdumpファイルをローカルにダウンロードする
|
|
s3_cp_process = subprocess.Popen(['aws', 's3', 'cp', s3_file_path, './dump.gz'], stderr=subprocess.PIPE)
|
|
_, error = s3_cp_process.communicate()
|
|
if s3_cp_process.returncode != 0:
|
|
logger.error(f'`aws s3 cp`実行時にエラーが発生しました。{"" if error is None else error.decode("utf-8")}')
|
|
return constants.BATCH_EXIT_CODE_SUCCESS
|
|
# S3コマンドの標準エラーはクローズしておく
|
|
s3_cp_process.stderr.close()
|
|
|
|
# gzipコマンドを実行してdumpファイルを解凍する
|
|
gzip_process = subprocess.Popen(['gunzip', '-c', './dump.gz'], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
|
# mysqlコマンドを実行し、dumpを復元する
|
|
mysql_process = subprocess.Popen(
|
|
['mysql', f'--defaults-file={my_cnf_path}', '-P', f"{environment.DB_PORT}", environment.DB_SCHEMA],
|
|
stdin=gzip_process.stdout, stderr=subprocess.PIPE
|
|
)
|
|
# gzipの標準出力をmysqlに接続したため、標準出力をクローズする
|
|
gzip_process.stdout.close()
|
|
|
|
_, error = mysql_process.communicate()
|
|
if mysql_process.returncode != 0:
|
|
logger.error(f'コマンド実行時にエラーが発生しました。{"" if error is None else error.decode("utf-8")}')
|
|
return constants.BATCH_EXIT_CODE_SUCCESS
|
|
logger.info('[NOTICE]実消化＆アルトマークダンプ復元スクリプト：終了（正常終了）')
|
|
return constants.BATCH_EXIT_CODE_SUCCESS
|
|
|
|
except Exception as e:
|
|
logger.exception(f'実消化＆アルトマークダンプ復元スクリプト中に想定外のエラーが発生しました :{e}')
|
|
return constants.BATCH_EXIT_CODE_SUCCESS
|