python压测demo

最新推荐文章于 2024-08-14 15:42:21 发布

weiwei200

最新推荐文章于 2024-08-14 15:42:21 发布

阅读量1.3k

点赞数 1

分类专栏：工具

本文链接：https://blog.csdn.net/weiwei200/article/details/92591635

版权

工具专栏收录该内容

13 篇文章 1 订阅

订阅专栏

import requests
import time
import concurrent.futures
import json


class APITest(object):

    def __init__(self):
        self.url = 'http://'
        self.token = ''
        self.headers = {'Content-Type': 'application/json',
                        'app_token': self.token}

        self.total_time = 0

        self.data = {

        }

    def common_get_events_list(self):
        t = time.time()
        ret = requests.post(url=self.url, headers=self.headers, data=json.dumps(self.data))
        # print(ret)
        end = (time.time() - t) % 60
        return end

    def time_total(self):
        with concurrent.futures.ThreadPoolExecutor(max_workers=5000) as executor:
            res = {executor.submit(self.common_get_events_list) for i in range(10)}

            for future in concurrent.futures.as_completed(res):
                try:
                    data = future.result()
                except Exception as exc:
                    pass
                else:
                    self.total_time += data
        print('common_get_events_list:接口响应的总时间为%s秒' % str(self.total_time))
        print('common_get_events_list:平均时间为%s秒' % str(self.total_time / 10))
        print('-' * 50)


if __name__ == '__main__':
    api = APITest()
    api.common_get_events_list()
    api.time_total()