training-guides/labs/tools/restore-cluster.sh
utsav dusad cf3b1fa923 Rename Bash scripts
Rename scripts in /labs/lib with proper naming convention.

Bug #1451071

Change-Id: I3a1f096414866846a7f9572804c5db852d76a36e
2015-05-03 22:44:56 +05:30

76 lines
1.9 KiB
Bash
Executable File

#!/bin/bash
set -o errexit -o nounset
TOP_DIR=$(cd "$(dirname "$0")/.." && pwd)
source "$TOP_DIR/config/paths"
source "$CONFIG_DIR/deploy.osbash"
source "$OSBASH_LIB_DIR/functions-host.sh"
CONTROLLER_VM=controller
NETWORK_VM=network
COMPUTE_VM=compute
function usage {
# Setting to empty string selects latest (current snapshot)
echo "Usage: $0 {current|<controller_snapshot_name>} {list-snapshots}"
echo " current: restore to currently active snapshot"
echo " list-snapshots: to list the snapshots of the vms"
exit
}
function cluster_restore {
vboxmanage controlvm $CONTROLLER_VM poweroff >/dev/null 2>&1 || rc=$?
sleep 1
if [ -n "$CONTROLLER_SNAPSHOT" ]; then
echo "Restoring $CONTROLLER_SNAPSHOT."
vboxmanage snapshot $CONTROLLER_VM restore "$CONTROLLER_SNAPSHOT"
else
echo "Restoring current snapshot."
vboxmanage snapshot $CONTROLLER_VM restorecurrent
fi
vboxmanage controlvm $COMPUTE_VM poweroff >/dev/null 2>&1 || rc=$?
sleep 1
vboxmanage snapshot $COMPUTE_VM restorecurrent
vboxmanage controlvm $NETWORK_VM poweroff >/dev/null 2>&1 || rc=$?
sleep 1
vboxmanage snapshot $NETWORK_VM restorecurrent
}
function cluster_start {
vboxmanage startvm $CONTROLLER_VM -t headless
vboxmanage startvm $COMPUTE_VM -t headless
vboxmanage startvm $NETWORK_VM -t headless
}
function list_snapshots {
for node in $CONTROLLER_VM $COMPUTE_VM $NETWORK_VM; do
echo -e "\n$node node's Snapshot"
vboxmanage snapshot $node list
echo
echo
sleep 1
done
exit 0
}
# Call the main brains
if [ $# -eq 0 ]; then
usage
elif [ "$1" = "list-snapshots" ]; then
list_snapshots
elif [ "$1" = "current" ]; then
CONTROLLER_SNAPSHOT=""
else
CONTROLLER_SNAPSHOT=$1
fi
echo "Restoring cluster snapshots."
cluster_restore
echo "Starting VMs."
cluster_start >/dev/null