90 行
2.2 KiB
Bash
可执行文件
90 行
2.2 KiB
Bash
可执行文件
#!/usr/bin/env bash
|
|
set -euo pipefail
|
|
|
|
BASE_URL="${BASE_URL:-http://127.0.0.1:3000}"
|
|
DEVICE_FINGERPRINT="${DEVICE_FINGERPRINT:-benchmark-assets}"
|
|
ADMIN_IDENTIFIER="${ADMIN_IDENTIFIER:-finance_mgr@hyinda.com}"
|
|
ADMIN_PASSWORD="${ADMIN_PASSWORD:-FinanceMgrPassword1234}"
|
|
ITERATIONS="${ITERATIONS:-10}"
|
|
ASSET_EXACT_ENDPOINT="${ASSET_EXACT_ENDPOINT:-/api/assets?q=ZC-001&mine=0}"
|
|
ASSET_FUZZY_ENDPOINT="${ASSET_FUZZY_ENDPOINT:-/api/assets?q=%E6%B5%8B%E8%AF%95&mine=0}"
|
|
|
|
if ! command -v curl >/dev/null 2>&1; then
|
|
echo "[ERROR] curl not found" >&2
|
|
exit 1
|
|
fi
|
|
|
|
if ! command -v jq >/dev/null 2>&1; then
|
|
echo "[ERROR] jq not found" >&2
|
|
exit 1
|
|
fi
|
|
|
|
TMP_DIR="$(mktemp -d)"
|
|
cleanup() {
|
|
rm -rf "$TMP_DIR"
|
|
}
|
|
trap cleanup EXIT
|
|
|
|
login() {
|
|
local out_file="$1"
|
|
local status
|
|
status="$(curl -sS -o "$out_file" -w "%{http_code}" \
|
|
-X POST "$BASE_URL/api/auth/login" \
|
|
-H "content-type: application/json" \
|
|
-H "x-device-fingerprint: $DEVICE_FINGERPRINT" \
|
|
-d "{\"identifier\":\"$ADMIN_IDENTIFIER\",\"password\":\"$ADMIN_PASSWORD\"}")"
|
|
|
|
if [[ "$status" != "200" ]]; then
|
|
echo "[ERROR] login failed: HTTP $status" >&2
|
|
cat "$out_file" >&2
|
|
exit 1
|
|
fi
|
|
|
|
jq -er '.data.accessToken' "$out_file"
|
|
}
|
|
|
|
benchmark_endpoint() {
|
|
local label="$1"
|
|
local endpoint="$2"
|
|
local token="$3"
|
|
|
|
local total_ms=0
|
|
local max_ms=0
|
|
local min_ms=999999
|
|
|
|
for _ in $(seq 1 "$ITERATIONS"); do
|
|
local body_file="$TMP_DIR/assets.json"
|
|
local elapsed
|
|
elapsed="$(curl -sS -o "$body_file" -w "%{time_total}" \
|
|
-H "authorization: Bearer $token" \
|
|
"$BASE_URL$endpoint")"
|
|
|
|
local elapsed_ms
|
|
elapsed_ms="$(python3 - <<PY
|
|
value = float("$elapsed")
|
|
print(int(round(value * 1000)))
|
|
PY
|
|
)"
|
|
|
|
total_ms=$((total_ms + elapsed_ms))
|
|
if (( elapsed_ms > max_ms )); then
|
|
max_ms=$elapsed_ms
|
|
fi
|
|
if (( elapsed_ms < min_ms )); then
|
|
min_ms=$elapsed_ms
|
|
fi
|
|
done
|
|
|
|
local avg_ms=$((total_ms / ITERATIONS))
|
|
echo "$label $endpoint avg=${avg_ms}ms min=${min_ms}ms max=${max_ms}ms iterations=$ITERATIONS"
|
|
}
|
|
|
|
echo "[1/2] Login"
|
|
TOKEN="$(login "$TMP_DIR/login.json")"
|
|
|
|
echo "[2/2] Assets search benchmark"
|
|
benchmark_endpoint "exact" "$ASSET_EXACT_ENDPOINT" "$TOKEN"
|
|
benchmark_endpoint "fuzzy" "$ASSET_FUZZY_ENDPOINT" "$TOKEN"
|
|
|
|
echo "[OK] Assets benchmark completed"
|