Files
training-labs/labs/osbash/tools/restore-cluster.sh
Roger Luethi c53a789f43 Fix sourcing of config/provider.*
Source the provider config file only in the provider-specific library.
It makes more sense and also fixes a bug that prevented repeat-test
to work properly with repeated tests on KVM.

Change-Id: I0658aba5fe9b08a1ce01f26d4f909f80c621f54c
2016-04-16 08:22:50 +02:00

155 lines
4.0 KiB
Bash
Executable File

#!/bin/bash
set -o errexit -o nounset
TOP_DIR=$(cd "$(dirname "$0")/.." && pwd)
source "$TOP_DIR/config/localrc"
source "$TOP_DIR/config/paths"
source "$CONFIG_DIR/deploy.osbash"
source "$OSBASH_LIB_DIR/functions-host.sh"
source "$OSBASH_LIB_DIR/$PROVIDER-functions.sh"
OSBASH=exec_cmd
function usage {
# Setting to empty string selects latest (current snapshot)
echo "Usage: $0 {-l|-c|-t <SNAP>} [-s]"
echo ""
echo "-h Help"
echo "-l List snapshots for node VMs"
echo "-c Restore cluster node VMs to current snapshot"
echo "-t SNAP Restore cluster to target snapshot"
echo "-s Start each node VMs after restoring it"
exit
}
function list_snapshots {
for vm_name in $(script_cfg_get_nodenames); do
if ! vm_exists "$vm_name"; then
echo "VM $vm_name does not exist. Skipping..."
continue
fi
echo -e "Snapshot list for $vm_name node:"
vm_snapshot_list_tree "$vm_name"
echo
done
exit 0
}
while getopts :chlst: opt; do
case $opt in
c)
CURRENT=yes
;;
h)
usage
;;
l)
list_snapshots
;;
s)
START=yes
;;
t)
TARGET_SNAPSHOT=$OPTARG
;;
:)
echo "Error: -$OPTARG needs argument"
;;
?)
echo "Error: invalid option -$OPTARG"
echo
usage
;;
esac
done
# Remove processed options from arguments
shift $(( OPTIND - 1 ));
if [ $# -ne 0 ]; then
usage
elif [ -z "${TARGET_SNAPSHOT:-}" -a -z "${CURRENT:-""}" ]; then
echo
echo "Error: no target snapshot given."
echo
usage
elif [ -n "${TARGET_SNAPSHOT:-}" -a -n "${CURRENT:-""}" ]; then
echo
echo "Error: conflicting options: target snapshot name and -c."
echo
usage
fi
# Find target_snapshot in scripts_cfg and set global *_SNAPSHOT variables
# to the correct snapshot name for each node (to allow building from there)
function set_snapshot_vars {
local target_snapshot=$1
local found=0
local config_name=$(get_distro_name "$DISTRO")_cluster
local scripts_cfg="$CONFIG_DIR/scripts.$config_name"
while read -r line; do
if [[ $line =~ ^cmd\ snapshot.*-n\ ([^ ]*)\ (.*) ]]; then
# Node name (e.g. controller)
node=${BASH_REMATCH[1]}
# Snapshot name (e.g. keystone_installed)
snapshot=${BASH_REMATCH[2]}
# Global variable name (e.g. CONTROLLER_SNAPSHOT)
# Can't use ${node,,} (OS X bash version is only 3.2)
var_name=$(echo "$node"|tr "a-z" "A-Z")_SNAPSHOT
if [ "$snapshot" = "$target_snapshot" ]; then
# Can't use associative arrays (OS X bash version is only 3.2)
eval "${var_name}=$snapshot"
found=1
elif [ $found -eq 0 ]; then
eval "${var_name}=$snapshot"
fi
fi
done < "$scripts_cfg"
if [ $found -eq 0 ]; then
echo "ERROR: snapshot '$target_snapshot' not found"
exit 1
fi
}
if [ -n "${TARGET_SNAPSHOT:-}" ]; then
set_snapshot_vars "$TARGET_SNAPSHOT"
fi
for vm_name in $(script_cfg_get_nodenames); do
if ! vm_exists "$vm_name"; then
echo "VM $vm_name does not exist. Skipping..."
continue
fi
vm_power_off "$vm_name"
vm_wait_for_shutdown "$vm_name"
if [ "${CURRENT:-""}" = "yes" ]; then
vm_snapshot_restore_current "$vm_name"
if [ "${START:-""}" = "yes" ]; then
vm_boot "$vm_name"
fi
else
# Global variable name (e.g. CONTROLLER_SNAPSHOT)
# (use tr due to OS X bash limitation)
var_name=$(echo "$vm_name"|tr "a-z" "A-Z")_SNAPSHOT
if [ -z "${!var_name:=""}" ]; then
vm_delete "$vm_name"
else
vm_snapshot_restore "$vm_name" "${!var_name}"
if [ "${START:-""}" = "yes" ]; then
vm_boot "$vm_name"
fi
fi
fi
done