#!/usr/bin/env bash
# ============================================================
# One-click: start SGLang on both nodes + run ~20h benchmark.
# Usage: bash start_and_bench.sh
# ============================================================
set -euo pipefail

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"

log() { echo "[$(date '+%Y-%m-%d %H:%M:%S')] $*"; }

# ---- Step 1: Start server on both nodes ----
log "=== Step 1: Starting SGLang multi-node server ==="
bash "${SCRIPT_DIR}/start_sglang_multinode.sh"

# ---- Step 2: Run benchmark ----
log "=== Step 2: Starting ~20h benchmark ==="
bash "${SCRIPT_DIR}/run_batch.sh"

log "=== All done! ==="