Add gzip and archive functionality to mongo backup

Dockerfile

@@ -1,5 +1,5 @@
-FROM mongo:latest
-MAINTAINER Steven Tsang <[email protected]>
+FROM library/mongo:3.4
+LABEL maintainer="Steven Tsang <[email protected]>"
 
 RUN apt-get update && apt-get -y install cron && mkdir -p /backup
 

README.md

@@ -30,6 +30,8 @@ Moreover, if you link `tsangste/mongodb-backup` to a mongodb container(e.g. offi
     CRON_TIME       the interval of cron job to run mongodump. `0 0 * * *` by default, which is every day at 00:00
     MAX_BACKUPS     the number of backups to keep. When reaching the limit, the old backup will be discarded. No limit, by default
     INIT_BACKUP     if set, create a backup when the container launched
+    USE_GZIP        if set, sets dump and restore to use gzip
+    USE_ARCHIVE     if set, dump and restore with a single file format and can be used with gzip
 
 ## Restore from a backup
 
@@ -39,4 +41,4 @@ See the list of backups, you can run:
 
 To restore database from a certain backup, simply run:
 
-    docker exec mongodb-backup /restore.sh /backup/2015.08.06.171901
+    docker exec mongodb-backup /restore.sh 2015.08.06.171901

docker-compose.development.yml

@@ -0,0 +1,6 @@
+version: '3.2'
+services:
+  mongodbbackup:
+    build:
+      context: .
+      dockerfile: Dockerfile

docker-compose.yml

@@ -1,17 +1,21 @@
-mongodb:
-  image: 'mongo:latest'
-  environment:
-    - MONGODB_PASS=mypass
-  ports:
-    - "27017:27017"
-    - "28017:28017"
-mongodbbackup:
-  image: 'tsangste/mongodb-backup:latest'
-  links:
-    - mongodb
-  environment:
-    - CRON_TIME=20 3 * * * #Change to your favorate cron job schedule
-    - MAX_BACKUPS=10
-    - INIT_BACKUP=yes
-  volumes:
-    - /host/backup:/backup #Change to the host folder where you want to store the backups
+version: '3.2'
+services:
+  mongodb:
+    image: 'library/mongo:3.4'
+    environment:
+      - MONGODB_PASS=mypass
+    ports:
+      - "27017:27017"
+      - "28017:28017"
+  mongodbbackup:
+    image: 'tsangste/mongodb-backup:latest'
+    links:
+      - mongodb
+    environment:
+      - CRON_TIME=20 3 * * * #Change to your favourite cron job schedule
+      - MAX_BACKUPS=10
+      - INIT_BACKUP=yes
+      - USE_GZIP=yes
+      - USE_ARCHIVE=yes
+    volumes:
+      - /srv/backup:/backup #Change to the host folder where you want to store the backups

run.sh

@@ -15,17 +15,33 @@ MONGODB_PASS=${MONGODB_PASS:-${MONGODB_ENV_MONGODB_PASS}}
 [[ ( -n "${MONGODB_PASS}" ) ]] && PASS_STR=" --password ${MONGODB_PASS}"
 [[ ( -n "${MONGODB_DB}" ) ]] && DB_STR=" --db ${MONGODB_DB}"
 
-BACKUP_CMD="mongodump --out /backup/"'${BACKUP_NAME}'" --host ${MONGODB_HOST} --port ${MONGODB_PORT} ${USER_STR}${PASS_STR}${DB_STR} ${EXTRA_OPTS}"
+use_gzip_backup () {
+    if [ -n "${USE_GZIP}" ]; then
+        echo ".gz --gzip"
+    else
+        echo ""
+    fi
+}
+
+use_archive_backup () {
+    if [ -n "${USE_ARCHIVE}" ]; then
+        echo "--archive="
+    else
+        echo "--out "
+    fi
+}
+
+BACKUP_CMD="mongodump $(use_archive_backup)/backup/"'${BACKUP_NAME}'"$(use_gzip_backup) --host ${MONGODB_HOST} --port ${MONGODB_PORT} ${USER_STR}${PASS_STR}${DB_STR}  ${EXTRA_OPTS}"
 
 echo "=> Creating backup script"
 rm -f /backup.sh
 cat <<EOF >> /backup.sh
 #!/bin/bash
 MAX_BACKUPS=${MAX_BACKUPS}
-BACKUP_NAME=\$(date +\%Y.\%m.\%d.\%H\%M\%S)
+BACKUP_NAME=\$(date +\%Y.\%m.\%d.\%H\%M\c%S)
 
 echo "=> Backup started"
-if ${BACKUP_CMD} ;then
+if ${BACKUP_CMD}; then
     echo "   Backup succeeded"
 else
     echo "   Backup failed"
@@ -44,6 +60,20 @@ echo "=> Backup done"
 EOF
 chmod +x /backup.sh
 
+use_gzip_restore () {
+    if [ -n "${USE_GZIP}" ]; then
+        echo "--gzip"
+    fi
+}
+
+use_archive_restore () {
+    if [ -n "${USE_ARCHIVE}" ]; then
+        echo "--archive="
+    fi
+}
+
+RESTORE_CMD="mongorestore --objcheck --host ${MONGODB_HOST} --port ${MONGODB_PORT} ${USER_STR}${PASS_STR} $(use_gzip_restore) $(use_archive_restore)"
+
 echo "=> Creating restore script"
 rm -f /restore.sh
 cat <<EOF >> /restore.sh
@@ -53,7 +83,7 @@ name="\$1"
 shift 1
 echo "=> Restore database from \$name"
 echo \$*
-if mongorestore --objcheck --host ${MONGODB_HOST} --port ${MONGODB_PORT} ${USER_STR}${PASS_STR} \$* /backup/\$name; then
+if  ${RESTORE_CMD}\$*/backup/\$name; then
     echo "   Restore succeeded"
 else
     echo "   Restore failed"