Refactor and restore function to backup script

This commit is contained in:
Wojciech Kozlowski 2019-07-29 21:46:22 +01:00
parent fa7af91461
commit f9ded87409

175
backup.sh
View File

@ -2,9 +2,10 @@
set -e
CYAN='\033[01;36m'
YELLOW='\033[01;33m'
RED='\033[01;31m'
GREEN='\033[01;32m'
YELLOW='\033[01;33m'
CYAN='\033[01;36m'
NC='\033[00m'
SCRIPT=$(readlink -f $0)
@ -14,63 +15,155 @@ DIRNAME=$(dirname $SCRIPT)
# Get the list of volumes.
# -----------------------------------------------------------------------------
echo -e "${CYAN}[${SCRIPT}] Extract list of volumes ${NC}"
function get_volumes {
# Find the line where "services:" start
services_line=$(grep -n services docker-compose.yml | awk '{split($0, a, ":"); print a[1]}')
# Find the line where "services:" start
services_line=$(grep -n services docker-compose.yml | \
awk '{split($0, a, ":"); print a[1]}')
# The volumes are listed between "volumes:" and "services:"
volume_list=$(head -n $services_line docker-compose.yml | awk '/volumes:/,/services:/')
# The volumes are listed between "volumes:" and "services:"
volume_list=$(head -n $services_line docker-compose.yml | \
awk '/volumes:/,/services:/')
# Split into array
IFS=':'; volumes=($volume_list); unset IFS;
# Split into array
IFS=':'; volumes=($volume_list); unset IFS;
# Trim whitespace
for ((i = 0; i < ${#volumes[@]}; i++)); do
volumes[$i]=$(echo -e "${volumes[$i]}" | tr -d '[:space:]')
done
# Trim whitespace
for ((i = 0; i < ${#volumes[@]}; i++)); do
volumes[$i]=$(echo -e "${volumes[$i]}" | tr -d '[:space:]')
done
# Verify that the first entry is "volumes" and the last "services"
if [ ${volumes[0]} != "volumes" ] || [ "${volumes[-1]}" != "services" ]; then
echo -e "${RED}Unexpected input${NC}" >&2
exit 1
fi
# Verify that the first entry is "volumes" and the last "services"
if [ ${volumes[0]} != "volumes" ] || [ "${volumes[-1]}" != "services" ]
then
echo -e "${RED}Unexpected input${NC}" >&2
exit 1
fi
# Remove first and last entry - they will be "volumes" and " services"
let len=${#volumes[@]}-2
volumes=("${volumes[@]:1:$len}")
# Remove first and last entry - they will be "volumes" and " services"
let len=${#volumes[@]}-2
volumes=("${volumes[@]:1:$len}")
# Print final list
echo -e "${YELLOW}Volumes${NC}:"
for vol in "${volumes[@]}"; do
echo -e " - ${YELLOW}${vol} ${NC}"
done
echo ${volumes[*]}
}
# -----------------------------------------------------------------------------
# Start the server.
# -----------------------------------------------------------------------------
function server_start {
echo -e "${CYAN}[${SCRIPT}] Restart loki-server ${NC}"
service loki-server start
}
# -----------------------------------------------------------------------------
# Stop the server.
# -----------------------------------------------------------------------------
echo -e "${CYAN}[${SCRIPT}] Stop loki-server ${NC}"
function server_stop {
service loki-server stop
echo -e "${CYAN}[${SCRIPT}] Stop loki-server ${NC}"
service loki-server stop
}
# -----------------------------------------------------------------------------
# Back up volumes.
# -----------------------------------------------------------------------------
for vol in "${volumes[@]}"; do
echo -e "${CYAN}[${SCRIPT}] Back up ${YELLOW}${vol}${CYAN} volume ${NC}"
function backup {
set -o xtrace
docker run --rm -v loki_${vol}:/opt/${vol} -v /srv/backup:/opt/backup debian:stable-slim bash -c \
"cd /opt/${vol} && tar cf /opt/backup/${vol}.tar ."
set +o xtrace
volumes=$1
for vol in "${volumes[@]}"
do
echo -e "${CYAN}[${SCRIPT}] Back up ${YELLOW}${vol}${CYAN} volume${NC}"
set -o xtrace
docker run --rm \
-v loki_${vol}:/opt/${vol} \
-v /srv/backup:/opt/backup \
debian:stable-slim \
bash -c "cd /opt/${vol} && tar cf /opt/backup/${vol}.tar ."
set +o xtrace
done
}
# -----------------------------------------------------------------------------
# Restore volumes.
# -----------------------------------------------------------------------------
function restore {
volumes=$1
for vol in "${volumes[@]}"
do
echo -e "${CYAN}[${SCRIPT}] Restore ${YELLOW}${vol}${CYAN} volume${NC}"
set -o xtrace
docker run --rm \
-v loki_${vol}:/opt/${vol} \
-v /srv/backup:/opt/backup \
debian:stable-slim \
bash -c "cd /opt/${vol} && tar xf /opt/backup/${vol}.tar"
set +o xtrace
done
}
# -----------------------------------------------------------------------------
# Main.
# -----------------------------------------------------------------------------
while getopts "br" option
do
case ${option} in
b )
echo -e "${CYAN}[${SCRIPT}] Extract list of volumes ${NC}"
volumes=($(get_volumes))
echo -e "${YELLOW}Volumes${NC}:"
for vol in "${volumes[@]}"
do
echo -e " - ${YELLOW}${vol} ${NC}"
done
server_stop
backup ${volumes}
server_start
exit 0
;;
r )
echo -e "${CYAN}[${SCRIPT}] Extract list of volumes ${NC}"
volumes=($(get_volumes))
echo -e "${YELLOW}Volumes${NC}:"
for vol in "${volumes[@]}"
do
echo -e " - ${YELLOW}${vol} ${NC}"
done
server_stop
restore ${volumes}
server_start
exit 0
;;
\? )
echo -e "${GREEN} Usage: backup.sh [-b|-r]${NC}"
exit 1
;;
esac
done
# -----------------------------------------------------------------------------
# Restart the server.
# -----------------------------------------------------------------------------
echo -e "${CYAN}[${SCRIPT}] Restart loki-server ${NC}"
service loki-server start
# If we get here then no options were triggered
echo -e "${GREEN} Usage: backup.sh [-b|-r]${NC}"