CSDN

JMeter 基于脚本实现代码共享

2023-09-25 00:38 由 授客 发表于 #其他

需求描述

需求是这样的：执行某次压测任务时，压测涉及的前端接口，要求携带一个userName请求头，该请求头值为实际用户名经过DES加密后，再采用Base64加密后的值，为此，编写了一段加密代码，发送请求前，对用户名进行加密，并将加密结果存储为变量。最后将代码添加到 JSR233采样器中，如下

DES加密代码

import java.util.Base64;
import javax.crypto.Cipher;
import java.security.Key;
import javax.crypto.SecretKeyFactory;
import javax.crypto.spec.DESKeySpec;


def encryptUsername(userName) {	
	String keyStr = 'jPQQqFT3lwg=' //  
	
	DESKeySpec dks = new DESKeySpec(keyStr.getBytes('UTF-8'));  
	SecretKeyFactory keyFactory = SecretKeyFactory.getInstance('DES');  
	Key key = keyFactory.generateSecret(dks); 
	        
	Cipher cipher = Cipher.getInstance('DES');
	cipher.init(Cipher.ENCRYPT_MODE, key); 
	
	byte[] encodedBytes = cipher.doFinal(userName.getBytes()) 

	def encodedString = Base64.getEncoder().encodeToString(encodedBytes);
	return encodedString
}

vars.put("userName", encryptUsername(vars.get('tsUserName')))
log.info('userName in script：' +vars.get("userName")) # 非调试时注释掉

此外，还有一些接口，在发送请求前，需要先请求通用配置类接口（比如获取组织结构树），以获取接口入参所需数据，也是采用类似如上的方法，即通过添加（前置）JSR233采样器。

按以上做法，即将代码编写在JSR233采样器中本身是没有问题的，问题在于相同的代码，被放在了n个采样器中（为了适配需求，比如不同页面的接口要求放在不同线程组中，配置不同的并发用户数，或者请求好些接口前都需要执行这份代码），这样当这份代码因为存在缺陷需要修改时，将需要修改n个地方，非常的繁琐。

解决方法

咋样避免上述的这种情况呢？我们可以将脚本文件化--将脚本放在一个文件中，而不是放在界面Script输入框中，如下

这里需要注意两点：

  1. File Name 输入的脚本文件所在路径，可以是相对路径(相对于JMeter bin目录)，也可以是绝对路径(建议将脚本所在目录参数化，这样，一改全改，避免在不同环境执行jmx脚本时，因为脚本路径不一样需要多处修改脚本路径)
  2. 如果File Name 输入脚本路径，下面Script输入框中输入的代码将不再被执行（上述示例为验证此观点，特别在输入框中添加了日志打印代码，发现该日志打印代码未被执行）