def delete(self, cluster_id):
"""Delete cluster
It is automatically backed up with timestamps as tags
:param cluster_id: target cluster id
"""
if not cluster_util.validate_id(cluster_id):
raise ClusterIdError(cluster_id)
center = Center()
center.update_ip_port()
success = center.check_hosts_connection()
if success:
center.stop_redis(force=True)
path_of_fb = config.get_path_of_fb(cluster_id)
props_path = path_of_fb['redis_properties']
hosts = config.get_props(props_path, 'sr2_redis_master_hosts', [])
if not center.check_include_localhost(hosts):
hosts += [config.get_local_ip()]
tag = time.strftime("%Y%m%d%H%M%S", time.gmtime())
cluster_backup_dir = 'cluster_{}_bak_{}'.format(cluster_id, tag)
for host in hosts:
center.cluster_backup(host, cluster_id, cluster_backup_dir)
msg = message.get('cluster_delete_complete')
msg = msg.format(cluster_id=cluster_id)
logger.info(msg)
def run_deploy(cluster_id=None,
history_save=True,
clean=False,
strategy="none"):
"""Install LightningDB package.
:param cluster_id: cluster id
:param history_save: save input history and use as default
:param clean: delete redis log, node configuration
:param strategy:
none(default): normal deploy,
zero-downtime: re-deploy without stop
"""
# validate cluster id
if cluster_id is None:
cluster_id = config.get_cur_cluster_id(allow_empty_id=True)
if cluster_id < 0:
msg = message.get('error_invalid_cluster_on_deploy')
logger.error(msg)
return
if not cluster_util.validate_id(cluster_id):
raise ClusterIdError(cluster_id)
# validate option
if not isinstance(history_save, bool):
msg = message.get('error_option_type_not_boolean')
msg = msg.format(option='history-save')
logger.error(msg)
return
logger.debug("option '--history-save': {}".format(history_save))
if not isinstance(clean, bool):
msg = message.get('error_option_type_not_boolean')
msg = msg.format(option='clean')
logger.error(msg)
return
logger.debug("option '--clean': {}".format(clean))
strategy_list = ["none", "zero-downtime"]
if strategy not in strategy_list:
msg = message.get('error_deploy_strategy').format(value=strategy,
list=strategy_list)
logger.error(msg)
return
if strategy == "zero-downtime":
run_cluster_use(cluster_id)
_deploy_zero_downtime(cluster_id)
return
_deploy(cluster_id, history_save, clean)
def delete(self, cluster_id):
"""Delete cluster
It is automatically backed up with timestamps as tags
:param cluster_id: target cluster id
"""
if not cluster_util.validate_id(cluster_id):
raise ClusterIdError(cluster_id)
path_of_fb = config.get_path_of_fb(cluster_id)
props_path = path_of_fb['redis_properties']
hosts = config.get_props(props_path, 'sr2_redis_master_hosts', [])
tag = time.strftime("%Y%m%d%H%M%S", time.gmtime())
cluster_backup_dir = 'cluster_{}_bak_{}'.format(cluster_id, tag)
for host in hosts:
Center().cluster_backup(host, cluster_id, cluster_backup_dir)
msg = message.get('cluster_delete_complete')
msg = msg.format(cluster_id=cluster_id)
logger.info(msg)
def restore(self, cluster_id, tag=None):
"""Restore cluster
:param cluster_id: target cluster id
:param tag: Tag of backup, if omitted, restore the most recent backup file
"""
logger.debug('cluster restore: cluster_id={}, tag={}'.format(
cluster_id,
tag
))
if not cluster_util.validate_id(cluster_id):
raise ClusterIdError(cluster_id)
# find restore folder with tag (local)
path_of_fb = config.get_path_of_fb(cluster_id)
cluster_backup_path = path_of_fb['cluster_backup_path']
if tag is None:
backup_list = os.listdir(cluster_backup_path)
pattern = 'cluster_{}_bak_'.format(cluster_id)
filtered = filter(lambda x: x.startswith(pattern), backup_list)
sorted_list = sorted(list(filtered))
if not sorted_list:
msg = message.get('error_not_found_any_backup')
logger.error('BackupNotExistError: ' + msg)
return
tag = sorted_list[-1]
logger.debug("tag option is empty, auto select: {}".format(tag))
cluster_restore_dir = tag
backup_path = os.path.join(cluster_backup_path, cluster_restore_dir)
if not os.path.isdir(backup_path):
msg = message.get('error_not_found_backup').format(tag=tag)
logger.error('BackupNotExistError: ' + msg)
return
# get hosts from cluster props
props_path = os.path.join(
backup_path,
'tsr2-assembly-1.0.0-SNAPSHOT',
'conf',
'redis.properties'
)
hosts = config.get_props(props_path, 'sr2_redis_master_hosts', [])
# check status of hosts
success = Center().check_hosts_connection(hosts, True)
if not success:
msg = message.get('error_exist_unavailable_host')
logger.error(msg)
return
logger.debug('Connection of all hosts ok.')
success = Center().check_include_localhost(hosts)
if not success:
msg = message.get('error_not_include_localhost')
logger.error(msg)
return
# check all host tag folder: OK / NOT FOUND
msg = message.get('check_backup_info')
logger.info(msg)
buf = []
for host in hosts:
client = net.get_ssh(host)
if not net.is_dir(client, backup_path):
logger.debug('cannot find backup dir: {}-{}'.format(
host,
cluster_restore_dir
))
buf.append([host, color.red('NOT FOUND')])
client.close()
if buf:
utils.print_table([['HOST', 'RESULT'] + buf])
return
logger.info('OK')
# backup cluster
new_tag = time.strftime("%Y%m%d%H%M%S", time.gmtime())
cluster_backup_dir = 'cluster_{}_bak_{}'.format(cluster_id, new_tag)
for host in hosts:
Center().cluster_backup(host, cluster_id, cluster_backup_dir)
# restore cluster
command = "cp -r {} {}/cluster_{}".format(
backup_path,
path_of_fb['base_directory'],
cluster_id
)
for host in hosts:
msg = message.get('restore_cluster')
msg = msg.format(tag=cluster_backup_dir, host=host)
logger.info(msg)
client = net.get_ssh(host)
net.ssh_execute(client, command)
client.close()
logger.info("OK")