start-multiple.sh

#!/bin/bash

# Colors for output
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
RED='\033[0;31m'
BLUE='\033[0;34m'
NC='\033[0m' # No Color

# Directory for logs
LOG_DIR="./logs"
mkdir -p "$LOG_DIR"

# PID file to track running processes
PID_FILE="./running_processes.txt"

# Function to validate positive integer input
validate_number() {
    if [[ "$1" =~ ^[0-9]+$ ]] && [ "$1" -gt 0 ]; then
        return 0
    else
        return 1
    fi
}

# Initialize PID file if it doesn't exist
if [ ! -f "$PID_FILE" ]; then
    touch "$PID_FILE"
fi

echo -e "${GREEN}=== Temporal Rate Limit Tester - Interactive Startup ===${NC}"
echo ""
echo "This script will help you start workers, starters, and queriers."
echo "You can start each group separately and ramp up as needed."
echo ""

# Counter for how many of each type are already running
EXISTING_WORKERS=0
EXISTING_STARTERS=0
EXISTING_QUERIERS=0

if [ -s "$PID_FILE" ]; then
    EXISTING_WORKERS=$(grep "^worker:" "$PID_FILE" 2>/dev/null | wc -l | tr -d ' ')
    EXISTING_STARTERS=$(grep "^starter:" "$PID_FILE" 2>/dev/null | wc -l | tr -d ' ')
    EXISTING_QUERIERS=$(grep "^querier:" "$PID_FILE" 2>/dev/null | wc -l | tr -d ' ')
fi

# Ensure they're numbers
EXISTING_WORKERS=${EXISTING_WORKERS:-0}
EXISTING_STARTERS=${EXISTING_STARTERS:-0}
EXISTING_QUERIERS=${EXISTING_QUERIERS:-0}

if [ $EXISTING_WORKERS -gt 0 ] || [ $EXISTING_STARTERS -gt 0 ] || [ $EXISTING_QUERIERS -gt 0 ]; then
    echo -e "${BLUE}Currently running:${NC}"
    [ $EXISTING_WORKERS -gt 0 ] && echo "  - Workers: $EXISTING_WORKERS"
    [ $EXISTING_STARTERS -gt 0 ] && echo "  - Starters: $EXISTING_STARTERS"
    [ $EXISTING_QUERIERS -gt 0 ] && echo "  - Queriers: $EXISTING_QUERIERS"
    echo ""
fi

# Start Workers
echo -e "${YELLOW}=== WORKERS ===${NC}"
read -p "How many workers would you like to start? (0 to skip): " NUM_WORKERS

if validate_number "$NUM_WORKERS"; then
    echo ""
    echo -e "${YELLOW}Starting $NUM_WORKERS workers...${NC}"
    WORKER_START=$((EXISTING_WORKERS + 1))
    WORKER_END=$((EXISTING_WORKERS + NUM_WORKERS))
    for i in $(seq $WORKER_START $WORKER_END); do
        go run accumulator/worker/main.go > "$LOG_DIR/worker-$i.log" 2>&1 &
        PID=$!
        echo "worker:$PID" >> "$PID_FILE"
        echo "  - Worker $i started (PID: $PID)"
    done
    echo -e "${GREEN}✓ Workers started${NC}"
elif [ "$NUM_WORKERS" = "0" ]; then
    echo "  Skipping workers"
else
    echo -e "${RED}Invalid input. Skipping workers.${NC}"
fi
echo ""

# Start Starters
echo -e "${YELLOW}=== STARTERS (Workflow initiators) ===${NC}"
read -p "How many starters would you like to start? (0 to skip): " NUM_STARTERS

if validate_number "$NUM_STARTERS"; then
    echo ""
    echo -e "${YELLOW}Starting $NUM_STARTERS starters...${NC}"
    STARTER_START=$((EXISTING_STARTERS + 1))
    STARTER_END=$((EXISTING_STARTERS + NUM_STARTERS))
    for i in $(seq $STARTER_START $STARTER_END); do
        go run accumulator/starter/main.go > "$LOG_DIR/starter-$i.log" 2>&1 &
        PID=$!
        echo "starter:$PID" >> "$PID_FILE"
        echo "  - Starter $i started (PID: $PID)"
    done
    echo -e "${GREEN}✓ Starters started${NC}"
elif [ "$NUM_STARTERS" = "0" ]; then
    echo "  Skipping starters"
else
    echo -e "${RED}Invalid input. Skipping starters.${NC}"
fi
echo ""

# Start Queriers
echo -e "${YELLOW}=== QUERIERS ===${NC}"
read -p "How many queriers would you like to start? (0 to skip): " NUM_QUERIERS

if validate_number "$NUM_QUERIERS"; then
    echo ""
    echo -e "${YELLOW}Starting $NUM_QUERIERS queriers...${NC}"
    QUERIER_START=$((EXISTING_QUERIERS + 1))
    QUERIER_END=$((EXISTING_QUERIERS + NUM_QUERIERS))
    for i in $(seq $QUERIER_START $QUERIER_END); do
        go run accumulator/query/main.go > "$LOG_DIR/querier-$i.log" 2>&1 &
        PID=$!
        echo "querier:$PID" >> "$PID_FILE"
        echo "  - Querier $i started (PID: $PID)"
    done
    echo -e "${GREEN}✓ Queriers started${NC}"
elif [ "$NUM_QUERIERS" = "0" ]; then
    echo "  Skipping queriers"
else
    echo -e "${RED}Invalid input. Skipping queriers.${NC}"
fi
echo ""

# Calculate totals
TOTAL_WORKERS=$(grep "^worker:" "$PID_FILE" 2>/dev/null | wc -l | tr -d ' ')
TOTAL_STARTERS=$(grep "^starter:" "$PID_FILE" 2>/dev/null | wc -l | tr -d ' ')
TOTAL_QUERIERS=$(grep "^querier:" "$PID_FILE" 2>/dev/null | wc -l | tr -d ' ')

# Ensure they're numbers
TOTAL_WORKERS=${TOTAL_WORKERS:-0}
TOTAL_STARTERS=${TOTAL_STARTERS:-0}
TOTAL_QUERIERS=${TOTAL_QUERIERS:-0}

echo -e "${GREEN}=== Startup Complete ===${NC}"
echo ""
echo "Total processes now running:"
echo "  - Workers: $TOTAL_WORKERS"
echo "  - Starters: $TOTAL_STARTERS"
echo "  - Queriers: $TOTAL_QUERIERS"
echo ""
echo "Logs are being written to: $LOG_DIR/"
echo ""
echo "Useful commands:"
echo "  - View status:  ./list-processes.sh"
echo "  - Add more:     ./start-multiple.sh"
echo "  - Stop all:     ./stop-all.sh"