1. 项目概述当性能测试遇上RSA加密最近在做一个金融项目的性能压测接口安全要求高大量请求参数都要求用RSA非对称加密。直接用Jmeter自带的函数助手太天真了那点功能根本不够看处理不了动态密钥、长文本分段加密这些复杂场景。网上的教程要么是零散的脚本片段要么是讲原理不落地踩坑踩到怀疑人生。折腾了好一阵子总算把一套稳定、可复用的Jmeter二次开发方案跑通了今天就把从环境搭建、代码编写到集成调试的全过程结合我实际遇到的坑给你掰开揉碎了讲清楚。无论你是刚接触Jmeter二次开发的新手还是被加密接口压测困扰的老鸟这篇都能给你一套“抄作业”的完整方案。简单说这个方案的核心就是通过开发自定义的Jmeter函数或前置处理器将Java实现的RSA加密逻辑无缝嵌入到Jmeter的测试流程中。这样你就能在HTTP请求的“参数”或“消息体数据”里像调用__MD5一样轻松调用__RSAEncrypt来加密你的动态数据了。下面我们就从为什么需要二次开发开始一步步拆解。2. 核心需求与方案选型背后的逻辑2.1 为什么Jmeter原生支持不够用Jmeter内置的__digest函数主要支持MD5、SHA等哈希算法对于RSA这种非对称加密算法它是无能为力的。虽然可以通过__BeanShell或__groovy脚本执行Java代码但在高并发压测场景下这存在几个致命问题性能瓶颈BeanShell/Groovy脚本在每次迭代中动态编译执行开销巨大在几百上千的并发下脚本引擎本身就会成为性能瓶颈导致TPS每秒事务数数据严重失真。代码维护困难加密逻辑分散在各个线程组的脚本中一旦加密规则变更比如密钥更换、填充模式调整你需要修改所有脚本极易出错。功能局限难以实现复杂的加密流程比如从文件或前置接口动态获取公钥、对超长参数进行自动分段加密、处理加密后的字节到Base64或Hex字符串的转换等。因此将RSA加密逻辑封装成原生的Jmeter函数或插件是高性能、可维护压测的必然选择。编译后的Java类以Jar包形式引入由JVM直接执行效率与Jmeter自身组件无异。2.2 技术方案选型函数 vs. 插件Jmeter二次开发主要有两种形式自定义函数和自定义插件如Sampler、Config Element等。对于加密这种通用数据处理场景自定义函数是更轻量、更合适的选择。自定义函数开发一个继承org.apache.jmeter.functions.AbstractFunction的类。它可以在任何能输入字符串的地方使用比如HTTP请求的参数值、消息体、断言条件等使用灵活符合“加密”作为数据预处理步骤的定位。自定义插件例如开发一个“RSA加密前置处理器”。它更适合于需要复杂配置界面、或对请求进行一系列结构化处理的场景。对于单纯的加密计算来说有点“杀鸡用牛刀”增加了不必要的复杂度。我们的选择很明确实现一个自定义的Jmeter函数org.apache.jmeter.functions.AbstractFunction。接下来我们进入具体的实现环节。3. 开发环境准备与项目搭建工欲善其事必先利其器。别小看环境搭建这里藏着第一个坑。3.1 JDK与Jmeter版本对齐这是最容易被忽略却最容易导致编译失败或运行异常的点。你必须使用与目标Jmeter运行环境相同或更低版本的JDK进行编译。查看Jmeter的JDK版本在Jmeter的bin目录下运行jmeter -v命令输出信息里会明确写明使用的Java版本。比如Jmeter 5.6.2 可能运行在 Java 8 或 11 上。开发环境JDK你的IDE如IntelliJ IDEA或Eclipse和Maven/Gradle配置的JDK版本必须不高于这个版本。强烈建议直接使用相同版本。比如Jmeter用Java 8你就用Java 8开发。Jmeter核心Jar包你需要将Jmeter安装目录下lib/ext中的ApacheJMeter_core.jar和ApacheJMeter_functions.jar添加到项目的构建路径Classpath中。这两个包包含了我们继承AbstractFunction所必须的接口和类。实操心得我曾经用JDK 17编译了一个函数包放到运行在JDK 8的Jmeter里直接报UnsupportedClassVersionError。所以版本一致是红线。3.2 创建Maven项目及关键依赖我推荐使用Maven管理项目它能很好地处理依赖。在你的pom.xml中关键依赖如下dependencies !-- Jmeter核心依赖scope设为provided因为运行时Jmeter本身会提供 -- dependency groupIdorg.apache.jmeter/groupId artifactIdApacheJMeter_core/artifactId version5.6.2/version !-- 版本与你使用的Jmeter一致 -- scopeprovided/scope /dependency dependency groupIdorg.apache.jmeter/groupId artifactIdApacheJMeter_functions/artifactId version5.6.2/version scopeprovided/scope /dependency !-- 日志依赖Jmeter使用log4j2 -- dependency groupIdorg.apache.logging.log4j/groupId artifactIdlog4j-api/artifactId version2.17.2/version scopeprovided/scope /dependency dependency groupIdorg.apache.logging.log4j/groupId artifactIdlog4j-core/artifactId version2.17.2/version scopeprovided/scope /dependency !-- 单元测试 -- dependency groupIdjunit/groupId artifactIdjunit/artifactId version4.13.2/version scopetest/scope /dependency /dependencies注意scope设置为provided意味着这些包在编译和测试时需要但最终打包时不会包含进去因为Jmeter运行时会自带它们。这样可以避免包冲突也是Jmeter插件开发的通用做法。3.3 项目结构规划一个清晰的项目结构有助于管理代码。建议如下rsa-encryption-function/ ├── src/ │ ├── main/ │ │ ├── java/ │ │ │ └── com/ │ │ │ └── yourcompany/ │ │ │ └── jmeter/ │ │ │ └── functions/ │ │ │ ├── RSAEncryptionFunction.java // 核心函数类 │ │ │ └── util/ │ │ │ └── RSAUtil.java // RSA加密工具类 │ │ └── resources/ │ │ └── META-INF/ │ │ └── services/ │ │ └── org.apache.jmeter.functions.Function // SPI服务发现文件 │ └── test/ │ └── java/ │ └── ... // 单元测试 ├── pom.xml └── README.md重点在于src/main/resources/META-INF/services/目录下的那个文件。这是Java的SPIService Provider Interface机制Jmeter通过它来发现我们自定义的函数。文件名为org.apache.jmeter.functions.Function内容就是我们实现的函数类的全限定名例如com.yourcompany.jmeter.functions.RSAEncryptionFunction4. 核心代码实现从工具类到Jmeter函数4.1 RSA加密工具类 (RSAUtil.java)首先我们封装一个健壮的RSA加密工具类。这里采用最常用的RSA/ECB/PKCS1Padding模式并处理好Base64编码。package com.yourcompany.jmeter.functions.util; import javax.crypto.Cipher; import java.nio.charset.StandardCharsets; import java.security.KeyFactory; import java.security.PublicKey; import java.security.spec.X509EncodedKeySpec; import java.util.Base64; public class RSAUtil { private static final String TRANSFORMATION RSA/ECB/PKCS1Padding; private static final String ALGORITHM RSA; /** * 使用Base64编码的公钥字符串加密文本 * * param plainText 待加密明文 * param publicKeyBase64 Base64编码的PEM公钥字符串去除头尾标记 * return Base64编码的密文 * throws Exception 加密过程中的任何异常 */ public static String encryptWithBase64Key(String plainText, String publicKeyBase64) throws Exception { // 1. 解码Base64公钥字符串得到字节数组 byte[] keyBytes Base64.getDecoder().decode(publicKeyBase64); X509EncodedKeySpec keySpec new X509EncodedKeySpec(keyBytes); // 2. 生成公钥对象 KeyFactory keyFactory KeyFactory.getInstance(ALGORITHM); PublicKey publicKey keyFactory.generatePublic(keySpec); // 3. 初始化加密器并执行加密 Cipher cipher Cipher.getInstance(TRANSFORMATION); cipher.init(Cipher.ENCRYPT_MODE, publicKey); byte[] encryptedBytes cipher.doFinal(plainText.getBytes(StandardCharsets.UTF_8)); // 4. 将加密后的字节数组转换为Base64字符串返回 return Base64.getEncoder().encodeToString(encryptedBytes); } /** * 处理超长文本的加密RSA有长度限制需要分段 * 注意PKCS1Padding模式下RSA加密的明文长度 密钥长度/8 - 11 * 例如2048位密钥最大明文长度是 256 - 11 245字节 */ public static String encryptLongText(String longText, String publicKeyBase64, int keySize) throws Exception { int maxBlockSize keySize / 8 - 11; // 计算最大分段大小 byte[] data longText.getBytes(StandardCharsets.UTF_8); StringBuilder encryptedResult new StringBuilder(); // 分段加密 for (int offset 0; offset data.length; offset maxBlockSize) { int inputLen Math.min(data.length - offset, maxBlockSize); byte[] block new byte[inputLen]; System.arraycopy(data, offset, block, 0, inputLen); // 调用单次加密方法需要稍作修改以支持字节数组输入 String encryptedBlock encryptBytesWithBase64Key(block, publicKeyBase64); encryptedResult.append(encryptedBlock); // 可选在分段之间添加分隔符但通常不需要因为解密时知道固定长度 } return encryptedResult.toString(); } // 需要一个支持字节数组输入的encrypt方法内部方法供分段加密调用 private static String encryptBytesWithBase64Key(byte[] data, String publicKeyBase64) throws Exception { byte[] keyBytes Base64.getDecoder().decode(publicKeyBase64); X509EncodedKeySpec keySpec new X509EncodedKeySpec(keyBytes); KeyFactory keyFactory KeyFactory.getInstance(ALGORITHM); PublicKey publicKey keyFactory.generatePublic(keySpec); Cipher cipher Cipher.getInstance(TRANSFORMATION); cipher.init(Cipher.ENCRYPT_MODE, publicKey); byte[] encryptedBytes cipher.doFinal(data); return Base64.getEncoder().encodeToString(encryptedBytes); } }注意事项密钥格式这个工具类假设你提供的公钥是Base64编码的X.509格式即PEM文件去掉-----BEGIN PUBLIC KEY-----和-----END PUBLIC KEY-----头尾后的内容。如果你的公钥是其他格式如PKCS#8需要相应的KeySpec。分段加密encryptLongText方法提供了基础的分段逻辑。但在实际HTTP传输中服务端如何解析拼接后的分段密文是个问题。更常见的做法是在接口层面避免传输超长的RSA加密数据通常RSA只用于加密密钥如AES密钥或短摘要。如果必须加密长文本需要与后端约定好分段和拼接规则。异常处理工具类抛出了Exception在实际函数中我们需要捕获并妥善处理避免导致Jmeter线程崩溃。4.2 Jmeter自定义函数类 (RSAEncryptionFunction.java)这是核心让我们的加密逻辑成为Jmeter的一部分。package com.yourcompany.jmeter.functions; import com.yourcompany.jmeter.functions.util.RSAUtil; import org.apache.jmeter.engine.util.CompoundVariable; import org.apache.jmeter.functions.AbstractFunction; import org.apache.jmeter.functions.InvalidVariableException; import org.apache.jmeter.samplers.SampleResult; import org.apache.jmeter.samplers.Sampler; import org.apache.jmeter.threads.JMeterVariables; import org.apache.jorphan.util.JMeterException; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import java.util.*; public class RSAEncryptionFunction extends AbstractFunction { private static final Logger log LoggerFactory.getLogger(RSAEncryptionFunction.class); // 函数名称在Jmeter中调用的名字 private static final String KEY __RSAEncrypt; // 函数描述会显示在Jmeter的函数助手中 private static final ListString DESC Arrays.asList( Encrypt a string using RSA with a base64 encoded public key., String to encrypt, Base64 encoded public key (without BEGIN/END markers), Name of variable in which to store the result (optional) ); // 存储传入的参数值 private CompoundVariable[] values; // 存储计算结果的变量名可选 private CompoundVariable resultVariableName; Override public String execute(SampleResult previousResult, Sampler currentSampler) throws InvalidVariableException { // 1. 参数校验 if (values.length 2) { log.error({} function requires at least 2 parameters: plainText and publicKeyBase64, KEY); return ; } String plainText values[0].execute().trim(); String publicKeyBase64 values[1].execute().trim(); if (plainText.isEmpty() || publicKeyBase64.isEmpty()) { log.warn(Plain text or public key is empty for function {}, KEY); return ; } // 2. 执行加密 String encryptedText; try { // 这里调用我们工具类的方法假设文本不长使用非分段加密 encryptedText RSAUtil.encryptWithBase64Key(plainText, publicKeyBase64); } catch (Exception e) { log.error(RSA encryption failed for text: {}. Error: {}, plainText, e.getMessage(), e); // 在实际压测中你可能希望失败时返回一个特定标记或抛出异常这里返回空字符串 return ; } // 3. 如果指定了存储结果的变量名则将其存入Jmeter变量中 if (resultVariableName ! null) { String varName resultVariableName.execute().trim(); if (!varName.isEmpty()) { JMeterVariables vars getVariables(); if (vars ! null) { vars.put(varName, encryptedText); } } } // 4. 返回加密后的字符串供请求直接使用 return encryptedText; } Override public void setParameters(CollectionCompoundVariable parameters) throws InvalidVariableException { // 检查参数数量 checkParameterCount(parameters, 2, 3); Object[] params parameters.toArray(); values new CompoundVariable[params.length]; for (int i 0; i params.length; i) { values[i] (CompoundVariable) params[i]; } // 如果传入了第三个参数则是存储结果的变量名 if (params.length 3) { resultVariableName (CompoundVariable) params[2]; } else { resultVariableName null; } } Override public String getReferenceKey() { return KEY; } Override public ListString getArgumentDesc() { return DESC; } }代码关键点解析继承与关键方法必须继承AbstractFunction并实现execute,setParameters,getReferenceKey,getArgumentDesc四个核心方法。参数处理 (setParameters)Jmeter会将用户输入的参数如${mobile}, ${pubKey}作为CompoundVariable集合传入。我们需要在这里解析并存储它们。checkParameterCount是一个辅助方法确保参数数量在2到3个之间明文、公钥、可选变量名。执行加密 (execute)这是函数的核心。它从values数组中取出明文和公钥调用我们的RSAUtil进行加密。这里进行了简单的错误处理将异常捕获并记录日志避免因单个加密失败导致整个线程停止。变量存储如果用户提供了第三个参数变量名我们会将加密结果存入Jmeter的变量上下文 (JMeterVariables) 中这样后续的请求或断言可以通过${encryptedResult}来引用它。返回值execute方法的返回值就是函数执行的结果。在HTTP请求的参数值中填写${__RSAEncrypt(${mobile},${pubKey})}Jmeter就会用加密后的字符串替换这个函数调用。5. 打包、部署与在Jmeter中使用5.1 项目打包与SPI文件确保META-INF/services/org.apache.jmeter.functions.Function文件已正确创建并包含你的类全名。然后使用Maven打包mvn clean package这会在target目录下生成一个rsa-encryption-function-1.0-SNAPSHOT.jar假设你的artifactId是rsa-encryption-function。关键一步你需要检查生成的Jar包中是否包含了META-INF/services目录及其文件。可以用解压软件打开Jar包查看。如果没有需要配置Maven的maven-resources-plugin确保资源文件被正确复制。5.2 部署到Jmeter将上一步打好的Jar包复制到你的Jmeter安装目录的lib/ext文件夹下。这是Jmeter加载自定义插件和函数的默认位置。然后重启Jmeter。这是必须的Jmeter只会在启动时加载lib/ext目录下的新Jar包。5.3 在Jmeter中调用自定义函数重启后打开Jmeter添加一个线程组和一个HTTP请求。准备变量你可以使用“用户定义的变量”或通过“正则表达式提取器”从上一个请求中获取公钥和待加密的明文。假设我们有两个变量${mobile}13800138000和${publicKeyBase64}MIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIBCgKCAQEA...(你的公钥Base64字符串)。在HTTP请求中调用函数在HTTP请求的“参数”或“消息体数据”选项卡中在需要加密的值一栏直接输入函数调用。用法1直接使用返回值在参数值中填写${__RSAEncrypt(${mobile},${publicKeyBase64})}。发送请求时Jmeter会自动将其替换为加密后的密文。用法2存储到变量再使用填写${__RSAEncrypt(${mobile},${publicKeyBase64},encryptedMobile)}。这会将加密结果存储到名为encryptedMobile的变量中。然后在参数值中填写${encryptedMobile}。验证函数是否生效添加一个“调试取样器”Debug Sampler和“查看结果树”View Results Tree。运行测试计划查看调试取样器的响应数据你应该能看到encryptedMobile变量如果用了用法2及其对应的Base64密文值。更进一步的验证是用同样的公钥对应的私钥去解密这个密文看是否能得到原始明文13800138000。6. 高级功能扩展与性能优化基础功能跑通后我们可以让它更强大、更健壮。6.1 支持从文件读取公钥在实际项目中公钥可能放在一个.pem文件里。我们可以在函数中增加对文件路径的支持。修改RSAEncryptionFunction.execute方法中的公钥获取逻辑String publicKeyBase64 values[1].execute().trim(); // 判断第二个参数是直接的Base64字符串还是文件路径 String keyContent; if (publicKeyBase64.startsWith(file://)) { String filePath publicKeyBase64.substring(7); // 去掉 file:// 前缀 try { keyContent new String(Files.readAllBytes(Paths.get(filePath)), StandardCharsets.UTF_8); // 清理PEM格式的头尾标记和换行符 keyContent keyContent.replace(-----BEGIN PUBLIC KEY-----, ) .replace(-----END PUBLIC KEY-----, ) .replaceAll(\\s, ); // 移除所有空白字符 } catch (IOException e) { log.error(Failed to read public key file: {}, filePath, e); return ; } } else { keyContent publicKeyBase64; // 直接使用Base64字符串 } // 然后使用 keyContent 调用 RSAUtil.encryptWithBase64Key这样在Jmeter中就可以这样调用${__RSAEncrypt(${mobile},file:///path/to/public_key.pem)}。6.2 增加算法和填充模式的可配置性RSA算法可能有不同的填充模式如PKCS1Padding、OAEPWithSHA-256AndMGF1Padding等。我们可以增加一个可选参数来指定。首先修改工具类RSAUtil让encryptWithBase64Key方法接受一个transformation参数。 然后在RSAEncryptionFunction中修改DESC和setParameters以支持额外的参数并在execute方法中将其传递给工具类。例如函数调用可以变为${__RSAEncrypt(${mobile},${pubKey},RSA/ECB/OAEPWithSHA-256AndMGF1Padding)}。6.3 性能优化缓存公钥对象在高并发压测中每次加密都去解析Base64字符串并生成PublicKey对象是一个不小的开销。我们可以引入一个简单的缓存。在RSAUtil类中public class RSAUtil { private static final MapString, PublicKey PUBLIC_KEY_CACHE new ConcurrentHashMap(); public static String encryptWithBase64Key(String plainText, String publicKeyBase64) throws Exception { PublicKey publicKey PUBLIC_KEY_CACHE.get(publicKeyBase64); if (publicKey null) { // 缓存未命中创建并缓存 byte[] keyBytes Base64.getDecoder().decode(publicKeyBase64); X509EncodedKeySpec keySpec new X509EncodedKeySpec(keyBytes); KeyFactory keyFactory KeyFactory.getInstance(ALGORITHM); publicKey keyFactory.generatePublic(keySpec); PUBLIC_KEY_CACHE.putIfAbsent(publicKeyBase64, publicKey); // 使用putIfAbsent避免并发重复创建 } // ... 使用缓存的publicKey进行加密 ... } }使用ConcurrentHashMap保证线程安全。这样同一个公钥在整个压测过程中只会被解析一次显著提升性能。7. 调试技巧与常见问题排查二次开发过程中遇到问题很正常。这里分享几个我踩过的坑和解决方法。7.1 函数不生效或找不到症状在函数助手中找不到__RSAEncrypt或者在日志中看到Function __RSAEncrypt not found的错误。排查步骤检查Jar包位置确认Jar包是否放到了jmeter/lib/ext目录下并且重启了Jmeter。检查SPI文件用解压软件打开你的Jar包确认META-INF/services/org.apache.jmeter.functions.Function文件存在且内容是你的函数类全名如com.yourcompany.jmeter.functions.RSAEncryptionFunction注意不能有空格或换行。检查类路径冲突确保你的Jar包没有包含与Jmeter自带库如ApacheJMeter_core.jar重复的类。这就是为什么我们使用providedscope。查看Jmeter启动日志在Jmeter的bin目录下启动时加上-j logfile.txt参数如jmeter -j startup.log查看启动日志中是否有加载你的Jar包或相关错误信息。7.2 加密结果与服务端不一致症状Jmeter加密的数据后端解密失败或得到乱码。排查步骤确认公钥一致性百分之九十的问题出在公钥格式上。确保你提供给Jmeter函数的公钥字符串与后端使用的公钥是完全相同的Base64编码。一个常见的错误是包含了PEM文件的头尾标记或换行符。使用上文提到的清理方法replaceAll(\\s, )确保公钥是纯净的Base64字符串。确认算法和填充模式用Java代码加密和用其他语言如Python、Go加密默认的填充模式可能不同。必须明确指定并确保前后端一致例如都是RSA/ECB/PKCS1Padding。编码问题确保加密前的明文字符串编码一致通常使用UTF-8。在Java中使用getBytes(StandardCharsets.UTF_8)。分段问题如果加密长文本确认前后端的分段、拼接逻辑完全一致。强烈建议先用一个短的、固定的字符串如test123进行联调排除基础加解密问题。7.3 高并发下的性能问题或内存溢出症状压测时TPS上不去或者运行一段时间后Jmeter报OutOfMemoryError。排查与优化启用公钥缓存如上文“性能优化”部分所述这是必须做的。监控内存使用Jmeter的jmeter.bat或jmeter.sh脚本启动时可以调整JVM堆内存参数-Xms和-Xmx。对于大规模压测适当增加内存是必要的。避免在函数中创建大对象确保execute方法执行高效不要在每次调用时都创建大量的临时对象。我们的工具类方法已经是静态的且公钥被缓存这点做得不错。日志级别将自定义函数类的日志级别调高如设为WARN或ERROR避免在压测时产生大量INFO或DEBUG日志拖慢性能。可以在log4j2.xml中配置。7.4 如何调试自定义函数代码本地单元测试在打包前为RSAUtil和RSAEncryptionFunction编写JUnit测试用例模拟各种输入正常、空值、错误密钥等确保核心逻辑正确。远程调试这是最强大的手段。在IDE中以调试模式启动你的自定义函数项目并设置远程调试端口。然后在启动Jmeter时添加JVM参数jmeter -Jjava.rmi.server.hostnamelocalhost -Jserver_port1099 -Jserver.rmi.localport1099 -Jclient.rmi.localport1099 -s -j jmeter-server.log同时在jmeter.bat或jmeter.sh中在HEAP变量设置之后添加set JVM_ARGS%JVM_ARGS% -agentlib:jdwptransportdt_socket,servery,suspendn,address5005然后在IDE中配置远程调试连接到localhost:5005就可以在Jmeter执行时断点调试你的函数代码了。8. 实战构建一个完整的加密接口压测场景理论说再多不如一个实际例子。假设我们要压测一个“用户登录”接口请求参数username和password都需要RSA加密。测试计划结构线程组设置并发用户数、循环次数等。用户定义的变量定义publicKeyBase64你的公钥。CSV 数据文件设置读取一个CSV文件里面有多组username和password明文。HTTP请求登录接口。方法POST路径/api/login参数username:${__RSAEncrypt(${username},${publicKeyBase64})}password:${__RSAEncrypt(${password},${publicKeyBase64})}响应断言检查返回的JSON中是否包含success: true。查看结果树 聚合报告用于调试和查看性能指标。参数化与关联如果公钥也是动态从某个接口获取的你可以先添加一个“获取公钥”的HTTP请求用“正则表达式提取器”或“JSON提取器”将公钥值提取到变量如pubKeyFromServer中然后在登录请求的函数调用里使用这个变量。执行与监控运行测试在“查看结果树”中检查每个请求的“请求”选项卡确认username和password的值是否已被正确替换为长长的Base64密文。观察“聚合报告”中的吞吐量Throughput、平均响应时间、错误率等关键指标。通过这样一套流程你就将一个需要复杂加密的接口性能测试变得像测试普通接口一样简单和自动化。这套自定义函数的方法不仅适用于RSA完全可以举一反三用于实现国密SM2、SM4或者复杂的签名算法彻底解放Jmeter在安全测试方面的生产力。