make backup/restore InfluxDB scripts robust

smartemission · Jun 19, 2018 · 5068aad · 5068aad
1 parent 50a6eda
commit 5068aad
Show file tree

Hide file tree

Showing 4 changed files with 106 additions and 4 deletions.
diff --git a/services/influxdb-dc1/backup.sh b/services/influxdb-dc1/backup.sh
@@ -1,10 +1,33 @@
 #!/bin/bash
 # See https://www.influxdata.com/blog/backuprestore-of-influxdb-fromto-docker-containers/
 
-source influxdb.env
+SCRIPT_DIR=${0%/*}
+
+pushd ${SCRIPT_DIR}
+	if [ ! -f influxdb.env ]
+	then
+	    echo "Bestand influxdb.env niet gevonden."
+	    exit 1
+	fi
+    source influxdb.env
+popd
+
+# Make sure vars are set
+if [ -z "${SE_BACKUP_DIR}" ]
+then
+    echo "SE_BACKUP_DIR not set"
+    exit 1
+fi
+
+if [ -z "${SE_CONTAINER_NAME}" ]
+then
+    echo "SE_CONTAINER_NAME not set"
+    exit 1
+fi
 
 BACKUP_DIR="${SE_BACKUP_DIR}/${SE_CONTAINER_NAME}"
 mkdir -p ${BACKUP_DIR}
+
 rm -rf ${BACKUP_DIR}/*
 TARGET_DUMP_FILE=${SE_BACKUP_DIR}/influxdb_${INFLUXDB_DB}_data.tar.gz
 

diff --git a/services/influxdb-dc1/restore.sh b/services/influxdb-dc1/restore.sh
@@ -19,7 +19,35 @@ fi
 # Parameter ok
 ./stop.sh
 
-source influxdb.env
+SCRIPT_DIR=${0%/*}
+
+pushd ${SCRIPT_DIR}
+	if [ ! -f influxdb.env ]
+	then
+	    echo "Bestand influxdb.env niet gevonden."
+	    exit 1
+	fi
+    source influxdb.env
+popd
+
+# Make sure vars are set
+if [ -z "${SE_BACKUP_DIR}" ]
+then
+    echo "SE_BACKUP_DIR not set"
+    exit 1
+fi
+
+if [ -z "${SE_CONTAINER_NAME}" ]
+then
+    echo "SE_CONTAINER_NAME not set"
+    exit 1
+fi
+
+if [ -z "${SE_DATA_DIR}" ]
+then
+    echo "SE_BACKUP_DIR not set"
+    exit 1
+fi
 
 DATA_DIR="${SE_DATA_DIR}/${SE_CONTAINER_NAME}"
 BACKUP_DIR="${SE_BACKUP_DIR}/${SE_CONTAINER_NAME}"

diff --git a/services/influxdb/backup.sh b/services/influxdb/backup.sh
@@ -1,10 +1,33 @@
 #!/bin/bash
 # See https://www.influxdata.com/blog/backuprestore-of-influxdb-fromto-docker-containers/
 
-source influxdb.env
+SCRIPT_DIR=${0%/*}
+
+pushd ${SCRIPT_DIR}
+	if [ ! -f influxdb.env ]
+	then
+	    echo "Bestand influxdb.env niet gevonden."
+	    exit 1
+	fi
+    source influxdb.env
+popd
+
+# Make sure vars are set
+if [ -z "${SE_BACKUP_DIR}" ]
+then
+    echo "SE_BACKUP_DIR not set"
+    exit 1
+fi
+
+if [ -z "${SE_CONTAINER_NAME}" ]
+then
+    echo "SE_CONTAINER_NAME not set"
+    exit 1
+fi
 
 BACKUP_DIR="${SE_BACKUP_DIR}/${SE_CONTAINER_NAME}"
 mkdir -p ${BACKUP_DIR}
+
 rm -rf ${BACKUP_DIR}/*
 TARGET_DUMP_FILE=${SE_BACKUP_DIR}/influxdb_${INFLUXDB_DB}_data.tar.gz
 

diff --git a/services/influxdb/restore.sh b/services/influxdb/restore.sh
@@ -19,7 +19,35 @@ fi
 # Parameter ok
 ./stop.sh
 
-source influxdb.env
+SCRIPT_DIR=${0%/*}
+
+pushd ${SCRIPT_DIR}
+	if [ ! -f influxdb.env ]
+	then
+	    echo "Bestand influxdb.env niet gevonden."
+	    exit 1
+	fi
+    source influxdb.env
+popd
+
+# Make sure vars are set
+if [ -z "${SE_BACKUP_DIR}" ]
+then
+    echo "SE_BACKUP_DIR not set"
+    exit 1
+fi
+
+if [ -z "${SE_CONTAINER_NAME}" ]
+then
+    echo "SE_CONTAINER_NAME not set"
+    exit 1
+fi
+
+if [ -z "${SE_DATA_DIR}" ]
+then
+    echo "SE_BACKUP_DIR not set"
+    exit 1
+fi
 
 DATA_DIR="${SE_DATA_DIR}/${SE_CONTAINER_NAME}"
 BACKUP_DIR="${SE_BACKUP_DIR}/${SE_CONTAINER_NAME}"