Shell虚拟用户测试脚本

higkoo

testMain.sh

1. 
   
2. #!/bin/bash
   
3. # testMain.sh
   
4. ###############################################
   
5. # 脚本接受两个整型参数
   
6. # 第一个参数为虚拟用户数（进程数）
   
7. #                每个虚拟用户一个进程，默认为10
   
8. # 第二个参数为虚拟用户加载的时间间隔，
   
9. #        默认为15（秒）
   
10. # 运行示例：
    
11. #        ./testMain.sh
    
12. #        ./testMain.sh 5 10
    
13. ###############################################
    
14. baseDir=`pwd`
    
15. logDir=$baseDir/logs
    
16. psFile=$baseDir/processStatus.txt
    
17. rtFile=$logDir/responeTime.txt
    
18. vLogFile=$logDir/VuserLog.txt
    
19. 
    
20. #验证传入的第一个参数：虚拟用户数，默认10
    
21. if [[ -z $1 ]];then
    
22.         echo -en "\033[40;33mPlease enter the number of Vuser you want (default 10):\033[0m"
    
23.         read vuserNum
    
24.         if [[ -z $vuserNum ]];then
    
25.                 vuserNum=10
    
26.         fi
    
27.         if [[ $vuserNum =~ ^[[:digit:]]*$ ]]; then
    
28.                 vuserNum=$vuserNum
    
29.         else
    
30.                 vuserNum=10
    
31.         fi
    
32. else
    
33.         vuserNum=$1
    
34. fi
    
35. #验证传入的第二个参数，加载间隔时间，默认为0
    
36. if [[ -n $2 ]];then
    
37.         if [[ $vuserNum =~ ^[[:digit:]]*$ ]]; then
    
38.                 sleepTime=$2
    
39.         else
    
40.                 echo "Use default delay time 15 seconds."
    
41.                 sleepTime=15
    
42.         fi
    
43. else
    
44.         echo -en "\033[40;33mPlease enter a delay time for each Vuser (default 15s):\033[0m"
    
45.         read sleepTime
    
46.         if [[ -z $sleepTime ]];then
    
47.                 sleepTime=15
    
48.         fi
    
49.         if [[ $sleepTime =~ ^[[:digit:]]*$ ]]; then
    
50.                 vuserNum=$sleepTime
    
51.         else
    
52.                 vuserNum=15
    
53.         fi       
    
54. fi
    
55. if [[ -z $3 ]];then
    
56.         echo -en "\033[40;33mPlease enter the URL for test(default myBlog ^_^):\033[0m"
    
57.         read testURL
    
58.         if [[ -z $vuserNum ]];then
    
59.                 testURL="http://higkoo.blog.sohu.com"
    
60.         fi
    
61. fi
    
62. #如果日志目录不存在，则创建它
    
63. if [ ! -x "$logDir" ];then
    
64.         mkdir "$logDir"
    
65. fi
    
66. #新建/覆写psFile、logFile
    
67. echo "Vuser List:" > $psFile
    
68. echo "Start:" > $vLogFile
    
69. echo -e "\033[40;32mStart $vuserNum Vuser(s),each after $sleepTime second(s).\033[0m"
    
70. #创建虚拟用户进程
    
71. for((i=1;i<=$vuserNum;i++))
    
72. do
    
73. #参数1为虚拟用户的ID
    
74.         sleep $sleepTime
    
75.         /bin/bash $baseDir/request.sh $i $testURL > /dev/null 2>&1 &
    
76.         echo -e "\033[40;32m-- Add Vuser$i PID=$! start:"`date '+%Y%m%d%H%M%S.%N'`"\033[0m"
    
77. done
    
78. echo -e "\033[40;32mFinish Load ! There're $vuserNum Vuser(s) running.\033[0m"
    
79. echo -e "\033[40;36mVuser status controllor file: $psFile\033[0m"
    
80. echo -e "\033[40;36mVuser's status logfile: $vLogFile\033[0m"
    
81. echo -e "\033[40;36mResponeTime file: $rtFile\033[0m"
    

复制代码

1. 
   
2. #!/bin/bash
   
3. # request.sh
   
4. ###############################################
   
