寫了一個簡單的http benchmark指令碼

apache benchmark只能測一個單一url,我用shell寫了一個benchmark指令碼，支援並發測試，指定參數，不過沒有對回應時間和ret code做統計，適用在伺服器端做精確統計的情況。

#! /bin/bash

concurrency=50
workdirectory=/tmp/httpbenchmark_workdir/
inputurlfile=


function usage() {
  echo "Http Benchmark is used for sending http request in pararrel."
  echo "Usage: ./http_benchmark.sh [-c int] [-i string] [-w string]"
  echo "-c: Specify concurrency number.(Default: 50)"
  echo "-i: Specify input file, if it's empty,then read from standard input.(Default: \"\")"
  echo "-w: Specify work directory.(Default: \"/tmp/httpbenchmark_workdir/\")"
}


function thread_run() {
  fileno=$1
  while read url
  do
    curl $url -s -o /dev/null
  done < ${workdirectory}/spilits_${fileno}.lst 
}


while getopts 'c:i:w:h' OPTION
do
  case $OPTION in
    c) concurrency="$OPTARG"
       ;;
    i) inputurlfile="$OPTARG"
       ;;
    w) workdirectory="$OPTARG"
       ;;
    h) usage
       exit 0
       ;;
    ?) usage
       exit 1
       ;;
  esac
done
shift $((OPTIND - 1)) 


echo "Concurrency=${concurrency}"
echo "Workdirectory=${workdirectory}"
echo "InputUrlFile=${inputurlfile}"
rm -rf $workdirectory
mkdir -p $workdirectory


# split url list.
echo "Begin splitting urls..."
lineno=0
if [ -z $inputurlfile ]; then
  while read line
  do
    # as the 'split' command differs between different distribution, we do it manually.
    if [ -n $line ]; then
      fileno=`expr $lineno % $concurrency`
      echo $line >> ${workdirectory}/spilits_${fileno}.lst
      let lineno=$lineno+1
    fi
  done
else
  while read line
  do
    # as the 'split' command differs between different distribution, we do it manually.
    if [ -n $line ]; then
      fileno=`expr $lineno % $concurrency`
      echo $line >> ${workdirectory}/spilits_${fileno}.lst
      let lineno=$lineno+1
    fi
  done < $inputurlfile
fi
echo "Splitting done."


# 'curl' in pararrel.
for i in `seq 1 $concurrency`
do
  let pos=$i-1
  thread_run $pos &
done


# wait
echo "Waiting subprocesses joined.."
wait
echo "Benchmark finished!Please inspect server log."


# clean work directory.
rm -rf $workdirectory