【Python】模块学习之locust性能测试

阅读目录

• 背景
• 安装
• 测试代码
• 运行命令

---

欢迎转载，也请在明显位置注明出处 ：https://www.cnblogs.com/Detector/p/8254158.html 谢谢！

背景

locust是一个python的第三方库，用于做性能测试，可使用多台机器同时对一台服务器进行压测，使用其中一台机器作为主节点，进行分布式管理

博主测试接口的时候一直是使用python脚本，后来有个需求需要做性能测试，我第一时间就想到了locust，不需要再对测试脚本进行编辑，而且对测试机的性能要求较低

安装

locust的安装很简单，可以直接使用pip

pip install locustio

测试代码

说明：下列代码中只有sms()函数是最终测试的函数 ，其实的都是博主测试过程中的加密算法的辅助函数

?

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58

#! /usr/bin/python # coding:utf-8 """ @author:Bingo.he @file: locust_practis.py @time: 2017/12/29 """ from locust import HttpLocust, TaskSet, task import hashlib   class UserBehavior(TaskSet):       @staticmethod     def param_order(dic):         """         参数转换函数         :param dic: 字典类型参数         :return: key1=value1&key2=value2         """         l = []         if isinstance(dic, dict):             for k, v in dic.items():                 l.append(k + '=' + str(v))             l.sort()             # print ('&'.join(l))             return '&'.join(l)         else:             return 'Not a dict'       def get_signMsg(self, soure_para, bizKey):         """         API密钥生成函数         :param soure_para : 传递的参数         :param bizKey:         :return: str         """         param = self.param_order(soure_para)         return hashlib.md5((param + bizKey).encode('utf8')).hexdigest()       def pack_param(self):         soure_para = {             'bizNo': '0000001',             'mobiles': '18565100000,18565100001,18565100002,18565100003,18565100004,18565100005,18565100006,','content': u'{}',         }         sign_msg = self.get_signMsg(soure_para, 'GyCK9XC4qYbs')         soure_para['signMsg'] = sign_msg  # 添加签名信息         return soure_para       @task(1)  # 表示第一个执行，可以通过它组合指定场景     def sms(self):         param = self.pack_param()         self.client.post("/sms", param)     class WebsiteUser(HttpLocust):     task_set = UserBehavior     min_wait = 300  # 不同请求间间隔时间     max_wait = 600

运行命令

locust -f ../locust_files/locust_practice.py --host=http://example.com

参数说明

-f 指定文件

--host 指定要压测的服务器

 一般是监听在*：8089上，所以直接访问localhost：8089即可，图形页面如下