105 lines
3.2 KiB
Bash
Executable File
105 lines
3.2 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
BACKUP=$1
|
|
if [[ $BACKUP == "" ]]; then
|
|
echo "No backup name given. It should look like '2020-01-29'."
|
|
exit 1
|
|
fi
|
|
|
|
# Get current working directory (pwd doesn't cut it)
|
|
cwd=$(cd -P -- "$(dirname -- "$0")" && pwd -P)
|
|
# Set the environment:
|
|
set -o allexport
|
|
cat $cwd/../.env | grep "AWS_ACCESS_KEY_ID\|AWS_SECRET_ACCESS_KEY\|S3_BACKUP_PATH\|SKYNET_DB_USER\|SKYNET_DB_PASS\|SKYNET_DB_HOST\|SKYNET_DB_PORT" >.tmpenv
|
|
source .tmpenv
|
|
rm .tmpenv
|
|
set +o allexport
|
|
# Check for AWS credentials:
|
|
if [[ $AWS_ACCESS_KEY_ID == "" || $AWS_SECRET_ACCESS_KEY == "" ]]; then
|
|
echo "Missing AWS credentials!"
|
|
exit 1
|
|
fi
|
|
# Check for backup path:
|
|
if [[ $S3_BACKUP_PATH == "" ]]; then
|
|
echo "Missing S3_BACKUP_PATH!"
|
|
exit 1
|
|
fi
|
|
|
|
### COCKROACH DB ###
|
|
echo "Restoring CockroachDB."
|
|
# Check if the backup exists:
|
|
totalFoundObjects=$(aws s3 ls $S3_BACKUP_PATH/$BACKUP --recursive --summarize | grep "cockroach" | wc -l)
|
|
if [ "$totalFoundObjects" -eq "0" ]; then
|
|
echo "This backup doesn't exist!"
|
|
exit 1
|
|
fi
|
|
# Restore the backup:
|
|
docker exec cockroach \
|
|
cockroach sql \
|
|
--host cockroach:26257 \
|
|
--certs-dir=/certs \
|
|
--execute="ALTER DATABASE defaultdb RENAME TO defaultdb_backup;"
|
|
if [[ $? > 0 ]]; then
|
|
echo "Failed to rename existing CockroachDB database. Exiting."
|
|
exit $?
|
|
fi
|
|
docker exec cockroach \
|
|
cockroach sql \
|
|
--host cockroach:26257 \
|
|
--certs-dir=/certs \
|
|
--execute="RESTORE DATABASE defaultdb FROM '$S3_BACKUP_PATH/$BACKUP/cockroach?AWS_ACCESS_KEY_ID=$AWS_ACCESS_KEY_ID&AWS_SECRET_ACCESS_KEY=$AWS_SECRET_ACCESS_KEY';"
|
|
if [[ $? == 0 ]]; then
|
|
# Restoration succeeded, drop the backup.
|
|
docker exec cockroach \
|
|
cockroach sql \
|
|
--host cockroach:26257 \
|
|
--certs-dir=/certs \
|
|
--execute="DROP DATABASE defaultdb_backup;"
|
|
echo "CockroachDB restoration succeeded."
|
|
else
|
|
# Restoration failed, drop the new DB and put back the old one.
|
|
echo "CockroachDB restoration failed, rolling back."
|
|
docker exec cockroach \
|
|
cockroach sql \
|
|
--host cockroach:26257 \
|
|
--certs-dir=/certs \
|
|
--execute="DROP DATABASE defaultdb;"
|
|
docker exec cockroach \
|
|
cockroach sql \
|
|
--host cockroach:26257 \
|
|
--certs-dir=/certs \
|
|
--execute="ALTER DATABASE defaultdb_backup RENAME TO defaultdb;"
|
|
if [[ $? > 0 ]]; then
|
|
echo "ERROR: Rollback failed! Inspect manually!"
|
|
exit $?
|
|
else
|
|
echo "Rollback successful. Restoration cancelled. Exiting."
|
|
exit 0
|
|
fi
|
|
fi
|
|
|
|
### MONGO DB ###
|
|
# Check if the backup exists:
|
|
totalFoundObjects=$(aws s3 ls $S3_BACKUP_PATH/$BACKUP --recursive --summarize | grep "mongo.tgz" | wc -l)
|
|
if [ "$totalFoundObjects" -eq "0" ]; then
|
|
echo "This backup doesn't exist!"
|
|
exit 1
|
|
fi
|
|
# Get the backup from S3:
|
|
aws s3 cp $S3_BACKUP_PATH/$BACKUP/mongo.tgz mongo.tgz
|
|
# Prepare a clean `to_restore` dir:
|
|
rm -rf $cwd/../docker/data/mongo/db/backups/to_restore
|
|
mkdir -p $cwd/../docker/data/mongo/db/backups/to_restore
|
|
# Decompress the backup:
|
|
tar -xzf mongo.tgz -C $cwd/../docker/data/mongo/db/backups/to_restore
|
|
rm mongo.tgz
|
|
# Restore the backup:
|
|
# The name of the backup is not `mongo` due to the way we're creating it,
|
|
# it's $BACKUP.
|
|
docker exec mongo \
|
|
mongorestore \
|
|
mongodb://$SKYNET_DB_USER:$SKYNET_DB_PASS@$SKYNET_DB_HOST:$SKYNET_DB_PORT \
|
|
/data/db/backups/to_restore/$BACKUP
|
|
# Clean up:
|
|
rm -rf $cwd/../docker/data/mongo/db/backups/to_restore
|