#!/bin/bash

# CentOS 极速安装 wrk 脚本
# 针对安装慢、中途停止进行了优化

set -e

echo ">>> 开始极速安装流程..."

# 1. 权限检查
if [ "$EUID" -ne 0 ]; then 
  echo "错误：请使用 sudo 运行此脚本"
  exit 1
fi

# 2. 清理旧环境（解决安装一半停止的问题）
echo ">>> 正在清理旧的残留文件..."
rm -rf /tmp/wrk_build
rm -rf ./wrk
rm -f /usr/local/bin/wrk

# 3. 精简依赖安装 (跳过 update，只装必要的)
echo ">>> 正在安装核心编译组件 (极速模式)..."
yum install -y gcc make openssl-devel git

# 4. 浅层克隆源码 (只下载最新一层，速度极快)
echo ">>> 正在下载 wrk 源码 (depth=1)..."
git clone --depth 1 https://github.com/wg/wrk.git /tmp/wrk_build

# 5. 多核并行编译
echo ">>> 正在编译..."
cd /tmp/wrk_build
# 获取 CPU 核心数以加速编译
THREADS=$(nproc)
make -j $THREADS

# 6. 安装
echo ">>> 正在完成安装..."
cp wrk /usr/local/bin/
chmod +x /usr/local/bin/wrk

# 7. 最终清理
rm -rf /tmp/wrk_build

echo "--------------------------------------"
if command -v wrk >/dev/null 2>&1; then
    echo "✅ 安装成功！"
    wrk --version
else
    echo "❌ 安装失败，请检查网络是否能访问 GitHub"
    exit 1
fi
echo "--------------------------------------"