python压测demo

最新推荐文章于 2025-11-28 16:05:06 发布

原创最新推荐文章于 2025-11-28 16:05:06 发布 · 1.4k 阅读

5 ·

CC 4.0 BY-SA版权

工具专栏收录该内容

13 篇文章

订阅专栏

此博客主要围绕Python压测展开，提供了压测的demo。借助Python进行压测，可对系统性能进行评估，在信息技术领域，压测是保障系统稳定运行的重要手段。

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()