一个用于并发批量执行大量关键字的robot framework库。

原创已于 2023-12-29 19:25:50 修改 · 933 阅读

6 ·

本内容遵循CC 4.0 BY-SA版权协议

GEO检测

于 2023-12-29 19:25:23 首次发布

本文介绍了一个库，利用Python的concurrent.futures模块实现RobotFramework关键字的并发批量执行，以减少案例执行时间。通过ProcessPoolExecutor并发处理，避免了单线程下执行的性能瓶颈。

这个库用于并发批量执行大量robot framework关键字。从而缩短案例执行时间。

使用concurrent.futures.ProcessPoolExecutor进程池并发执行同一个robot framework关键字很多次。

源代码如下：

import concurrent.futures

from robot.libraries.BuiltIn import BuiltIn
from robot.output.logger import LOGGER

def _get_concurrent_keyword_results(futures):
    results = []
    for future in futures:
        result = future.result()
        if result!="":
            results.append(result)
    BuiltIn().run_keyword("log to console", f"Info: Run keywords {len(futures)} times, successful {len(results)} times")
    return results

def _run_keyword_without_log(keyword, *args):
    # robot不支持并发写xml日志，强行写xml日志可能导致日志被破坏，所以不写xml日志。
    LOGGER.unregister_xml_logger()
    try: 
        return BuiltIn().run_keyword(keyword, *args)
    except Exception as e:
        BuiltIn().run_keyword("log to console", f"Exception: {e}")
        # 如果不raise，返回值就是None，_get_concurrent_keyword_results将分不清楚返回值是None还是发生了异常。但是如果raise e，进程池里的进程可能会被终结，导致后续任务无法执行。
        # raise e
        # 用空字符串代表关键字执行出现了异常
        return  ""

def run_keyword_concurrently_on_each_para_group(keyword, concurrent_number, para_groups):
    '''
    这是并发运行keyword的关键字。
    keyword是要运行的关键字，concurrent_number指定同时最多有几个关键字运行。
    para_groups是一个二维列表，每个内层列表，是keyword的一次运行所需的参数。para_groups的长度就是关键字运行的次数。
    并发运行的进程数最大为20个，否则robot报错Maximum limit of started keywords exceeded.\n
    '''
    with concurrent.futures.ProcessPoolExecutor(int(concurrent_number)) as executor:
        futures = []
        for para_group in para_groups:
            future = executor.submit(_run_keyword_without_log, keyword, *para_group)
            futures.append(future)
        concurrent.futures.wait(futures)
        return _get_concurrent_keyword_results(futures)

标签