python异步压测代码

下面是通过python异步压测代码main.py文件

#-*- coding:utf-8 -*-

import sys
import os
import time

if len(sys.argv) != 3:
        print("bad param")
        sys.exit()
qps = int(sys.argv[1])
filename = sys.argv[2]

duration = 1.0 / qps
start = time.time() #单位为s
count = 0
while True:
        f = open(filename)
        for line in f:
                line = line.strip()
                count += 1
                command = "wget -q \"%s\" -O /dev/null &" % (line)
                print(command + ' ' + time.strftime('%Y-%m-%d %H:%M:%S'))
                os.system(command)
                if count % 3000 == 0:
                        os.system("rm -rf result/*")
            
                expect = start + duration * count
                cur = time.time()
                if cur > expect:
                        sleep = 0
                else:
                        sleep = expect - cur
                print(sleep)
                time.sleep(sleep)  # 单位为s，可以为小数，0.1就是休眠100毫秒

执行命令

python main.py 10 filename.txt

10:表示10qps

filename.txt文件里面是链接地址