#!/bin/bash

# 检查是否已安装Open-CAS
check_opencas() {
    if ! command -v casadm &> /dev/null; then
        echo "错误: Open-CAS未安装或无法在PATH中找到。"
        echo "请先安装Open-CAS: https://open-cas.github.io/"
        exit 1
    fi
    echo "Open-CAS已安装"
}

# 检查设备是否存在
check_devices() {
    local nvme_device=$(grep -oP 'devices_nvme: "\K[^"]+' config.yaml)
    local hdd_device=$(grep -oP 'devices_hdd: "\K[^"]+' config.yaml)
    
    if [ ! -b "$nvme_device" ]; then
        echo "错误: NVMe设备 $nvme_device 不存在或不是块设备"
        exit 1
    fi
    
    if [ ! -b "$hdd_device" ]; then
        echo "错误: HDD设备 $hdd_device 不存在或不是块设备"
        exit 1
    fi
    
    echo "设备检查通过: $nvme_device 和 $hdd_device 存在"
}

# 编译项目
build_project() {
    echo "编译项目..."
    go build -o bin/server cmd/server/main.go
    go build -o bin/client cmd/client/main.go
    
    if [ $? -ne 0 ]; then
        echo "编译失败"
        exit 1
    fi
    
    echo "编译成功"
}

# 启动服务器
start_server() {
    echo "启动服务器..."
    mkdir -p logs
    ./bin/server -config config.yaml -log-level info > logs/server.log 2>&1 &
    SERVER_PID=$!
    
    echo "服务器进程ID: $SERVER_PID"
    echo "等待服务器启动..."
    sleep 3
    
    # 检查服务器是否正常启动
    if ! ps -p $SERVER_PID > /dev/null; then
        echo "服务器启动失败，请检查日志: logs/server.log"
        exit 1
    fi
    
    echo "服务器已成功启动"
}

# 运行指定测试
run_test() {
    local test_type=$1
    local data_size=$2
    local block_size=$3
    
    echo "运行测试: $test_type (数据大小: ${data_size}MB, 块大小: ${block_size}KB)"
    
    ./bin/client -config config.yaml -test "$test_type" -data-size "$data_size" -block-size "$block_size" > "logs/client_${test_type}.log" 2>&1
    
    if [ $? -ne 0 ]; then
        echo "测试 $test_type 失败，请检查日志: logs/client_${test_type}.log"
        return 1
    fi
    
    echo "测试 $test_type 完成"
    return 0
}

# 运行所有测试
run_all_tests() {
    local data_size=$1
    local block_size=$2
    local concurrent=$3
    
    if [ "$concurrent" = "true" ]; then
        echo "并发运行所有测试 (数据大小: ${data_size}MB, 块大小: ${block_size}KB)"
        ./bin/client -config config.yaml -test all -data-size "$data_size" -block-size "$block_size" -concurrent > logs/client_all.log 2>&1
    else
        echo "顺序运行所有测试 (数据大小: ${data_size}MB, 块大小: ${block_size}KB)"
        ./bin/client -config config.yaml -test all -data-size "$data_size" -block-size "$block_size" > logs/client_all.log 2>&1
    fi
    
    if [ $? -ne 0 ]; then
        echo "运行所有测试失败，请检查日志: logs/client_all.log"
        return 1
    fi
    
    echo "所有测试完成"
    return 0
}

# 停止服务器
stop_server() {
    if [ -n "$SERVER_PID" ]; then
        echo "停止服务器 (PID: $SERVER_PID)..."
        kill $SERVER_PID
        wait $SERVER_PID 2>/dev/null
        echo "服务器已停止"
    fi
}

# 清理资源
cleanup() {
    echo "清理资源..."
    stop_server
    
    # 检查是否有遗留的缓存实例
    local cache_id=$(grep -oP 'cache_instance_id: "\K[^"]+' config.yaml)
    if casadm -L | grep -q "Cache Instance $cache_id"; then
        echo "停止缓存实例 $cache_id..."
        casadm -T -i $cache_id -f
    fi
    
    echo "清理完成"
}

# 主函数
main() {
    local test_type=${1:-"all"}
    local data_size=${2:-100}  # 默认100MB
    local block_size=${3:-4}   # 默认4KB
    local concurrent=${4:-"false"} # 默认非并发
    
    mkdir -p bin logs
    
    # 注册清理函数
    trap cleanup EXIT
    
    check_opencas
    check_devices
    build_project
    start_server
    
    if [ "$test_type" = "all" ]; then
        run_all_tests "$data_size" "$block_size" "$concurrent"
    else
        run_test "$test_type" "$data_size" "$block_size"
    fi
    
    echo "测试执行完毕"
}

# 执行主函数，参数: 测试类型 数据大小(MB) 块大小(KB) 是否并发
main "$@"