#!/bin/bash
# 快速压力测试脚本

# 配置
HOST="${1:-https://valuefrontier.cn}"
DURATION="${2:-30s}"

echo "=============================================="
echo "🚀 快速压力测试"
echo "   目标: $HOST"
echo "   时长: $DURATION"
echo "=============================================="

# 检查 wrk 是否安装
if ! command -v wrk &> /dev/null; then
    echo "❌ wrk 未安装，正在安装..."
    if command -v apt-get &> /dev/null; then
        apt-get update && apt-get install -y wrk
    elif command -v yum &> /dev/null; then
        yum install -y wrk
    else
        echo "请手动安装 wrk: https://github.com/wg/wrk"
        exit 1
    fi
fi

echo ""
echo "=== 测试 1: 首页 (100 连接) ==="
wrk -t4 -c100 -d$DURATION "$HOST/"

echo ""
echo "=== 测试 2: API 接口 (500 连接) ==="
wrk -t8 -c500 -d$DURATION "$HOST/api/stocks"

echo ""
echo "=== 测试 3: 高并发 (2000 连接) ==="
wrk -t12 -c2000 -d$DURATION "$HOST/api/stocks"

echo ""
echo "=== 测试 4: 极限测试 (5000 连接) ==="
wrk -t12 -c5000 -d$DURATION "$HOST/api/stocks"

echo ""
echo "=============================================="
echo "✅ 压力测试完成!"
echo "=============================================="