JMeter与Python的多重交响：从入门到高级应用（下）-51CTO.COM

接上文《JMeter与Python的多重交响：从入门到高级应用(上)》

在性能测试领域，Apache JMeter已经成为测试专业人士的首选工具，用于模拟用户行为、测量响应时间、评估系统性能。但在某些情境下，为了满足特定需求，我们需要更多的灵活性，比如引入Python来进行特定操作或处理复杂逻辑。

一、OS Process Sampler

在OS Process Sampler中，可以直接执行系统命令，这也包括执行Python脚本以及其他乱七八糟的脚本或者文件，但是我们这里只介绍关于调用python脚本的知识。

1.梳理步骤

先编写python脚本，可以接收参数也可以不接收参数，但是一定要使用 print 打印结果出来
如果是windows系统，编写一个.bat 文件，让jmeter直接执行文件，如果linux文件，则编写shell文件
启动jmeter，添加一个os process sample ，然后配置里面的信息
添加一个正则提取器，提取调用外部文件返回的结果就完成我们的所有操作了。

2.演练开始

下面开始步骤一的操作

如下：加密文件中的python代码：

import base64
import sys
from Crypto.Cipher import AES
import binascii

def add_to_16(text):
    while len(text) % 16 != 0:
        text += '\0'
    return text

def encrypt(data, password):
    if isinstance(password, str):
        password = password.encode('utf8')
    bs = AES.block_size
    pad = lambda s: s + (bs - len(s) % bs) * chr(bs - len(s) % bs)
    cipher = AES.new(password, AES.MODE_ECB)
    data = cipher.encrypt(pad(data).encode('utf8'))
    encrypt_data = binascii.b2a_hex(data)  # 输出hex
    # encrypt_data = base64.b64encode(data)         # 取消注释，输出Base64格式
    return encrypt_data.decode('utf8')

if __name__ == '__main__':
    data = sys.argv[1]  # 待加密数据
    # data = '1915'  # 待加密数据
    password = '5544223414143242332423423423423'  # 16,24,32位长的密码（密钥）
    password = add_to_16(password)
    encrypt_data = encrypt(data, password)
    # print('加密前数据：{}\n======================='.format(data))
    print(f"sign={encrypt_data}")

    # decrypt_data = decrypt(encrypt_data, password)
    # print('解密后的数据：{}'.format(decrypt_data))

上述代码的大概逻辑就是接收传进来的待加密字符串，然后进行AES加密，最后打印加密后的数据结果。

接着我们开始步骤二的操作用

windows举例，.bat 文件内容如下：

c:
cd C:\Users\chenyongzhi11\Desktop\
python .\do_AES.py %1

上面文件内容就是在命令行执行python文件，由于前面的python文件接收参数，我们这里使用 %1 这个占位来接收jmeter传进去的参数，我们把文件命名为：execute_python_script.bat

接着我们开始步骤三的操作

添加一个OS Process Sampler：

我们看下这个界面该如何配置：

这会调用外部Python脚本，并传入参数input_param。

最后一步操作

添加一个正则表达式提取器，编写正则，看看能不能提取到结果：

最后我们用debug sample檢測最终结果：

这样整个流程完成了，也就可以很方便的调用外部文件做接口自动化了。

二、其他方案

这里再简单介绍两种能够处理python代码的方案：

使用函数助手[jmeter-functions-execute-python-script-1.0.jar]链接：https://pan.baidu.com/s/1JrPW723es9rFbp18mNAvug?pwd=thjp 提取码：thjp这个就直接放入到：\lib\ext 下面就行，然后重启jmeter使用如图：

使用BeanShell Sampler组件这个需要一定的java代码能力，大伙可以自行看着玩，因烦不建议，前面的os process sample 舒服，也就是说，既然都要写beanshell了，直接java代码干就完事了，哈哈！。示例代码，不保证能用：

import java.io.BufferedReader;
import java.io.InputStreamReader;

//1. 命令里的路径改成自己脚本的路径  
String command = "/opt/homebrew/bin/python3 /Users/xxx/Code/python-mysql/gen_id.py";

Runtime rt = Runtime.getRuntime();
Process pr = rt.exec(command);
 
pr.waitFor();
  
BufferedReader b = new BufferedReader(new InputStreamReader(pr.getInputStream()));
String line = "";
StringBuilder response = new StringBuilder();
while ((line = b.readLine()) != null) {
    response.append(line);
}
  
String response_data = response.toString();
  
System.out.println(response_data);
log.info(response_data);
b.close();

// 2. 定义Jmeter中引用的变量名
vars.put("xxx",response_data); //把结果赋值给变量 ,方便后面调用

By the way,很多小伙伴反馈说既然用jmeter了，干嘛还往里整python代码，不是多此一举嘛？这里勇哥谈谈自己的几点愚见：

现有代码复用：有时候可能我们有一些现成的python代码用特定的操作或者业务逻辑，嵌入这些python代码就可以避免重复造轮子了
代码能力：很多测试人员的技术栈是偏python的，在使用jmeter做自动化测试时，利用python的灵活及强大的库工具就很容易入手了。
集成其他工具：可能有一些自己的python工具很好用，但是想集成到一起就可以考虑这样的偏方了

总之jmeter既然可以这样玩，那么给到用户也就多一种使用体验，多一种解决问题的可能性。