#!/bin/bash

# 门店总业绩统计接口性能测试

echo "=== 门店总业绩统计接口性能测试 ==="

# 获取Token
TOKEN=$(curl -s -X POST "http://localhost:2011/api/oauth/Login" \
  -H "Content-Type: application/x-www-form-urlencoded" \
  -d "account=admin&password=e10adc3949ba59abbe56e057f20f883e" | python3 -c "import sys, json; print(json.load(sys.stdin)['data']['token'])" 2>/dev/null)

if [ -z "$TOKEN" ]; then
  echo "❌ 获取Token失败"
  exit 1
fi

echo "✅ Token获取成功"
echo ""

# 测试1: 分页查询（每页10条）
echo "--- 测试1: 分页查询（每页10条） ---"
START=$(date +%s.%3N)
RESPONSE1=$(curl -s -w "\nTIME:%{time_total}" -X POST "http://localhost:2011/api/Extend/LqStatistics/get-store-total-performance-statistics-list" \
  -H "Authorization: $TOKEN" \
  -H "Content-Type: application/json" \
  -d '{"statisticsMonth": "202512", "pageIndex": 1, "pageSize": 10}')
END=$(date +%s.%3N)
TIME1=$(echo "$RESPONSE1" | grep "TIME:" | cut -d: -f2)
RESPONSE_BODY1=$(echo "$RESPONSE1" | sed '/TIME:/d')
TOTAL1=$(echo "$RESPONSE_BODY1" | python3 -c "import sys, json; data=json.load(sys.stdin); print(data.get('data', {}).get('pagination', {}).get('total', 0))" 2>/dev/null)
echo "响应时间: ${TIME1}秒"
echo "总记录数: $TOTAL1"
echo ""

# 测试2: 分页查询（每页50条）
echo "--- 测试2: 分页查询（每页50条） ---"
START=$(date +%s.%3N)
RESPONSE2=$(curl -s -w "\nTIME:%{time_total}" -X POST "http://localhost:2011/api/Extend/LqStatistics/get-store-total-performance-statistics-list" \
  -H "Authorization: $TOKEN" \
  -H "Content-Type: application/json" \
  -d '{"statisticsMonth": "202512", "pageIndex": 1, "pageSize": 50}')
END=$(date +%s.%3N)
TIME2=$(echo "$RESPONSE2" | grep "TIME:" | cut -d: -f2)
RESPONSE_BODY2=$(echo "$RESPONSE2" | sed '/TIME:/d')
TOTAL2=$(echo "$RESPONSE_BODY2" | python3 -c "import sys, json; data=json.load(sys.stdin); print(data.get('data', {}).get('pagination', {}).get('total', 0))" 2>/dev/null)
echo "响应时间: ${TIME2}秒"
echo "总记录数: $TOTAL2"
echo ""

# 测试3: 门店名称筛选
echo "--- 测试3: 门店名称筛选（搜索'紫荆'） ---"
START=$(date +%s.%3N)
RESPONSE3=$(curl -s -w "\nTIME:%{time_total}" -X POST "http://localhost:2011/api/Extend/LqStatistics/get-store-total-performance-statistics-list" \
  -H "Authorization: $TOKEN" \
  -H "Content-Type: application/json" \
  -d '{"statisticsMonth": "202512", "storeName": "紫荆", "pageIndex": 1, "pageSize": 10}')
END=$(date +%s.%3N)
TIME3=$(echo "$RESPONSE3" | grep "TIME:" | cut -d: -f2)
RESPONSE_BODY3=$(echo "$RESPONSE3" | sed '/TIME:/d')
TOTAL3=$(echo "$RESPONSE_BODY3" | python3 -c "import sys, json; data=json.load(sys.stdin); print(data.get('data', {}).get('pagination', {}).get('total', 0))" 2>/dev/null)
LIST_COUNT3=$(echo "$RESPONSE_BODY3" | python3 -c "import sys, json; data=json.load(sys.stdin); print(len(data.get('data', {}).get('list', [])))" 2>/dev/null)
echo "响应时间: ${TIME3}秒"
echo "总记录数: $TOTAL3"
echo "当前页记录数: $LIST_COUNT3"
echo ""

# 测试4: 连续5次请求，测试稳定性
echo "--- 测试4: 连续5次请求，测试稳定性 ---"
TIMES=()
for i in {1..5}; do
  START=$(date +%s.%3N)
  RESPONSE=$(curl -s -w "\nTIME:%{time_total}" -X POST "http://localhost:2011/api/Extend/LqStatistics/get-store-total-performance-statistics-list" \
    -H "Authorization: $TOKEN" \
    -H "Content-Type: application/json" \
    -d '{"statisticsMonth": "202512", "pageIndex": 1, "pageSize": 10}')
  END=$(date +%s.%3N)
  TIME=$(echo "$RESPONSE" | grep "TIME:" | cut -d: -f2)
  TIMES+=($TIME)
  echo "第${i}次请求: ${TIME}秒"
done

# 计算平均时间
SUM=0
for t in "${TIMES[@]}"; do
  SUM=$(echo "$SUM + $t" | bc)
done
AVG=$(echo "scale=3; $SUM / ${#TIMES[@]}" | bc)
echo "平均响应时间: ${AVG}秒"
echo ""

# 测试5: 查询不同月份（2025年11月）
echo "--- 测试5: 查询2025年11月数据 ---"
START=$(date +%s.%3N)
RESPONSE5=$(curl -s -w "\nTIME:%{time_total}" -X POST "http://localhost:2011/api/Extend/LqStatistics/get-store-total-performance-statistics-list" \
  -H "Authorization: $TOKEN" \
  -H "Content-Type: application/json" \
  -d '{"statisticsMonth": "202511", "pageIndex": 1, "pageSize": 10}')
END=$(date +%s.%3N)
TIME5=$(echo "$RESPONSE5" | grep "TIME:" | cut -d: -f2)
RESPONSE_BODY5=$(echo "$RESPONSE5" | sed '/TIME:/d')
TOTAL5=$(echo "$RESPONSE_BODY5" | python3 -c "import sys, json; data=json.load(sys.stdin); print(data.get('data', {}).get('pagination', {}).get('total', 0))" 2>/dev/null)
echo "响应时间: ${TIME5}秒"
echo "总记录数: $TOTAL5"
echo ""

echo "=== 性能测试总结 ==="
echo "1. 分页查询（10条）: ${TIME1}秒"
echo "2. 分页查询（50条）: ${TIME2}秒"
echo "3. 门店名称筛选: ${TIME3}秒"
echo "4. 连续5次平均: ${AVG}秒"
echo "5. 不同月份查询: ${TIME5}秒"
echo ""
echo "✅ 所有测试完成"