5. # 脚本由testMain.sh调用
   
6. # 接受一个参数：虚拟用户ID
   
7. ###############################################
   
8. #定义常规变量
   
9. baseDir=`pwd`
   
10. logDir=$baseDir/logs
    
11. psFile=$baseDir/processStatus.txt
    
12. rtFile=$logDir/responeTime.txt
    
13. vLogFile=$logDir/VuserLog.txt
    
14. myPid=$$
    
15. #计算响应时间函数，精确到毫秒，传入格式为`date '+%s.%N'`
    
16. getRetime(){
    
17.                 tSecond=`echo $2,$1|awk -F, '{print $1-$2}'`               
    
18. }
    
19. #将状态写入到文件
    
20. echo "Vuser$1,$myPid,1" >> $psFile
    
21. echo "Vuser:$1,pid:$myPid,start:"`date '+%Y%m%d.%H%M%S.%N'` >> $vLogFile
    
22. #初始化暂停标识
    
23. isPaused=0
    
24. #持续运行
    
25. while( true )
    
26. do
    
27. #获取运行状态，确定是否继续运行
    
28.         myStatus=`cat $psFile|grep $myPid|awk -F, '{print $3}'`
    
29.         if [ -z $myStatus ] || [ $myStatus = 0 ]
    
30.         then
    
31.                 echo "Vuser:$1,pid:$myPid,stop:"`date '+%Y%m%d.%H%M%S.%N'` >> $vLogFile
    
32.                 break
    
33.         else
    
34.                 if [ $myStatus = 2 ];then
    
35.                         if [ $isPaused = 0 ];then
    
36.                                 echo "Vuser:$1,pid:$myPid,pause:"`date '+%Y%m%d.%H%M%S.%N'` >> $vLogFile
    
37.                         fi
    
38.                         isPaused=1
    
39.                         sleep 1
    
40.                         continue
    
41.                 fi
    
42.         fi
    
43. #判断是否为Restart
    
44.                 if [ $isPaused = 1 ];then
    
45.                         echo "Vuser:$1,pid:$myPid,restart:"`date '+%Y%m%d.%H%M%S.%N'` >> $vLogFile
    
46.                         isPaused=0
    
47.                 fi
    
48. #测试代码
    
49.         startTime=`date '+%s.%N'`
    
50.         wget -S --output-document=$logDir/$$ $2 2> $logDir/$$.txt
    
51.         endTime=`date '+%s.%N'`
    
52.         #grep找到返回0，不存在返回1
    
53.         #查找服务器是否正确响应       
    
54.         if [ `grep "HTTP/1.1" $logDir/$$.txt |awk '{print $2}'` -ne "200" ]
    
55.         then
    
56.                 cat $logDir/$$.txt >> $logDir/$$.error.txt
    
57.                 cat $logDir/$$ >> $logDir/$$.error.txt
    
58.         else
    
59.                 #查找程序的返回值是否正确
    
60.                 grep "html" $logDir/$$        > /dev/null
    
61.                 if [ $? = 1 ]
    
62.                 then
    
63.                         echo "Cann't find :html"
    
64.                 else
    
65.                         getRetime $startTime $endTime
    
66.                         echo $tSecond >> $rtFile
    
67.                 fi               
    
68.         fi
    
69. #删除过程中产生的日志
    
70. rm -f $logDir/$$ $logDir/$$.txt
    
71. done
    

复制代码

把两个脚本放在同一个目录运行 `bash testMain.sh` ，会在同目录下生成一个processStatus.txt文件，里面是当前在运行的虚拟用户的信息和状态，把状态改为0，虚拟用户就停了，改为2，虚拟用户就暂停，再改为1就继续运行。

   同时生成Log目录，里面记录有响应时间和虚拟用户加载日志。
  脚本没有设置运行时长，停止可以修改processStatus.txt里虚拟用户对应的状态，或删除processStatus.txt里的内容，或删除这个文件，程序即停止运行。
或运行 kill -9 `ps aux | grep request.sh | cut -c9-14` 2>/dev/null 杀死所有测试进程。

    此脚本纯粹练手，修改一下当然也可以小用一下。欢迎拍砖！