#######################################################################
#!/bin/bash
# Script para comparar mq-deadline vs bfq automaticamente
# Requer: sudo, fio

DISK="/dev/sda" # ajuste se necessário
WORKDIR="$HOME/teste-scheduler"
RESULTS="$WORKDIR/resultados.txt"
SUMMARY="$WORKDIR/resumo.txt"

mkdir -p "$WORKDIR"
rm -f "$RESULTS" "$SUMMARY"
echo "== Teste iniciado em $(date) ==" > "$SUMMARY"


rodar_teste() {
local sched="$1"
echo "$sched" | sudo tee /sys/block/sda/queue/scheduler >/dev/null
echo ">> Scheduler: $sched" | tee -a "$RESULTS"

total_bw=0
for test in "seqread:--rw=read --bs=1M --size=1G" \
"seqwrite:--rw=write --bs=1M --size=1G" \
"randread:--rw=randread --bs=4k --size=512M" \
"randwrite:--rw=randwrite --bs=4k --size=512M"
do
name=${test%%:*}
opts=${test#*:}
echo "Executando $name..." | tee -a "$RESULTS"

bw=$(fio --name=$name $opts --numjobs=1 --runtime=15s --time_based \
--filename="$WORKDIR/testfile" 2>&1 | \
grep "BW=" | awk -F'BW=' '{print $2}' | awk '{print $1}')

echo "$name: $bw" | tee -a "$RESULTS"
# Converter para KB/s
unit=$(echo "$bw" | grep -o '[KMG]iB/s')
value=$(echo "$bw" | grep -o '^[0-9.]*')

case $unit in
KiB/s) kb=$(echo "$value" | awk '{print int($1)}') ;;
MiB/s) kb=$(echo "$value" | awk '{print int($1*1024)}') ;;
GiB/s) kb=$(echo "$value" | awk '{print int($1*1024*1024)}') ;;
*) kb=0 ;;
esac

total_bw=$((total_bw + kb))
done

echo "$sched total score (KB/s): $total_bw" >> "$SUMMARY"
echo $total_bw
}

# Rodar para os dois schedulers
score_deadline=$(rodar_teste mq-deadline)
score_bfq=$(rodar_teste bfq)


echo "" >> "$SUMMARY"
if [ "$score_deadline" -gt "$score_bfq" ]; then
echo "Melhor scheduler para este sistema: mq-deadline" | tee -a "$SUMMARY"
else
echo "Melhor scheduler para este sistema: bfq" | tee -a "$SUMMARY"
fi

echo "== Teste concluído em $(date) ==" >> "$SUMMARY"
cat "$SUMMARY"
##################################################################################