564 lines
15 KiB
Bash
564 lines
15 KiB
Bash
|
#!/bin/bash
|
||
|
|
|
||
|
|
# Rustelo Application Deployment Script
|
||
|
|
# This script handles deployment of the Rustelo application in various environments
|
||
|
|
|
||
|
|
set -e
|
||
|
|
|
||
|
|
# Colors for output
|
||
|
|
RED='\033[0;31m'
|
||
|
|
GREEN='\033[0;32m'
|
||
|
|
YELLOW='\033[1;33m'
|
||
|
|
BLUE='\033[0;34m'
|
||
|
|
NC='\033[0m' # No Color
|
||
|
|
|
||
|
|
# Default values
|
||
|
|
ENVIRONMENT="production"
|
||
|
|
COMPOSE_FILE="docker-compose.yml"
|
||
|
|
BUILD_ARGS=""
|
||
|
|
MIGRATE_DB=false
|
||
|
|
BACKUP_DB=false
|
||
|
|
HEALTH_CHECK=true
|
||
|
|
TIMEOUT=300
|
||
|
|
PROJECT_NAME="rustelo"
|
||
|
|
DOCKER_REGISTRY=""
|
||
|
|
IMAGE_TAG="latest"
|
||
|
|
FORCE_RECREATE=false
|
||
|
|
SCALE_REPLICAS=1
|
||
|
|
FEATURES="production"
|
||
|
|
USE_DEFAULT_FEATURES=false
|
||
|
|
|
||
|
|
# Function to print colored output
|
||
|
|
print_status() {
|
||
|
|
echo -e "${GREEN}[INFO]${NC} $1"
|
||
|
|
}
|
||
|
|
|
||
|
|
print_warning() {
|
||
|
|
echo -e "${YELLOW}[WARN]${NC} $1"
|
||
|
|
}
|
||
|
|
|
||
|
|
print_error() {
|
||
|
|
echo -e "${RED}[ERROR]${NC} $1"
|
||
|
|
}
|
||
|
|
|
||
|
|
print_debug() {
|
||
|
|
if [[ "$DEBUG" == "true" ]]; then
|
||
|
|
echo -e "${BLUE}[DEBUG]${NC} $1"
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
|
||
|
|
# Function to show usage
|
||
|
|
show_usage() {
|
||
|
|
cat << EOF
|
||
|
|
Usage: $0 [OPTIONS] COMMAND
|
||
|
|
|
||
|
|
Commands:
|
||
|
|
deploy Deploy the application
|
||
|
|
stop Stop the application
|
||
|
|
restart Restart the application
|
||
|
|
status Show deployment status
|
||
|
|
logs Show application logs
|
||
|
|
scale Scale application replicas
|
||
|
|
backup Create database backup
|
||
|
|
migrate Run database migrations
|
||
|
|
rollback Rollback to previous version
|
||
|
|
health Check application health
|
||
|
|
update Update application to latest version
|
||
|
|
clean Clean up unused containers and images
|
||
|
|
|
||
|
|
Options:
|
||
|
|
-e, --env ENV Environment (dev|staging|production) [default: production]
|
||
|
|
-f, --file FILE Docker compose file [default: docker-compose.yml]
|
||
|
|
-p, --project PROJECT Project name [default: rustelo]
|
||
|
|
-t, --tag TAG Docker image tag [default: latest]
|
||
|
|
-r, --registry REGISTRY Docker registry URL
|
||
|
|
-s, --scale REPLICAS Number of replicas [default: 1]
|
||
|
|
--migrate Run database migrations before deployment
|
||
|
|
--backup Create database backup before deployment
|
||
|
|
--no-health-check Skip health check after deployment
|
||
|
|
--force-recreate Force recreation of containers
|
||
|
|
--timeout SECONDS Deployment timeout [default: 300]
|
||
|
|
--build-arg ARG Docker build arguments
|
||
|
|
--features FEATURES Cargo features to enable [default: production]
|
||
|
|
--default-features Use default features instead of custom
|
||
|
|
--debug Enable debug output
|
||
|
|
-h, --help Show this help message
|
||
|
|
|
||
|
|
Examples:
|
||
|
|
$0 deploy # Deploy production
|
||
|
|
$0 deploy -e staging # Deploy staging
|
||
|
|
$0 deploy --migrate --backup # Deploy with migration and backup
|
||
|
|
$0 scale -s 3 # Scale to 3 replicas
|
||
|
|
$0 logs -f # Follow logs
|
||
|
|
$0 health # Check health status
|
||
|
|
$0 deploy --features "auth,metrics" # Deploy with specific features
|
||
|
|
$0 deploy --default-features # Deploy with all default features
|
||
|
|
|
||
|
|
Environment Variables:
|
||
|
|
DOCKER_REGISTRY Docker registry URL
|
||
|
|
RUSTELO_ENV Environment override
|
||
|
|
COMPOSE_PROJECT_NAME Docker compose project name
|
||
|
|
DATABASE_URL Database connection string
|
||
|
|
DEBUG Enable debug mode
|
||
|
|
EOF
|
||
|
|
}
|
||
|
|
|
||
|
|
# Function to parse command line arguments
|
||
|
|
parse_args() {
|
||
|
|
while [[ $# -gt 0 ]]; do
|
||
|
|
case $1 in
|
||
|
|
-e|--env)
|
||
|
|
ENVIRONMENT="$2"
|
||
|
|
shift 2
|
||
|
|
;;
|
||
|
|
-f|--file)
|
||
|
|
COMPOSE_FILE="$2"
|
||
|
|
shift 2
|
||
|
|
;;
|
||
|
|
-p|--project)
|
||
|
|
PROJECT_NAME="$2"
|
||
|
|
shift 2
|
||
|
|
;;
|
||
|
|
-t|--tag)
|
||
|
|
IMAGE_TAG="$2"
|
||
|
|
shift 2
|
||
|
|
;;
|
||
|
|
-r|--registry)
|
||
|
|
DOCKER_REGISTRY="$2"
|
||
|
|
shift 2
|
||
|
|
;;
|
||
|
|
-s|--scale)
|
||
|
|
SCALE_REPLICAS="$2"
|
||
|
|
shift 2
|
||
|
|
;;
|
||
|
|
--migrate)
|
||
|
|
MIGRATE_DB=true
|
||
|
|
shift
|
||
|
|
;;
|
||
|
|
--backup)
|
||
|
|
BACKUP_DB=true
|
||
|
|
shift
|
||
|
|
;;
|
||
|
|
--no-health-check)
|
||
|
|
HEALTH_CHECK=false
|
||
|
|
shift
|
||
|
|
;;
|
||
|
|
--force-recreate)
|
||
|
|
FORCE_RECREATE=true
|
||
|
|
shift
|
||
|
|
;;
|
||
|
|
--timeout)
|
||
|
|
TIMEOUT="$2"
|
||
|
|
shift 2
|
||
|
|
;;
|
||
|
|
--build-arg)
|
||
|
|
BUILD_ARGS="$BUILD_ARGS --build-arg $2"
|
||
|
|
shift 2
|
||
|
|
;;
|
||
|
|
--features)
|
||
|
|
FEATURES="$2"
|
||
|
|
shift 2
|
||
|
|
;;
|
||
|
|
--default-features)
|
||
|
|
USE_DEFAULT_FEATURES=true
|
||
|
|
shift
|
||
|
|
;;
|
||
|
|
--debug)
|
||
|
|
DEBUG=true
|
||
|
|
shift
|
||
|
|
;;
|
||
|
|
-h|--help)
|
||
|
|
show_usage
|
||
|
|
exit 0
|
||
|
|
;;
|
||
|
|
-*)
|
||
|
|
print_error "Unknown option: $1"
|
||
|
|
show_usage
|
||
|
|
exit 1
|
||
|
|
;;
|
||
|
|
*)
|
||
|
|
COMMAND="$1"
|
||
|
|
shift
|
||
|
|
;;
|
||
|
|
esac
|
||
|
|
done
|
||
|
|
}
|
||
|
|
|
||
|
|
# Function to validate environment
|
||
|
|
validate_environment() {
|
||
|
|
case $ENVIRONMENT in
|
||
|
|
dev|development)
|
||
|
|
ENVIRONMENT="development"
|
||
|
|
COMPOSE_FILE="docker-compose.yml"
|
||
|
|
;;
|
||
|
|
staging)
|
||
|
|
ENVIRONMENT="staging"
|
||
|
|
COMPOSE_FILE="docker-compose.staging.yml"
|
||
|
|
;;
|
||
|
|
prod|production)
|
||
|
|
ENVIRONMENT="production"
|
||
|
|
COMPOSE_FILE="docker-compose.yml"
|
||
|
|
;;
|
||
|
|
*)
|
||
|
|
print_error "Invalid environment: $ENVIRONMENT"
|
||
|
|
print_error "Valid environments: dev, staging, production"
|
||
|
|
exit 1
|
||
|
|
;;
|
||
|
|
esac
|
||
|
|
}
|
||
|
|
|
||
|
|
# Function to check prerequisites
|
||
|
|
check_prerequisites() {
|
||
|
|
print_status "Checking prerequisites..."
|
||
|
|
|
||
|
|
# Check if Docker is installed and running
|
||
|
|
if ! command -v docker &> /dev/null; then
|
||
|
|
print_error "Docker is not installed or not in PATH"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
if ! docker info &> /dev/null; then
|
||
|
|
print_error "Docker daemon is not running"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Check if Docker Compose is installed
|
||
|
|
if ! command -v docker-compose &> /dev/null; then
|
||
|
|
print_error "Docker Compose is not installed or not in PATH"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Check if compose file exists
|
||
|
|
if [[ ! -f "$COMPOSE_FILE" ]]; then
|
||
|
|
print_error "Compose file not found: $COMPOSE_FILE"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
print_status "Prerequisites check passed"
|
||
|
|
}
|
||
|
|
|
||
|
|
# Function to set environment variables
|
||
|
|
set_environment_vars() {
|
||
|
|
export COMPOSE_PROJECT_NAME="${PROJECT_NAME}"
|
||
|
|
export DOCKER_REGISTRY="${DOCKER_REGISTRY}"
|
||
|
|
export IMAGE_TAG="${IMAGE_TAG}"
|
||
|
|
export ENVIRONMENT="${ENVIRONMENT}"
|
||
|
|
|
||
|
|
# Source environment-specific variables
|
||
|
|
if [[ -f ".env.${ENVIRONMENT}" ]]; then
|
||
|
|
print_status "Loading environment variables from .env.${ENVIRONMENT}"
|
||
|
|
source ".env.${ENVIRONMENT}"
|
||
|
|
elif [[ -f ".env" ]]; then
|
||
|
|
print_status "Loading environment variables from .env"
|
||
|
|
source ".env"
|
||
|
|
fi
|
||
|
|
|
||
|
|
print_debug "Environment variables set:"
|
||
|
|
print_debug " COMPOSE_PROJECT_NAME=${COMPOSE_PROJECT_NAME}"
|
||
|
|
print_debug " DOCKER_REGISTRY=${DOCKER_REGISTRY}"
|
||
|
|
print_debug " IMAGE_TAG=${IMAGE_TAG}"
|
||
|
|
print_debug " ENVIRONMENT=${ENVIRONMENT}"
|
||
|
|
print_debug " FEATURES=${FEATURES}"
|
||
|
|
print_debug " USE_DEFAULT_FEATURES=${USE_DEFAULT_FEATURES}"
|
||
|
|
}
|
||
|
|
|
||
|
|
# Function to build Docker images
|
||
|
|
build_images() {
|
||
|
|
print_status "Building Docker images..."
|
||
|
|
|
||
|
|
local build_cmd="docker-compose -f $COMPOSE_FILE build"
|
||
|
|
|
||
|
|
if [[ -n "$BUILD_ARGS" ]]; then
|
||
|
|
build_cmd="$build_cmd $BUILD_ARGS"
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Add feature arguments to build args
|
||
|
|
if [[ "$USE_DEFAULT_FEATURES" == "false" ]]; then
|
||
|
|
build_cmd="$build_cmd --build-arg CARGO_FEATURES=\"$FEATURES\" --build-arg NO_DEFAULT_FEATURES=\"true\""
|
||
|
|
else
|
||
|
|
build_cmd="$build_cmd --build-arg CARGO_FEATURES=\"\" --build-arg NO_DEFAULT_FEATURES=\"false\""
|
||
|
|
fi
|
||
|
|
|
||
|
|
if [[ "$DEBUG" == "true" ]]; then
|
||
|
|
print_debug "Build command: $build_cmd"
|
||
|
|
fi
|
||
|
|
|
||
|
|
if ! $build_cmd; then
|
||
|
|
print_error "Failed to build Docker images"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
print_status "Docker images built successfully"
|
||
|
|
}
|
||
|
|
|
||
|
|
# Function to create database backup
|
||
|
|
create_backup() {
|
||
|
|
if [[ "$BACKUP_DB" == "true" ]]; then
|
||
|
|
print_status "Creating database backup..."
|
||
|
|
|
||
|
|
local backup_file="backup_$(date +%Y%m%d_%H%M%S).sql"
|
||
|
|
|
||
|
|
if docker-compose -f "$COMPOSE_FILE" exec -T db pg_dump -U postgres rustelo_prod > "$backup_file"; then
|
||
|
|
print_status "Database backup created: $backup_file"
|
||
|
|
else
|
||
|
|
print_error "Failed to create database backup"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
|
||
|
|
# Function to run database migrations
|
||
|
|
run_migrations() {
|
||
|
|
if [[ "$MIGRATE_DB" == "true" ]]; then
|
||
|
|
print_status "Running database migrations..."
|
||
|
|
|
||
|
|
if docker-compose -f "$COMPOSE_FILE" run --rm migrate; then
|
||
|
|
print_status "Database migrations completed successfully"
|
||
|
|
else
|
||
|
|
print_error "Database migrations failed"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
|
||
|
|
# Function to deploy application
|
||
|
|
deploy_application() {
|
||
|
|
print_status "Deploying application..."
|
||
|
|
|
||
|
|
local compose_cmd="docker-compose -f $COMPOSE_FILE up -d"
|
||
|
|
|
||
|
|
if [[ "$FORCE_RECREATE" == "true" ]]; then
|
||
|
|
compose_cmd="$compose_cmd --force-recreate"
|
||
|
|
fi
|
||
|
|
|
||
|
|
if [[ "$SCALE_REPLICAS" -gt 1 ]]; then
|
||
|
|
compose_cmd="$compose_cmd --scale app=$SCALE_REPLICAS"
|
||
|
|
fi
|
||
|
|
|
||
|
|
if [[ "$DEBUG" == "true" ]]; then
|
||
|
|
print_debug "Deploy command: $compose_cmd"
|
||
|
|
fi
|
||
|
|
|
||
|
|
if ! $compose_cmd; then
|
||
|
|
print_error "Failed to deploy application"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
print_status "Application deployed successfully"
|
||
|
|
}
|
||
|
|
|
||
|
|
# Function to wait for application to be ready
|
||
|
|
wait_for_health() {
|
||
|
|
if [[ "$HEALTH_CHECK" == "true" ]]; then
|
||
|
|
print_status "Waiting for application to be healthy..."
|
||
|
|
|
||
|
|
local start_time=$(date +%s)
|
||
|
|
local health_url="http://localhost:3030/health"
|
||
|
|
|
||
|
|
while true; do
|
||
|
|
local current_time=$(date +%s)
|
||
|
|
local elapsed=$((current_time - start_time))
|
||
|
|
|
||
|
|
if [[ $elapsed -gt $TIMEOUT ]]; then
|
||
|
|
print_error "Health check timeout after ${TIMEOUT} seconds"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
if curl -f -s "$health_url" > /dev/null 2>&1; then
|
||
|
|
print_status "Application is healthy"
|
||
|
|
break
|
||
|
|
fi
|
||
|
|
|
||
|
|
print_debug "Health check failed, retrying in 5 seconds... (${elapsed}s elapsed)"
|
||
|
|
sleep 5
|
||
|
|
done
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
|
||
|
|
# Function to show deployment status
|
||
|
|
show_status() {
|
||
|
|
print_status "Deployment status:"
|
||
|
|
docker-compose -f "$COMPOSE_FILE" ps
|
||
|
|
|
||
|
|
print_status "Container resource usage:"
|
||
|
|
docker stats --no-stream --format "table {{.Container}}\t{{.CPUPerc}}\t{{.MemUsage}}\t{{.NetIO}}\t{{.BlockIO}}"
|
||
|
|
}
|
||
|
|
|
||
|
|
# Function to show logs
|
||
|
|
show_logs() {
|
||
|
|
local follow_flag=""
|
||
|
|
if [[ "$1" == "-f" ]]; then
|
||
|
|
follow_flag="-f"
|
||
|
|
fi
|
||
|
|
|
||
|
|
docker-compose -f "$COMPOSE_FILE" logs $follow_flag
|
||
|
|
}
|
||
|
|
|
||
|
|
# Function to scale application
|
||
|
|
scale_application() {
|
||
|
|
print_status "Scaling application to $SCALE_REPLICAS replicas..."
|
||
|
|
|
||
|
|
if docker-compose -f "$COMPOSE_FILE" up -d --scale app="$SCALE_REPLICAS"; then
|
||
|
|
print_status "Application scaled successfully"
|
||
|
|
else
|
||
|
|
print_error "Failed to scale application"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
|
||
|
|
# Function to stop application
|
||
|
|
stop_application() {
|
||
|
|
print_status "Stopping application..."
|
||
|
|
|
||
|
|
if docker-compose -f "$COMPOSE_FILE" down; then
|
||
|
|
print_status "Application stopped successfully"
|
||
|
|
else
|
||
|
|
print_error "Failed to stop application"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
|
||
|
|
# Function to restart application
|
||
|
|
restart_application() {
|
||
|
|
print_status "Restarting application..."
|
||
|
|
|
||
|
|
if docker-compose -f "$COMPOSE_FILE" restart; then
|
||
|
|
print_status "Application restarted successfully"
|
||
|
|
else
|
||
|
|
print_error "Failed to restart application"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
|
||
|
|
# Function to check application health
|
||
|
|
check_health() {
|
||
|
|
print_status "Checking application health..."
|
||
|
|
|
||
|
|
local health_url="http://localhost:3030/health"
|
||
|
|
|
||
|
|
if curl -f -s "$health_url" | jq '.status' | grep -q "healthy"; then
|
||
|
|
print_status "Application is healthy"
|
||
|
|
|
||
|
|
# Show detailed health information
|
||
|
|
curl -s "$health_url" | jq .
|
||
|
|
else
|
||
|
|
print_error "Application is not healthy"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
|
||
|
|
# Function to update application
|
||
|
|
update_application() {
|
||
|
|
print_status "Updating application..."
|
||
|
|
|
||
|
|
# Pull latest images
|
||
|
|
docker-compose -f "$COMPOSE_FILE" pull
|
||
|
|
|
||
|
|
# Restart with new images
|
||
|
|
docker-compose -f "$COMPOSE_FILE" up -d --force-recreate
|
||
|
|
|
||
|
|
print_status "Application updated successfully"
|
||
|
|
}
|
||
|
|
|
||
|
|
# Function to rollback application
|
||
|
|
rollback_application() {
|
||
|
|
print_warning "Rollback functionality not implemented yet"
|
||
|
|
print_warning "Please manually specify the desired image tag and redeploy"
|
||
|
|
}
|
||
|
|
|
||
|
|
# Function to clean up
|
||
|
|
cleanup() {
|
||
|
|
print_status "Cleaning up unused containers and images..."
|
||
|
|
|
||
|
|
# Remove stopped containers
|
||
|
|
docker container prune -f
|
||
|
|
|
||
|
|
# Remove unused images
|
||
|
|
docker image prune -f
|
||
|
|
|
||
|
|
# Remove unused volumes
|
||
|
|
docker volume prune -f
|
||
|
|
|
||
|
|
# Remove unused networks
|
||
|
|
docker network prune -f
|
||
|
|
|
||
|
|
print_status "Cleanup completed"
|
||
|
|
}
|
||
|
|
|
||
|
|
# Main function
|
||
|
|
main() {
|
||
|
|
# Parse command line arguments
|
||
|
|
parse_args "$@"
|
||
|
|
|
||
|
|
# Validate command
|
||
|
|
if [[ -z "$COMMAND" ]]; then
|
||
|
|
print_error "No command specified"
|
||
|
|
show_usage
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Validate environment
|
||
|
|
validate_environment
|
||
|
|
|
||
|
|
# Check prerequisites
|
||
|
|
check_prerequisites
|
||
|
|
|
||
|
|
# Set environment variables
|
||
|
|
set_environment_vars
|
||
|
|
|
||
|
|
# Execute command
|
||
|
|
case $COMMAND in
|
||
|
|
deploy)
|
||
|
|
build_images
|
||
|
|
create_backup
|
||
|
|
run_migrations
|
||
|
|
deploy_application
|
||
|
|
wait_for_health
|
||
|
|
show_status
|
||
|
|
;;
|
||
|
|
stop)
|
||
|
|
stop_application
|
||
|
|
;;
|
||
|
|
restart)
|
||
|
|
restart_application
|
||
|
|
wait_for_health
|
||
|
|
;;
|
||
|
|
status)
|
||
|
|
show_status
|
||
|
|
;;
|
||
|
|
logs)
|
||
|
|
show_logs "$@"
|
||
|
|
;;
|
||
|
|
scale)
|
||
|
|
scale_application
|
||
|
|
;;
|
||
|
|
backup)
|
||
|
|
create_backup
|
||
|
|
;;
|
||
|
|
migrate)
|
||
|
|
run_migrations
|
||
|
|
;;
|
||
|
|
rollback)
|
||
|
|
rollback_application
|
||
|
|
;;
|
||
|
|
health)
|
||
|
|
check_health
|
||
|
|
;;
|
||
|
|
update)
|
||
|
|
update_application
|
||
|
|
wait_for_health
|
||
|
|
;;
|
||
|
|
clean)
|
||
|
|
cleanup
|
||
|
|
;;
|
||
|
|
*)
|
||
|
|
print_error "Unknown command: $COMMAND"
|
||
|
|
show_usage
|
||
|
|
exit 1
|
||
|
|
;;
|
||
|
|
esac
|
||
|
|
}
|
||
|
|
|
||
|
|
# Run main function
|
||
|
|
main "$@"
|