web api性能測試

這邊簡單的給出shell腳本

注意先安裝：wrk和gnuplot

#!/bin/bash# Copyright 2020 Lingfei Kong <colin404@foxmail.com>. All rights reserved.
# Use of this source code is governed by a MIT style
# license that can be found in the LICENSE file.: << EOF
API 性能測試腳本，會自動執行 wrk 命令，采集數據、分析數據并調用 gnuplot 畫圖使用方式 ( 測試 API 性能)：
1. 啟動 iam-apiserver(8080端口)
2. 執行測試腳本: ./wrktest.sh腳本會生成 .dat 的數據文件，每列含義為：并發數 QPS 平均響應時間 成功率使用方式 (對比2次測試結果)
1. 性能測試：./wrktest.sh iam-apiserver http://127.0.0.1:8080/healthz
2. 執行命令： ./wrktest.sh diff apiserver.dat http.dat> Note: 需要確保系統安裝了 wrk 和 gnuplot 工具
EOFiamroot="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd -P)"
wrkdir="${iamroot}/_output/wrk"
jobname="apiserver"
duration="300s"
threads=$((3 * `grep -c processor /proc/cpuinfo`))source "${iamroot}/scripts/lib/color.sh"# Set wrk options
iam::wrk::setup() {#concurrent="200 500 1000 3000 5000 10000 15000 20000 25000 50000 100000 200000 500000 1000000"concurrent="200 500 1000 3000 5000 10000 15000 20000 25000 50000"cmd="wrk -t${threads} -d${duration} -T30s --latency"
}# Print usage infomation
iam::wrk::usage()
{cat << EOFUsage: $0 [OPTION] [diff] URL
Performance automation test script.URL                    HTTP request url, like: http://127.0.0.1:8080/healthzdiff                   Compare two performance test resultsOPTIONS:-h                     Usage information-n                     Performance test task name, default: apiserver-d                     Directory used to store performance data and gnuplot graphic, default: _output/wrkReprot bugs to <colin404@foxmail.com>.
EOF
}# Convert plot data to useable data
function iam::wrk::convert_plot_data()
{echo "$1" | awk -v datfile="${wrkdir}/${datfile}" ' {if ($0 ~ "Running") {common_time=$2}
if ($0 ~ "connections") {connections=$4common_threads=$1
}
if ($0 ~ "Latency   ") {avg_latency=convertLatency($2)
}
if ($0 ~ "50%") {p50=convertLatency($2)
}
if ($0 ~ "75%") {p75=convertLatency($2)
}
if ($0 ~ "90%") {p90=convertLatency($2)
}
if ($0 ~ "99%") {p99=convertLatency($2)
}
if ($0 ~ "Requests/sec") {qps=$2
}
if ($0 ~ "requests in") {allrequest=$1
}
if ($0 ~ "Socket errors") {err=$4+$6+$8+$10
}
}
END {
rate=sprintf("%.2f", (allrequest-err)*100/allrequest)
print connections,qps,avg_latency,rate >> datfile
}function convertLatency(s) {if (s ~ "us") {sub("us", "", s)return s/1000}
if (s ~ "ms") {sub("ms", "", s)return s
}
if (s ~ "s") {sub("s", "", s)return s * 1000
}
}'
}# Remove existing data file
function iam::wrk::prepare()
{rm -f ${wrkdir}/${datfile}
}# Plot according to gunplot data file
function iam::wrk::plot() {gnuplot <<  EOF
set terminal png enhanced #輸出格式為png文件
set ylabel 'QPS'
set xlabel 'Concurrent'
set y2label 'Average Latency (ms)'
set key top left vertical noreverse spacing 1.2 box
set tics out nomirror
set border 3 front
set style line 1 linecolor rgb '#00ff00' linewidth 2 linetype 3 pointtype 2
set style line 2 linecolor rgb '#ff0000' linewidth 1 linetype 3 pointtype 2
set style data linespointsset grid #顯示網格
set xtics nomirror rotate #by 90#只需要一個x軸
set mxtics 5
set mytics 5 #可以增加分刻度
set ytics nomirror
set y2ticsset autoscale  y
set autoscale y2set output "${wrkdir}/${qpsttlb}"  #指定數據文件名稱
set title "QPS & TTLB\nRunning: ${duration}\nThreads: ${threads}"
plot "${wrkdir}/${datfile}" using 2:xticlabels(1) w lp pt 7 ps 1 lc rgbcolor "#EE0000" axis x1y1 t "QPS","${wrkdir}/${datfile}" using 3:xticlabels(1) w lp pt 5 ps 1 lc rgbcolor "#0000CD" axis x2y2 t "Avg Latency (ms)"unset y2tics
unset y2label
set ytics nomirror
set yrange[0:100]
set output "${wrkdir}/${successrate}"  #指定數據文件名稱
set title "Success Rate\nRunning: ${duration}\nThreads: ${threads}"
plot "${wrkdir}/${datfile}" using 4:xticlabels(1) w lp pt 7 ps 1 lc rgbcolor "#F62817" t "Success Rate"
EOF
}# Plot diff graphic
function iam::wrk::plot_diff()
{gnuplot <<  EOF
set terminal png enhanced #輸出格式為png文件
set xlabel 'Concurrent'
set ylabel 'QPS'
set y2label 'Average Latency (ms)'
set key below left vertical noreverse spacing 1.2 box autotitle columnheader
set tics out nomirror
set border 3 front
set style line 1 linecolor rgb '#00ff00' linewidth 2 linetype 3 pointtype 2
set style line 2 linecolor rgb '#ff0000' linewidth 1 linetype 3 pointtype 2
set style data linespoints#set border 3 lt 3 lw 2   #這會讓你的坐標圖的border更好看
set grid #顯示網格
set xtics nomirror rotate #by 90#只需要一個x軸
set mxtics 5
set mytics 5 #可以增加分刻度
set ytics nomirror
set y2tics#set pointsize 0.4 #點的像素大小
#set datafile separator '\t' #數據文件的字段用\t分開set autoscale  y
set autoscale y2#設置圖像的大小 為標準大小的2倍
#set size 2.3,2set output "${wrkdir}/${t1}_${t2}.qps.ttlb.diff.png"  #指定數據文件名稱
set title "QPS & TTLB\nRunning: ${duration}\nThreads: ${threads}"
plot "/tmp/plot_diff.dat" using 2:xticlabels(1) w lp pt 7 ps 1 lc rgbcolor "#EE0000" axis x1y1 t "${t1} QPS","/tmp/plot_diff.dat" using 5:xticlabels(1) w lp pt 7 ps 1 lc rgbcolor "#EE82EE" axis x1y1 t "${t2} QPS","/tmp/plot_diff.dat" using 3:xticlabels(1) w lp pt 5 ps 1 lc rgbcolor "#0000CD" axis x2y2 t "${t1} Avg Latency (ms)", "/tmp/plot_diff.dat" using 6:xticlabels(1) w lp pt 5 ps 1 lc rgbcolor "#6495ED" axis x2y2 t "${t2} Avg Latency (ms)"unset y2tics
unset y2label
set ytics nomirror
set yrange[0:100]
set output "${wrkdir}/${t1}_${t2}.successrate.diff.png"  #指定數據文件名稱
set title "Success Rate\nRunning: ${duration}\nThreads: ${threads}"
plot "/tmp/plot_diff.dat" using 4:xticlabels(1) w lp pt 7 ps 1 lc rgbcolor "#EE0000" t "${t1} Success Rate","/tmp/plot_diff.dat" using 7:xticlabels(1) w lp pt 7 ps 1 lc rgbcolor "#EE82EE" t "${t2} Success Rate"
EOF
}# Start API performance testing
iam::wrk::start_performance_test() {iam::wrk::preparefor c in ${concurrent}dowrkcmd="${cmd} -c ${c} $1"echo "Running wrk command: ${wrkcmd}"result=`eval ${wrkcmd}`iam::wrk::convert_plot_data "${result}"doneecho -e "\nNow plot according to ${COLOR_MAGENTA}${wrkdir}/${datfile}${COLOR_NORMAL}"iam::wrk::plot &> /dev/nullecho -e "QPS graphic file is: ${COLOR_MAGENTA}${wrkdir}/${qpsttlb}${COLOR_NORMAL}
Success rate graphic file is: ${COLOR_MAGENTA}${wrkdir}/${successrate}${COLOR_NORMAL}"
}while getopts "hd:n:" opt;docase ${opt} ind)wrkdir=${OPTARG};;n)jobname=${OPTARG};;?)iam::wrk::usageexit 0;;esac
doneshift $(($OPTIND-1))mkdir -p ${wrkdir}
case $1 in"diff")if [ "$#" -lt 3 ];theniam::wrk::usageexit 0fit1=$(basename $2|sed 's/.dat//g') # 對比圖中紅色線條名稱t2=$(basename $3|sed 's/.dat//g') # 對比圖中粉色線條名稱join $2 $3 > /tmp/plot_diff.datiam::wrk::plot_diff `basename $2` `basename $3`exit 0;;*)if [ "$#" -lt 1 ];theniam::wrk::usageexit 0fiurl="$1"qpsttlb="${jobname}_qps_ttlb.png"successrate="${jobname}_successrate.png"datfile="${jobname}.dat"iam::wrk::setupiam::wrk::start_performance_test "${url}";;
esac

直接使用wrk命令

wrk -t144 -c3000 -d30s -T30s --latency 你的url