lustre/tests/rundbench

   1 #!/bin/sh
   2
   3 LUSTRE=${LUSTRE:-$(cd $(dirname $0)/..; echo $PWD)}
   4 . $LUSTRE/tests/test-framework.sh
   5
   6 usage() {
   7     echo "-C use chroot instead of cd"
   8     echo "-D DIR - use 'DIR' as work directory"
   9     echo
  10     exit;
  11 }
  12
  13 PATH=${DBENCH_LIB}:${PATH}
  14 MOUNT=${MOUNT:-/mnt/lustre}
  15 DIR=${DIR:-$MOUNT/`hostname`}
  16
  17 PREFIX="on"
  18 while getopts "CD:" opt $*; do
  19         case $opt in
  20                 D) [ -d $OPTARG ] && DIR=$OPTARG ;;
  21                 C) CHROOT="yes" ;;
  22                 \?) usage ;;
  23         esac
  24 done
  25
  26 mkdir -p $DIR
  27 TGT=$DIR/client.txt
  28 CLIENT_PREFIX="${DBENCH_LIB} /usr/share/dbench /usr/local/share /usr/lib/dbench"
  29 CLIENT_FILE="client.txt client_plain.txt dbench_client"
  30 if ! which dbench > /dev/null 2>&1 ; then
  31     [ "$MISSING_DBENCH_OK" ] || { error "dbench is not installed !" && exit 3; }
  32     skip "$0: dbench is not installed"
  33     exit 0
  34 fi
  35 CLIENT=""
  36
  37 for prefix in $CLIENT_PREFIX; do
  38         for file in $CLIENT_FILE; do
  39                 if [ -s "${prefix}/${file}" ]; then
  40                         CLIENT="${prefix}/${file}";
  41                         break;
  42                 fi
  43         done
  44         [ "x$CLIENT" != "x" ] && break;
  45 done
  46
  47 if [ -n "$SRC" -a -s "$SRC" ]; then
  48         CLIENT=${SRC}
  49 fi
  50
  51 [ ! -s "$CLIENT" ] && \
  52     skip "$0: no client file found for dbench DBENCH_LIB=$DBENCH_LIB SRC=$SRC" && \
  53         exit 0
  54
  55 [ ! -s "$TGT" ] && echo "copying $CLIENT to $TGT" && cp $CLIENT $TGT
  56 [ ! -s "$TGT" ] && \
  57     echo "$0: $TGT file doesn't exist after cp $CLIENT $TGT" && exit 1
  58
  59 if [ "x$CHROOT" == "xyes" ]; then
  60         echo "copying necessary libs to $DIR"
  61         cp `which dbench` $DIR
  62         LIBS71=$(ldd $DIR/dbench|sed -e 's/\t*//' -e 's/.*=> //' -e 's/ .*//' -e 's/^\///')
  63         (cd / && tar chf - $LIBS71) | (cd $DIR && tar xvf -)
  64         [ $? != 0 ] && echo "can't copy libs $LIBS71 to $DIR" && exit 1
  65         RUN="chroot $DIR"
  66         PREFIX="in"
  67         PATH=.:/:$PATH
  68 fi
  69
  70 shift $((OPTIND - 1))
  71
  72 trap '
  73 echo kill dbench main pid=$DBENCHPID
  74 kill $DBENCHPID
  75 rm -rf dbench $LIBS71 client.txt
  76 exit 0
  77 ' TERM
  78
  79 cd $DIR
  80 echo "running 'dbench $@' $PREFIX $PWD at `date`"
  81
  82 $RUN dbench -c client.txt $@ &
  83 DBENCHPID=$!
  84 echo "dbench PID=$DBENCHPID"
  85 wait $DBENCHPID
  86 RC=$?
  87 [ $RC -ne 0 ] && killall -9 dbench
  88
  89 rm -rf dbench $LIBS71 client.txt
  90 exit $RC