77 lines
2.2 KiB
Bash
77 lines
2.2 KiB
Bash
|
#!/bin/bash
|
||
|
|
# AO3 Mirror - 启动脚本
|
||
|
|
# 启动 4 个 uvicorn worker 进程 + Caddy 负载均衡
|
||
|
|
|
||
|
|
set -e
|
||
|
|
|
||
|
|
BASE_DIR="/home/ubuntu/ao3-mirror"
|
||
|
|
LOGS_DIR="$BASE_DIR"
|
||
|
|
PID_DIR="/dev/shm/ao3"
|
||
|
|
WORKER_COUNT=2
|
||
|
|
START_PORT=8081
|
||
|
|
|
||
|
|
mkdir -p "$PID_DIR"
|
||
|
|
|
||
|
|
# Kill any existing workers
|
||
|
|
echo "[Startup] Killing existing workers..."
|
||
|
|
for pid_file in "$PID_DIR"/worker-*.pid; do
|
||
|
|
[ -f "$pid_file" ] && kill "$(cat "$pid_file")" 2>/dev/null || true
|
||
|
|
done
|
||
|
|
sleep 1
|
||
|
|
|
||
|
|
# Also kill by port
|
||
|
|
for port in $(seq $START_PORT $((START_PORT + WORKER_COUNT - 1))); do
|
||
|
|
fuser -k "${port}/tcp" 2>/dev/null || true
|
||
|
|
done
|
||
|
|
|
||
|
|
# Start backend workers
|
||
|
|
echo "[Startup] Starting $WORKER_COUNT backend workers..."
|
||
|
|
for i in $(seq 0 $((WORKER_COUNT - 1))); do
|
||
|
|
port=$((START_PORT + i))
|
||
|
|
log_file="$LOGS_DIR/worker-$i.log"
|
||
|
|
pid_file="$PID_DIR/worker-$i.pid"
|
||
|
|
|
||
|
|
cd "$BASE_DIR"
|
||
|
|
|
||
|
|
# Pin worker i to CPU i (2-core machine: CPU 0, CPU 1)
|
||
|
|
taskset -c $i python3 -m uvicorn app:app \
|
||
|
|
--host 127.0.0.1 \
|
||
|
|
--port "$port" \
|
||
|
|
--workers 1 \
|
||
|
|
--loop uvloop \
|
||
|
|
--http httptools \
|
||
|
|
--log-level warning \
|
||
|
|
--timeout-keep-alive 30 \
|
||
|
|
>> "$log_file" 2>&1 &
|
||
|
|
|
||
|
|
echo $! > "$pid_file"
|
||
|
|
echo "[Startup] Worker $i started on port $port (PID: $!)"
|
||
|
|
done
|
||
|
|
|
||
|
|
# Wait for workers to be ready
|
||
|
|
echo "[Startup] Waiting for workers to become healthy..."
|
||
|
|
for i in $(seq 0 $((WORKER_COUNT - 1))); do
|
||
|
|
port=$((START_PORT + i))
|
||
|
|
for attempt in $(seq 1 15); do
|
||
|
|
if curl -sf "http://127.0.0.1:$port/health" > /dev/null 2>&1; then
|
||
|
|
echo "[Startup] Worker $i (port $port) is healthy"
|
||
|
|
break
|
||
|
|
fi
|
||
|
|
sleep 1
|
||
|
|
done
|
||
|
|
done
|
||
|
|
|
||
|
|
# Configure Caddy
|
||
|
|
echo "[Startup] Applying Caddyfile..."
|
||
|
|
sudo caddy fmt "$BASE_DIR/Caddyfile" --overwrite 2>/dev/null || true
|
||
|
|
sudo cp "$BASE_DIR/Caddyfile" /etc/caddy/Caddyfile
|
||
|
|
sudo systemctl reload caddy 2>/dev/null || sudo systemctl restart caddy
|
||
|
|
echo "[Startup] Caddy reloaded"
|
||
|
|
|
||
|
|
echo "[Startup] AO3 Mirror is running!"
|
||
|
|
echo " Workers: 2 (ports 8081-8082, pinned to CPUs 0-1)"
|
||
|
|
echo " Caddy: port 443 (agento3.miscs.dev)"
|
||
|
|
echo " Stats: https://agento3.miscs.dev/stats"
|
||
|
|
echo " Health: https://agento3.miscs.dev/health"
|
||
|
|
echo " Logs: $LOGS_DIR/worker-*.log"
|