78 lines
3.1 KiB
Bash
78 lines
3.1 KiB
Bash
echo "===TEST:disk==="
|
||
|
||
TEST_DIR=${AUTOCHECK_DISK_DIR:-/tmp/auto-check-disk}
|
||
mkdir -p "$TEST_DIR"
|
||
|
||
START_TIME=$(date +%s%N)
|
||
|
||
if command -v fio >/dev/null 2>&1; then
|
||
FIO_IMG="$TEST_DIR/fio.img"
|
||
BW_LOG="$TEST_DIR/ac-bw"
|
||
IOPS_LOG="$TEST_DIR/ac-iops"
|
||
rm -f "$BW_LOG"*.log "$IOPS_LOG"*.log "$FIO_IMG"
|
||
|
||
set +e
|
||
OUTPUT=$(fio --name=autocheck --rw=randrw --rwmixread=70 --bs=4k \
|
||
--size="${AUTOCHECK_DISK_SIZE_MB:-512}M" --iodepth=32 --numjobs=1 \
|
||
--runtime="${AUTOCHECK_DURATION:-30}" --time_based --direct=1 \
|
||
--group_reporting --output-format=normal \
|
||
--write_bw_log="$BW_LOG" --write_iops_log="$IOPS_LOG" \
|
||
--log_avg_msec=1000 --filename="$FIO_IMG" 2>&1)
|
||
EXIT_CODE=$?
|
||
set -e
|
||
|
||
READ_LINE=$(echo "$OUTPUT" | grep -E '^[[:space:]]*read:' | head -1)
|
||
WRITE_LINE=$(echo "$OUTPUT" | grep -E '^[[:space:]]*write:' | head -1)
|
||
READ_IOPS=$(echo "$READ_LINE" | grep -oP 'IOPS=\K[0-9.]+' | head -1)
|
||
READ_BW=$(echo "$READ_LINE" | grep -oP 'BW=\K[0-9.]+' | head -1)
|
||
WRITE_IOPS=$(echo "$WRITE_LINE" | grep -oP 'IOPS=\K[0-9.]+' | head -1)
|
||
WRITE_BW=$(echo "$WRITE_LINE" | grep -oP 'BW=\K[0-9.]+' | head -1)
|
||
SCORE=$(awk -v r="${READ_BW:-0}" -v w="${WRITE_BW:-0}" 'BEGIN{printf "%.1f", r+w}')
|
||
IOPS_TOTAL=$(awk -v r="${READ_IOPS:-0}" -v w="${WRITE_IOPS:-0}" 'BEGIN{printf "%.1f", r+w}')
|
||
|
||
END_TIME=$(date +%s%N)
|
||
DURATION_MS=$(( (END_TIME - START_TIME) / 1000000 ))
|
||
if [ "$EXIT_CODE" -eq 0 ]; then echo 'status:pass'; else echo 'status:fail'; fi
|
||
echo "duration:${DURATION_MS}ms"
|
||
echo "score:${SCORE}"
|
||
echo "metric:iops_total=${IOPS_TOTAL}"
|
||
[ -n "$READ_IOPS" ] && echo "metric:read_iops=${READ_IOPS}"
|
||
[ -n "$WRITE_IOPS" ] && echo "metric:write_iops=${WRITE_IOPS}"
|
||
[ -n "$READ_BW" ] && echo "metric:read_bw=${READ_BW}"
|
||
[ -n "$WRITE_BW" ] && echo "metric:write_bw=${WRITE_BW}"
|
||
[ -n "$READ_BW" ] && [ -n "$WRITE_BW" ] && echo "metric:bw_total=${SCORE}"
|
||
echo "output:${OUTPUT}"
|
||
|
||
# IOPS / 吞吐 曲线（按秒聚合）
|
||
BW_LOG_FILE=$(ls "$BW_LOG"*.log 2>/dev/null | head -1)
|
||
IOPS_LOG_FILE=$(ls "$IOPS_LOG"*.log 2>/dev/null | head -1)
|
||
echo "===SAMPLES==="
|
||
echo "time,iops_total,bw_total"
|
||
if [ -n "$BW_LOG_FILE" ] && [ -n "$IOPS_LOG_FILE" ]; then
|
||
awk -F',' '
|
||
FNR==NR { t=int($1/1000); if($3==0) ri[t]=$2; else wi[t]=$2; seen[t]=1; next }
|
||
{ t=int($1/1000); if($3==0) rb[t]=$2; else wb[t]=$2; seen[t]=1 }
|
||
END { for (k in seen) printf "%d,%.0f,%.1f\n", k, ri[k]+wi[k], (rb[k]+wb[k])/1024 }
|
||
' "$IOPS_LOG_FILE" "$BW_LOG_FILE" | sort -n
|
||
fi
|
||
echo "===END_SAMPLES==="
|
||
|
||
rm -f "$FIO_IMG" "$BW_LOG"*.log "$IOPS_LOG"*.log
|
||
else
|
||
# 无 fio，降级为 stress-ng 磁盘 IO 压测（无 IOPS/吞吐曲线）
|
||
set +e
|
||
OUTPUT=$(stress-ng --iomix "${AUTOCHECK_THREADS:-2}" --iomix-bytes "${AUTOCHECK_DISK_MB:-512}M" --timeout "${AUTOCHECK_DURATION:-30}s" --temp-path /tmp --metrics-brief 2>&1)
|
||
EXIT_CODE=$?
|
||
set -e
|
||
SCORE=$(echo "$OUTPUT" | awk '/^stress-ng:/ && $4=="iomix" { v=$(NF-1)+0; if(v>0) s+=v } END{printf "%.1f", s+0}')
|
||
|
||
END_TIME=$(date +%s%N)
|
||
DURATION_MS=$(( (END_TIME - START_TIME) / 1000000 ))
|
||
if [ "$EXIT_CODE" -eq 0 ]; then echo 'status:pass'; else echo 'status:fail'; fi
|
||
echo "duration:${DURATION_MS}ms"
|
||
echo "score:${SCORE}"
|
||
echo "output:${OUTPUT}"
|
||
fi
|
||
|
||
echo ''
|