Update nvme_test.sh

Author: gpas45

nvme_test.sh

@@ -1,68 +1,92 @@
 #!/bin/bash
-# Тест дисков NVMe
-
+# Тест дисков NVMe (профиль CrystalDiskMark: SEQ1M Q8T1, SEQ128K Q32T1, RND4K Q32T8, RND4K Q1T1)
+# Парсинг через JSON-вывод fio + jq, все значения в MB/s (десятичные, как у fio/CDM)
 set -u
 set -o pipefail
 
-# Проверка: установлен ли fio
-if ! command -v fio >/dev/null 2>&1; then
-  echo "Ошибка: 'fio' не найден (команда отсутствует в PATH). Установите fio и повторите."
-  echo "Debian/Ubuntu: sudo apt install -y fio"
-  exit 127
-fi
+# --- Проверка зависимостей ---------------------------------------------------
+for cmd in fio jq; do
+  if ! command -v "$cmd" >/dev/null 2>&1; then
+    echo "Ошибка: '$cmd' не найден (команда отсутствует в PATH)."
+    echo "Debian/Ubuntu: sudo apt install -y $cmd"
+    exit 127
+  fi
+done
 
+# --- Параметры ----------------------------------------------------------------
 TESTFILE_PATH='/980PRO/fiotest.tmp'
 TESTFILE_SIZE='1000m'
 TEST_LOOPS='5'
 
+# --- Проверка каталога ---------------------------------------------------------
+TESTDIR=$(dirname "$TESTFILE_PATH")
+if [ ! -d "$TESTDIR" ] || [ ! -w "$TESTDIR" ]; then
+  echo "Ошибка: каталог '$TESTDIR' не существует или недоступен для записи."
+  exit 1
+fi
+
+# --- Уборка тестового файла при любом выходе -----------------------------------
+trap 'rm -f "$TESTFILE_PATH"' EXIT
+
 FIO_BASE=(fio
   --loops="$TEST_LOOPS"
   --size="$TESTFILE_SIZE"
   --filename="$TESTFILE_PATH"
   --ioengine=libaio
   --direct=1
+  --group_reporting
+  --output-format=json
 )
 
 pause_io() { sync; sleep 3; }
 
-# Оставляем исходную логику парсинга: 3-е поле и вырезаем "bw=...MiB/s"
-# (как было: substr($3, 2, length($3)-3))
-extract_bw() {
-  grep -E "$1" | awk '{print substr($3, 2, length($3)-3)}'
+# run_test <read|write> <доп. параметры fio...>
+# Возвращает скорость в MB/s (десятичных, bw_bytes/1e6) или "ERR" при сбое.
+run_test() {
+  local direction="$1"; shift
+  local json bw
+  if ! json=$("${FIO_BASE[@]}" "$@" 2>/dev/null); then
+    echo "ERR"; return 1
+  fi
+  bw=$(jq -r --arg d "$direction" '.jobs[0][$d].bw_bytes' <<<"$json" 2>/dev/null)
+  if [ -z "$bw" ] || [ "$bw" = "null" ]; then
+    echo "ERR"; return 1
+  fi
+  awk -v b="$bw" 'BEGIN { printf "%.1f", b / 1000000 }'
 }
 
 print_header() {
-  echo " +----------------------------------------+"
-  printf " | %12s | %10s | %10s |\n" "Mode" "Read" "Write"
-  echo " +----------------------------------------+"
+  echo " +------------------------------------------------+"
+  printf " | %12s | %14s | %14s |\n" "Mode" "Read, MB/s" "Write, MB/s"
+  echo " +------------------------------------------------+"
 }
 print_row() {
-  printf " | %12s | %10s | %10s |\n" "$1" "$2" "$3"
-  echo " +----------------------------------------+"
+  printf " | %12s | %14s | %14s |\n" "$1" "$2" "$3"
+  echo " +------------------------------------------------+"
 }
 
+# --- Тесты ---------------------------------------------------------------------
 pause_io
-SEQ1MQ8T1R=$("${FIO_BASE[@]}" --name=SeqReadQ8  --bs=1m   --iodepth=8  --rw=read  | extract_bw 'READ:')
+SEQ1MQ8T1R=$(run_test read  --name=SeqReadQ8  --bs=1m --iodepth=8 --rw=read)
 pause_io
-SEQ1MQ8T1W=$("${FIO_BASE[@]}" --name=SeqWriteQ8 --bs=1m   --iodepth=8  --rw=write | extract_bw 'WRITE:')
-
+SEQ1MQ8T1W=$(run_test write --name=SeqWriteQ8 --bs=1m --iodepth=8 --rw=write)
 print_header
 print_row "SEQ1MQ8T1" "$SEQ1MQ8T1R" "$SEQ1MQ8T1W"
 
 pause_io
-SEQ128KQ32T1R=$("${FIO_BASE[@]}" --name=SeqReadQ32  --bs=128k --iodepth=32 --rw=read  | extract_bw 'READ:')
+SEQ128KQ32T1R=$(run_test read  --name=SeqReadQ32  --bs=128k --iodepth=32 --rw=read)
 pause_io
-SEQ128KQ32T1W=$("${FIO_BASE[@]}" --name=SeqWriteQ32 --bs=128k --iodepth=32 --rw=write | extract_bw 'WRITE:')
+SEQ128KQ32T1W=$(run_test write --name=SeqWriteQ32 --bs=128k --iodepth=32 --rw=write)
 print_row "SEQ128KQ32T1" "$SEQ128KQ32T1R" "$SEQ128KQ32T1W"
 
 pause_io
-RND4KQ32T8R=$("${FIO_BASE[@]}" --name=RandReadQ32  --bs=4k --iodepth=32 --numjobs=8 --rw=randread  | extract_bw 'READ:')
+RND4KQ32T8R=$(run_test read  --name=RandReadQ32  --bs=4k --iodepth=32 --numjobs=8 --rw=randread)
 pause_io
-RND4KQ32T8W=$("${FIO_BASE[@]}" --name=RandWriteQ32 --bs=4k --iodepth=32 --numjobs=8 --rw=randwrite | extract_bw 'WRITE:')
+RND4KQ32T8W=$(run_test write --name=RandWriteQ32 --bs=4k --iodepth=32 --numjobs=8 --rw=randwrite)
 print_row "RND4KQ32T8" "$RND4KQ32T8R" "$RND4KQ32T8W"
 
 pause_io
-RND4KQ1T1R=$("${FIO_BASE[@]}" --name=RandReadQ1  --bs=4k --iodepth=1 --rw=randread  | extract_bw 'READ:')
+RND4KQ1T1R=$(run_test read  --name=RandReadQ1  --bs=4k --iodepth=1 --rw=randread)
 pause_io
-RND4KQ1T1W=$("${FIO_BASE[@]}" --name=RandWriteQ1 --bs=4k --iodepth=1 --rw=randwrite | extract_bw 'WRITE:')
+RND4KQ1T1W=$(run_test write --name=RandWriteQ1 --bs=4k --iodepth=1 --rw=randwrite)
 print_row "RND4KQ1T1" "$RND4KQ1T1R" "$RND4KQ1T1W"