212 lines
9.3 KiB
Python
Executable File
212 lines
9.3 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
from my_error import MyError
|
|
import sys
|
|
import mysql.connector
|
|
from mysql.connector import errorcode
|
|
import logging
|
|
import json
|
|
import config
|
|
import opts
|
|
import run_modules
|
|
import actions
|
|
import normal_ddl_actions_post
|
|
import normal_dml_actions_post
|
|
import each_tenant_dml_actions_post
|
|
import each_tenant_ddl_actions_post
|
|
import special_upgrade_action_post
|
|
|
|
# 由于用了/*+read_consistency(WEAK) */来查询，因此升级期间不能允许创建或删除租户
|
|
|
|
class UpgradeParams:
|
|
log_filename = config.post_upgrade_log_filename
|
|
sql_dump_filename = config.post_upgrade_sql_filename
|
|
rollback_sql_filename = config.post_upgrade_rollback_sql_filename
|
|
|
|
def config_logging_module(log_filenamme):
|
|
logging.basicConfig(level=logging.INFO,\
|
|
format='[%(asctime)s] %(levelname)s %(filename)s:%(lineno)d %(message)s',\
|
|
datefmt='%Y-%m-%d %H:%M:%S',\
|
|
filename=log_filenamme,\
|
|
filemode='w')
|
|
# 定义日志打印格式
|
|
formatter = logging.Formatter('[%(asctime)s] %(levelname)s %(filename)s:%(lineno)d %(message)s', '%Y-%m-%d %H:%M:%S')
|
|
#######################################
|
|
# 定义一个Handler打印INFO及以上级别的日志到sys.stdout
|
|
stdout_handler = logging.StreamHandler(sys.stdout)
|
|
stdout_handler.setLevel(logging.INFO)
|
|
# 设置日志打印格式
|
|
stdout_handler.setFormatter(formatter)
|
|
# 将定义好的stdout_handler日志handler添加到root logger
|
|
logging.getLogger('').addHandler(stdout_handler)
|
|
|
|
|
|
def dump_sql_to_file(dump_filename, tenant_id_list):
|
|
normal_ddls_str = normal_ddl_actions_post.get_normal_ddl_actions_sqls_str()
|
|
normal_dmls_str = normal_dml_actions_post.get_normal_dml_actions_sqls_str()
|
|
each_tenant_dmls_str = each_tenant_dml_actions_post.get_each_tenant_dml_actions_sqls_str(tenant_id_list)
|
|
dump_file = open(dump_filename, 'w')
|
|
dump_file.write('# 以下是upgrade_post.py脚本中的步骤\n')
|
|
dump_file.write('# 仅供upgrade_post.py脚本运行失败需要人肉的时候参考\n')
|
|
dump_file.write('\n\n')
|
|
dump_file.write('# normal ddl\n')
|
|
dump_file.write(normal_ddls_str + '\n')
|
|
dump_file.write('\n\n')
|
|
dump_file.write('# normal dml\n')
|
|
dump_file.write(normal_dmls_str + '\n')
|
|
dump_file.write('\n\n')
|
|
dump_file.write('# each tenant dml\n')
|
|
dump_file.write(each_tenant_dmls_str + '\n')
|
|
dump_file.write('\n\n')
|
|
dump_file.write('# do special upgrade actions\n')
|
|
dump_file.write('# please run ./upgrade_post.py -h [host] -P [port] -u [user] -p [password] -m special_action\n')
|
|
dump_file.write('\n\n')
|
|
dump_file.close()
|
|
|
|
def check_before_upgrade(query_cur, upgrade_params):
|
|
check_server_version(query_cur)
|
|
return
|
|
|
|
# 混部阶段执行POST脚本，会导致租户级系统表创建可能多数派处于旧binary而被GC,
|
|
# 需要规避OCP升级流程异常导致的混部阶段执行POST脚本的问题
|
|
def check_server_version(query_cur):
|
|
sql = """select distinct(substring_index(build_version, '_', 1)) from __all_server""";
|
|
(desc, results) = query_cur.exec_query(sql);
|
|
if len(results) != 1:
|
|
raise MyError("servers build_version not match")
|
|
else:
|
|
logging.info("check server version success")
|
|
|
|
def print_stats():
|
|
logging.info('==================================================================================')
|
|
logging.info('============================== STATISTICS BEGIN ==================================')
|
|
logging.info('==================================================================================')
|
|
logging.info('succeed run sql(except sql of special actions): \n\n%s\n', actions.get_succ_sql_list_str())
|
|
logging.info('commited sql(except sql of special actions): \n\n%s\n', actions.get_commit_sql_list_str())
|
|
logging.info('==================================================================================')
|
|
logging.info('=============================== STATISTICS END ===================================')
|
|
logging.info('==================================================================================')
|
|
|
|
def do_upgrade(my_host, my_port, my_user, my_passwd, my_module_set, upgrade_params):
|
|
try:
|
|
conn = mysql.connector.connect(user = my_user,
|
|
password = my_passwd,
|
|
host = my_host,
|
|
port = my_port,
|
|
database = 'oceanbase',
|
|
raise_on_warnings = True)
|
|
cur = conn.cursor(buffered=True)
|
|
try:
|
|
query_cur = actions.QueryCursor(cur)
|
|
# 开始升级前的检查
|
|
check_before_upgrade(query_cur, upgrade_params)
|
|
# 获取租户id列表
|
|
tenant_id_list = actions.fetch_tenant_ids(query_cur)
|
|
if len(tenant_id_list) <= 0:
|
|
logging.error('distinct tenant id count is <= 0, tenant_id_count: %d', len(tenant_id_list))
|
|
raise MyError('no tenant id')
|
|
logging.info('there has %s distinct tenant ids: [%s]', len(tenant_id_list), ','.join(str(tenant_id) for tenant_id in tenant_id_list))
|
|
conn.commit()
|
|
|
|
actions.refresh_commit_sql_list()
|
|
dump_sql_to_file(upgrade_params.sql_dump_filename, tenant_id_list)
|
|
logging.info('================succeed to dump sql to file: {0}==============='.format(upgrade_params.sql_dump_filename))
|
|
|
|
if run_modules.MODULE_DDL in my_module_set:
|
|
logging.info('================begin to run ddl===============')
|
|
conn.autocommit = True
|
|
normal_ddl_actions_post.do_normal_ddl_actions(cur)
|
|
logging.info('================succeed to run ddl===============')
|
|
conn.autocommit = False
|
|
|
|
if run_modules.MODULE_EACH_TENANT_DDL in my_module_set:
|
|
has_run_ddl = True
|
|
logging.info('================begin to run each tenant ddl===============')
|
|
conn.autocommit = True
|
|
each_tenant_ddl_actions_post.do_each_tenant_ddl_actions(cur, tenant_id_list)
|
|
logging.info('================succeed to run each tenant ddl===============')
|
|
conn.autocommit = False
|
|
|
|
if run_modules.MODULE_NORMAL_DML in my_module_set:
|
|
logging.info('================begin to run normal dml===============')
|
|
normal_dml_actions_post.do_normal_dml_actions(cur)
|
|
logging.info('================succeed to run normal dml===============')
|
|
conn.commit()
|
|
actions.refresh_commit_sql_list()
|
|
logging.info('================succeed to commit dml===============')
|
|
|
|
if run_modules.MODULE_EACH_TENANT_DML in my_module_set:
|
|
logging.info('================begin to run each tenant dml===============')
|
|
conn.autocommit = True
|
|
each_tenant_dml_actions_post.do_each_tenant_dml_actions(cur, tenant_id_list)
|
|
conn.autocommit = False
|
|
logging.info('================succeed to run each tenant dml===============')
|
|
|
|
if run_modules.MODULE_SPECIAL_ACTION in my_module_set:
|
|
logging.info('================begin to run special action===============')
|
|
conn.autocommit = True
|
|
special_upgrade_action_post.do_special_upgrade(conn, cur, tenant_id_list, my_user, my_passwd)
|
|
conn.autocommit = False
|
|
actions.refresh_commit_sql_list()
|
|
logging.info('================succeed to commit special action===============')
|
|
except Exception, e:
|
|
logging.exception('run error')
|
|
raise e
|
|
finally:
|
|
# 打印统计信息
|
|
print_stats()
|
|
# 将回滚sql写到文件中
|
|
actions.dump_rollback_sql_to_file(upgrade_params.rollback_sql_filename)
|
|
cur.close()
|
|
conn.close()
|
|
except mysql.connector.Error, e:
|
|
logging.exception('connection error')
|
|
raise e
|
|
except Exception, e:
|
|
logging.exception('normal error')
|
|
raise e
|
|
|
|
def do_upgrade_by_argv(argv):
|
|
upgrade_params = UpgradeParams()
|
|
opts.change_opt_defult_value('log-file', upgrade_params.log_filename)
|
|
opts.parse_options(argv)
|
|
if not opts.has_no_local_opts():
|
|
opts.deal_with_local_opts()
|
|
else:
|
|
opts.check_db_client_opts()
|
|
log_filename = opts.get_opt_log_file()
|
|
upgrade_params.log_filename = log_filename
|
|
# 日志配置放在这里是为了前面的操作不要覆盖掉日志文件
|
|
config_logging_module(upgrade_params.log_filename)
|
|
try:
|
|
host = opts.get_opt_host()
|
|
port = int(opts.get_opt_port())
|
|
user = opts.get_opt_user()
|
|
password = opts.get_opt_password()
|
|
cmd_module_str = opts.get_opt_module()
|
|
module_set = set([])
|
|
all_module_set = run_modules.get_all_module_set()
|
|
cmd_module_list = cmd_module_str.split(',')
|
|
for cmd_module in cmd_module_list:
|
|
if run_modules.ALL_MODULE == cmd_module:
|
|
module_set = module_set | all_module_set
|
|
elif cmd_module in all_module_set:
|
|
module_set.add(cmd_module)
|
|
else:
|
|
raise MyError('invalid module: {0}'.format(cmd_module))
|
|
logging.info('parameters from cmd: host=\"%s\", port=%s, user=\"%s\", password=\"%s\", module=\"%s\", log-file=\"%s\"',\
|
|
host, port, user, password, module_set, log_filename)
|
|
do_upgrade(host, port, user, password, module_set, upgrade_params)
|
|
except mysql.connector.Error, e:
|
|
logging.exception('mysql connctor error')
|
|
logging.exception('run error, maybe you can reference ' + upgrade_params.rollback_sql_filename + ' to rollback it')
|
|
raise e
|
|
except Exception, e:
|
|
logging.exception('normal error')
|
|
logging.exception('run error, maybe you can reference ' + upgrade_params.rollback_sql_filename + ' to rollback it')
|
|
raise e
|
|
|
|
|
|
|