这个库用于并发批量执行大量robot framework关键字。从而缩短案例执行时间。
使用concurrent.futures.ProcessPoolExecutor进程池并发执行同一个robot framework关键字很多次。
源代码如下:
import concurrent.futuresfrom robot.libraries.BuiltIn import BuiltIn
from robot.output.logger import LOGGERdef _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 resultsdef _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)