python locust 能压测数据库_性能压测工具之 Locust

Locust 是一款基于python语言开发的一款新的性能压测工具，它是以协程(比线程还小)的方式运行。

先看一张 locust 运行图：

源码如下：

#!/usr/bin/env python

#-- coding: utf-8 --

from locust import HttpLocust, TaskSet, task

import subprocess

import json

import time

import random

import hashlib

import os

class UserBer(TaskSet):

def on_start(self):

pass

@task

def pubsend(self):

url = "/pub/send"

f = open("F:/python/data.json", 'r')

data = f.read()

data = json.loads(data)

data["from"]["time"] = int(time.time())

data["from"]["nonce"] = random.randint(100000, 999999)

pubToken = [data["from"]["no"], data["from"]["pub"], data["from"]["pubtoken"]

pubToken.sort()

pubToken = "".join(pubToken)

sha1 = hashlib.sha1(pubToken.encode('utf-8'))

sha1 = sha1.hexdigest()

data["from"]["pubtoken"] = sha1

dataCop = json.dumps(data)

head = {"Content-Type": "application/json"}

req = self.client.post(url=url,data=dataCop,headers=head)

if req.status_code == 200 and req.['success']==True:

print("响应正常！")

else:

print("服务端返回异常！")

class MobileUserLocust(HttpLocust):

task_set = UserBer

host = "http://hlep.his.com"

min_wait = 1000

max_wait = 3000

if name == "main":

os.system("locust -f locust-script.py")

说明：wins 下 pip install locustio ；运行脚本后 在浏览器中输入： localhost:8089