python多进程并行之multiprocessing

最新推荐文章于 2024-05-25 02:30:00 发布

lenka_qin

最新推荐文章于 2024-05-25 02:30:00 发布

阅读量172

点赞数

分类专栏： Python

本文链接：https://blog.youkuaiyun.com/oHuandream/article/details/114005615

版权

Python 专栏收录该内容

1 篇文章

订阅专栏

本文介绍了一种使用Python的多线程池实现的测试脚本，通过数据库获取用例列表，每个用例随机执行并记录运行时间和结果。展示了如何设置并发数，以及测试开始和结束的时间戳计算，以评估并发耗时。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >


import datetime
import random
import time
from multiprocessing.dummy import Pool as ThreadPool

# 修改为从数据库中读取用例列表


def get_case_list():
    return [i for i in range(1, 20)]

# 具体的执行方法， 记录测试结果和测试时间


def run_case(case_id):
    run_time = random.randrange(1, 10)
    time.sleep(run_time)
    print('run case: %d, result is pass, run time is : %d seconds\n' % (case_id, run_time))


# 设置并发数量
threadNum = 5
# 设置多进程执行
pool = ThreadPool(threadNum)

# 格式化成2016-03-20 11:45:39形式
# print(f'并发数为: {threadNum}')
start = datetime.datetime.now()
print('Start:', start)
pool.map(run_case, get_case_list())
end = datetime.datetime.now()
print('End:', end)
print(f'{threadNum}个进程并发耗时: {(end - start).seconds} 秒')
pool.close()  # 调用join之前，先调用close函数，否则会出错。执行完close后不会有新的进程加入到pool,join函数等待所有子进程结束
pool.join()