prepare-bursty-service.sh

#!/bin/bash

# Function for consistent log formatting
log_info() {
    echo -e "\n[INFO] $(date '+%Y-%m-%d %H:%M:%S') - $1"
}

log_error() {
    echo -e "\n[ERROR] $(date '+%Y-%m-%d %H:%M:%S') - $1" >&2
}

log_success() {
    echo -e "\n[SUCCESS] $(date '+%Y-%m-%d %H:%M:%S') - $1"
}

# List of all bursty services
BURSTY_SERVICES=(
    "ts-cancel-service"
    "ts-basic-service"
    "ts-travel-service"
    # "ts-preserve-service"
    "ts-seat-service"
)

# Initialize array for tracking failed services
failed_services=()

# Function to randomly select a service
select_random_service() {
    local array_length=${#BURSTY_SERVICES[@]}
    local random_index=$((RANDOM % array_length))
    echo "${BURSTY_SERVICES[$random_index]}"
}

# Check arguments and handle --no-groundtruth flag
if [ "$1" = "--no-groundtruth" ]; then
    if [ "$#" -ne 5 ]; then
        log_error "Missing required arguments"
        echo "Prerequiste: 1. install maven, 2.login to dockerhub use \"docker login\", 3. make sure train-ticket repo is already in cacti-exp branch"
        echo "Usage: $0 --no-groundtruth <bursty-period> <burst-rate> <burst-duration> <tag-name>"
        echo "Example: $0 --no-groundtruth 60 5 10 v1.0.0"
        echo "Parameters:"
        echo "  - bursty-period: Time between bursts in seconds (e.g., 60 for 1 minute)"
        echo "  - burst-rate: Requests per second during burst (e.g., 5)"
        echo "  - burst-duration: Duration of each burst in seconds (e.g., 10)"
        exit 1
    fi
    TARGET_SERVICE=$(select_random_service)
    BURSTY_PERIOD_SECONDS=$2
    BURST_REQUESTS_PER_SEC=$3
    BURST_DURATION_SECONDS=$4
    TAG_NAME=$5
    log_info "Random target service selected: $TARGET_SERVICE"
else
    if [ "$#" -ne 5 ]; then
        log_error "Missing required arguments"
        echo "Prerequiste: 1. install maven, 2.login to dockerhub use \"docker login\", 3. make sure train-ticket repo is already in cacti-exp branch"
        echo "Usage: $0 <target-service-name> <bursty-period> <burst-rate> <burst-duration> <tag-name>"
        echo "Note: the first parameter could be --no-groundtruth flag that randomly select a trigger serivce to enable burstness"
        echo "Example: $0 ts-cancel-service 60 5 10 v1.0.0"
        echo "Parameters:"
        echo "  - bursty-period: Time between bursts in seconds (e.g., 60 for 1 minute)"
        echo "  - burst-rate: Requests per second during burst (e.g., 5)"
        echo "  - burst-duration: Duration of each burst in seconds (e.g., 10)"
        exit 1
    fi
    TARGET_SERVICE=$1
    BURSTY_PERIOD_SECONDS=$2
    BURST_REQUESTS_PER_SEC=$3
    BURST_DURATION_SECONDS=$4
    TAG_NAME=$5
fi

log_info "Starting service update process with following parameters:"
echo "Target Service: $TARGET_SERVICE"
echo "Bursty Period: $BURSTY_PERIOD_SECONDS seconds"
echo "Burst Rate: $BURST_REQUESTS_PER_SEC requests/second"
echo "Burst Duration: $BURST_DURATION_SECONDS seconds"
echo "Tag Name: $TAG_NAME"

# Function to get controller path based on service name
get_controller_path() {
    local service=$1
    local service_part=$(echo $service | sed 's/ts-\(.*\)-service/\1/')
    local controller_name="$(tr '[:lower:]' '[:upper:]' <<< ${service_part:0:1})${service_part:1}Controller.java"
    
    echo "${service}/src/main/java/${service_part}/controller/${controller_name}"
}

# Function to clean up any local changes and update to latest remote version
cleanup_and_update() {
    log_info "Starting repository cleanup and update"
    log_info "Starting repository cleanup and update"
    echo "Cleaning up local changes..."
    git restore . 2>/dev/null || true
    git clean -fd 2>/dev/null || true
    
    echo "Switching to cacti-exp branch..."
    git switch cacti-exp
    
    echo "Pulling latest changes..."
    git pull origin cacti-exp --ff-only
    
    if [ $? -ne 0 ]; then
        log_error "Failed to update to latest version. Exiting."
        log_error "Failed to update to latest version. Exiting."
        exit 1
    fi
    log_success "Repository cleanup and update completed"
}

# Function to get path based on service name
get_service_path() {
    local service=$1
    
    # Special path and naming for basic service
    if [ "$service" = "ts-basic-service" ]; then
        echo "${service}/src/main/java/fdse/microservice/service/BasicServiceImpl.java"
    elif [ "$service" = "ts-cancel-service" ]; then
        echo "${service}/src/main/java/cancel/service/CancelServiceImpl.java"
    elif [ "$service" = "ts-seat-service" ]; then
        echo "${service}/src/main/java/seat/service/SeatServiceImpl.java"
    # elif [ "$service" = "ts-travel-service" ]; then
    #     echo "${service}/src/main/java/travel/service/TravelServiceImpl.java"
    else
        # Controller path for other services
        local service_part=$(echo $service | sed 's/ts-\(.*\)-service/\1/')
        local controller_name="$(tr '[:lower:]' '[:upper:]' <<< ${service_part:0:1})${service_part:1}Controller.java"
        echo "${service}/src/main/java/${service_part}/controller/${controller_name}"
    fi
}

# Function to update burst parameters in a service
update_service_params() {
    local service=$1
    local period=$2
    local rate=$3
    local duration=$4
    
    log_info "Updating parameters for service: $service"
    local file_path=$(get_service_path "$service")
    
    if [ ! -f "$file_path" ]; then
        log_error "File not found at $file_path"
        return 1
    fi
    
    echo "Target file path: $file_path"
    echo "Applying changes:"
    
    # Create backup of original file
    cp "$file_path" "${file_path}.bak"
    
    if [ "$service" = "ts-basic-service" ] || [ "$service" = "ts-cancel-service" ] || [ "$service" = "ts-seat-service" ]; then
        # BasicServiceImpl parameters
        echo "  - BURST_PERIOD_SECONDS: $period"
        echo "  - BURST_REQUESTS_PER_SEC: $rate"
        echo "  - BURST_DURATION_SECONDS: $duration"
        
        sed -i "s/private static final int BURST_PERIOD_SECONDS = [0-9]\+;/private static final int BURST_PERIOD_SECONDS = ${period};/" "$file_path"
        sed -i "s/private static final int BURST_REQUESTS_PER_SEC = [0-9]\+;/private static final int BURST_REQUESTS_PER_SEC = ${rate};/" "$file_path"
        sed -i "s/private static final int BURST_DURATION_SECONDS = [0-9]\+;/private static final int BURST_DURATION_SECONDS = ${duration};/" "$file_path"
        
        # Verify changes for BasicServiceImpl
        if grep -q "BURST_PERIOD_SECONDS = ${period}" "$file_path" && \
           grep -q "BURST_REQUESTS_PER_SEC = ${rate}" "$file_path" && \
           grep -q "BURST_DURATION_SECONDS = ${duration}" "$file_path"; then
            log_success "Successfully updated parameters for $service"
            rm "${file_path}.bak"
        else
            log_error "Failed to update all parameters for $service"
            log_info "Restoring backup file"
            mv "${file_path}.bak" "$file_path"
            return 1
        fi
    else
        # Controller parameters for other services
        echo "  - BURSTY_PERIOD_SECONDS: $period"
        echo "  - BURST_REQUESTS_PER_SEC: $rate"
        echo "  - BURST_DURATION_SECONDS: $duration"
        
        sed -i "s/private static final int BURSTY_PERIOD_SECONDS = [0-9]\+;/private static final int BURSTY_PERIOD_SECONDS = ${period};/" "$file_path"
        sed -i "s/private static final int BURST_REQUESTS_PER_SEC = [0-9]\+;/private static final int BURST_REQUESTS_PER_SEC = ${rate};/" "$file_path"
        sed -i "s/private static final int BURST_DURATION_SECONDS = [0-9]\+;/private static final int BURST_DURATION_SECONDS = ${duration};/" "$file_path"
        
        # Verify changes for Controllers
        if grep -q "BURSTY_PERIOD_SECONDS = ${period}" "$file_path" && \
           grep -q "BURST_REQUESTS_PER_SEC = ${rate}" "$file_path" && \
           grep -q "BURST_DURATION_SECONDS = ${duration}" "$file_path"; then
            log_success "Successfully updated parameters for $service"
            rm "${file_path}.bak"
        else
            log_error "Failed to update all parameters for $service"
            log_info "Restoring backup file"
            mv "${file_path}.bak" "$file_path"
            return 1
        fi
    fi
}

# Function to build and push Docker image
build_and_push_docker() {
    local service=$1
    local tag=$2
    
    log_info "Building and pushing Docker image for $service"
    
    # Navigate to the service directory
    cd "${service}" || { log_error "Failed to navigate to ${service} directory"; return 1; }

    # Build Docker image
    log_info "Building Docker image for $service:$tag"
    if ! sudo docker build -t "docclabgroup/${service}:${tag}" .; then
        log_error "Docker build failed for $service"
        cd ..
        return 1
    fi

    # Push Docker image
    log_info "Pushing Docker image for $service:$tag"
    if ! sudo docker push "docclabgroup/${service}:${tag}"; then
        log_error "Docker push failed for $service"
        cd ..
        return 1
    fi
    
    cd .. || { log_error "Failed to navigate back from ${service} directory"; return 1; }
    log_success "Successfully built and pushed Docker image for $service"
}

# Function to check rollout status
check_all_rollouts() {
    local services=("$@")
    local failed_rollouts=()
    local success=true

    log_info "Checking rollout status for all services"
    
    for service in "${services[@]}"; do
        log_info "Checking rollout status for $service"
        if ! kubectl rollout status "deployment/${service}" --timeout=300s; then
            log_error "Rollout failed for $service"
            failed_rollouts+=("$service")
            success=false
        else
            log_success "Rollout completed successfully for $service"
        fi
    done

    if [ "$success" = false ]; then
        log_error "The following services failed to roll out: ${failed_rollouts[*]}"
        return 1
    fi
    
    log_success "All services rolled out successfully"
    return 0
}

# Main execution starts here
log_info "Starting main execution"
cd /local/train-ticket || { log_error "Failed to navigate to train-ticket directory"; exit 1; }
sudo chown -R $(whoami) .

# Initial cleanup and update
cleanup_and_update

# First Phase: Update Parameters
log_info "Phase 1: Updating service parameters"
echo "Total services to process: ${#BURSTY_SERVICES[@]}"

for service in "${BURSTY_SERVICES[@]}"; do
    echo -e "\n----------------------------------------"
    log_info "Updating parameters for service: $service"

    if [ "$service" = "$TARGET_SERVICE" ]; then
        log_info "$service is the target service - applying specified burst parameters"
        update_service_params "$service" "$BURSTY_PERIOD_SECONDS" "$BURST_REQUESTS_PER_SEC" "$BURST_DURATION_SECONDS"
    else
        log_info "$service is not the target service - setting zero burst parameters"
        update_service_params "$service" "0" "0" "0"
    fi
    echo "----------------------------------------"
done

# Second Phase: Maven Build
log_info "Phase 2: Building project with Maven"
if ! mvn clean install -DskipTests; then
    log_error "Maven build failed"
    exit 1
fi
log_success "Maven build completed"

# Third Phase: Build and Push Docker Images
log_info "Phase 3: Building and pushing Docker images"
for service in "${BURSTY_SERVICES[@]}"; do
    echo -e "\n----------------------------------------"
    if ! build_and_push_docker "$service" "$TAG_NAME"; then
        failed_services+=("$service")
    fi
    echo "----------------------------------------"
done

if [ ${#failed_services[@]} -ne 0 ]; then
    log_error "Failed to build/push the following services: ${failed_services[*]}"
    exit 1
fi

# Fourth Phase: Update Kubernetes Deployments
log_info "Phase 4: Updating Kubernetes deployments"
echo "Updating all service images simultaneously..."

for service in "${BURSTY_SERVICES[@]}"; do
    log_info "Setting new image for $service"
    kubectl set image "deployment/${service}" "${service}=docclabgroup/${service}:${TAG_NAME}" &
done

# Wait for all kubectl set image commands to complete
log_info "Waiting for all image updates to complete"
wait
log_success "All deployment image updates initiated"

# Fifth Phase: Check Rollout Status
log_info "Phase 5: Checking rollout status for all services"
if ! check_all_rollouts "${BURSTY_SERVICES[@]}"; then
    log_error "Some deployments failed to roll out properly"
    exit 1
fi

log_success "All deployments completed successfully!"
echo -e "\nFinal Configuration Summary:"
echo "Target Service: ${TARGET_SERVICE}"
echo "Target Configuration:"
echo "- Bursty Period: ${BURSTY_PERIOD_SECONDS} seconds"
echo "- Burst Rate: ${BURST_REQUESTS_PER_SEC} requests per second"
echo "- Burst Duration: ${BURST_DURATION_SECONDS} seconds"
echo "- Image Tag: ${TAG_NAME}"
echo "All other services have been set to zero burst parameters"