bench

   1 #!/bin/sh
   2
   3 _root="$(dirname "${0}")"; . "${_root}/lib/sh/env.sh"
   4 want log suexec
   5
   6 [ "$(hostname -s)" = "nabihinew" ] || wtf "Run this on nabihinew only!"
   7
   8 suexec "${0}" "${@}"
   9
  10 want stopwatch
  11
  12 # Environment and needed stuff
  13 tests="${_root}/tests"
  14
  15 # Log maintenance
  16 logs="${_root}/log"
  17 [ -d "${logs}" ] || mkdir -p "${logs}"
  18 run_log="${logs}/bench-$(date +%Y-%m-%d_%H:%M:%S).log"
  19
  20 # Scratch directory maintenance
  21 scratch="/bonnie"
  22 clean_scratch() {
  23   [ -d "${scratch}" ] || mkdir -p "${scratch}"
  24   mount | grep -q ' on '"${scratch}"' ' && umount "${scratch}" || true
  25   find "${scratch}" ${bonnie_uid:+-uid "${bonnie_uid}"} -xdev -delete
  26 }
  27 clean_scratch
  28
  29 # Bonnie setup
  30 bonnie_uid="4444"
  31 bonnie_opts="-q -d /bonnie -u ${bonnie_uid} -s 12288 -n 16:16384:0 -f 0"
  32 bonnie_log="${logs}/bonnie-%s-$(date +%Y-%m-%d_%H:%M:%S).csv"
  33 run_bonnie() {
  34   chown "${bonnie_uid}" "${scratch}"
  35   /usr/local/sbin/bonnie++ ${bonnie_opts} -m "${test}" >> "$(printf "${bonnie_log}" "${test}")"
  36 }
  37
  38 # Default methods
  39 setup() {
  40   clean_scratch
  41 }
  42 run() {
  43   run_bonnie
  44 }
  45 teardown() {
  46   clean_scratch
  47 }
  48
  49 # Main event
  50 if [ "${#}" -eq 0 ]
  51 then
  52   log "Considering tests: $(cd "${tests}"; echo *)" | tee -a "${run_log}"
  53   script -a "${run_log}" sh -c "( cd '${tests}'; ls -1 ) | xargs -L1 '${0}'"
  54 else
  55   test="${1}"; shift
  56   testfile="${tests}/${test}"
  57   [ -f "${testfile}" ] || wtf "Test '${test}' does not exist"
  58   ( log "Preparing test: ${test}"
  59     . "${testfile}"
  60     setup
  61     sync
  62     log "$(stopwatch run start)"
  63     trap "teardown" EXIT HUP INT TERM KILL
  64     run; result=$?
  65     log "$(stopwatch run stop) Test result ${result}"
  66     teardown
  67   )
  68 fi