Gan, Jimmy
92 lines
2.4 KiB
Bash
92 lines
2.4 KiB
Bash
#!/bin/bash
|
|
|
|
# Configuration
|
|
BACKUP_DIR="/backups"
|
|
DB_HOST="mold_cost_db"
|
|
DB_NAME="mold_cost"
|
|
DB_USER="postgres"
|
|
DB_PASSWORD="postgres"
|
|
|
|
# Set PostgreSQL password environment variable
|
|
export PGPASSWORD="$DB_PASSWORD"
|
|
|
|
# Function to list available backups
|
|
list_backups() {
|
|
echo "Available backups:"
|
|
ls -la "$BACKUP_DIR"/backup_*.sql.gz 2>/dev/null | while read -r line; do
|
|
echo " $line"
|
|
done
|
|
}
|
|
|
|
# Function to restore from backup
|
|
restore_backup() {
|
|
local backup_file="$1"
|
|
|
|
if [ ! -f "$backup_file" ]; then
|
|
echo "ERROR: Backup file '$backup_file' not found!"
|
|
list_backups
|
|
exit 1
|
|
fi
|
|
|
|
echo "WARNING: This will overwrite the current database!"
|
|
echo "Backup file: $backup_file"
|
|
read -p "Are you sure you want to continue? (y/N): " -n 1 -r
|
|
echo
|
|
|
|
if [[ $REPLY =~ ^[Yy]$ ]]; then
|
|
echo "Restoring database from backup..."
|
|
|
|
# Drop and recreate database
|
|
echo "Dropping existing database..."
|
|
dropdb -h "$DB_HOST" -U "$DB_USER" --no-password "$DB_NAME" 2>/dev/null || true
|
|
|
|
echo "Creating new database..."
|
|
createdb -h "$DB_HOST" -U "$DB_USER" --no-password "$DB_NAME"
|
|
|
|
echo "Restoring data..."
|
|
if gunzip -c "$backup_file" | psql -h "$DB_HOST" -U "$DB_USER" -d "$DB_NAME" --no-password; then
|
|
echo "Database restored successfully!"
|
|
else
|
|
echo "ERROR: Database restore failed!"
|
|
exit 1
|
|
fi
|
|
else
|
|
echo "Restore cancelled."
|
|
fi
|
|
}
|
|
|
|
# Main script logic
|
|
if [ $# -eq 0 ]; then
|
|
echo "Usage: $0 [list|restore <backup_file>]"
|
|
echo ""
|
|
echo "Commands:"
|
|
echo " list - List available backups"
|
|
echo " restore <backup_file> - Restore database from backup file"
|
|
echo ""
|
|
echo "Examples:"
|
|
echo " $0 list"
|
|
echo " $0 restore /backups/backup_20241201_120000.sql.gz"
|
|
exit 1
|
|
fi
|
|
|
|
case "$1" in
|
|
"list")
|
|
list_backups
|
|
;;
|
|
"restore")
|
|
if [ -z "$2" ]; then
|
|
echo "ERROR: Please specify a backup file to restore from."
|
|
echo "Usage: $0 restore <backup_file>"
|
|
exit 1
|
|
fi
|
|
restore_backup "$2"
|
|
;;
|
|
*)
|
|
echo "ERROR: Unknown command '$1'"
|
|
echo "Usage: $0 [list|restore <backup_file>]"
|
|
exit 1
|
|
;;
|
|
esac
|
|
|
|
# Clear password from environment
|
|
unset PGPASSWORD |