jmeter添加自定义扩展函数之图片base64编码示例详解

<!-- https://mvnrepository.com/artifact/org.apache.jmeter/ApacheJMeter_core -->
<dependency>
    <groupId>org.apache.jmeter</groupId>
    <artifactId>ApacheJMeter_core</artifactId>
    <version>3.2</version>
</dependency>

<!-- Https://mvnrepository.com/artifact/org.apache.jmeter/ApacheJMeter_functions -->
<dependency>
    <groupId>org.apache.jmeter</groupId>
    <artifactId>ApacheJMeter_functions</artifactId>
    <version>3.2</version>
</dependency>

package com.mytest.functions;

import java.io.FileInputStream;
import java.io.InputStream;
import java.util.Collection;
import java.util.LinkedList;
import java.util.List;
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 sun.misc.BASE64Encoder;
public class MyBase64 extends AbstractFunction{
    //自定义function的描述
    private static final List<String> desc = new LinkedList<String>();
    static {
        desc.add("图片路径");
    }
        desc.add("图片base64后存放变量");
    private static final String KEY = "__MyBase64";
   
    //存放传入参数的值的变量
    private Object[] values;
    
    //描述参数
    public List<String> getArgumentDesc() {
        // TODO Auto-generated method stub
        return desc;
    @Override
    //函数的执行
    public synchronized String execute(SampleResult arg0, Sampler arg1) throws InvalidVariableException {
        JMeterVariables localJMeterVariables = getVariables();
        String str1 = ((CompoundVariable)this.values[0]).execute();
        String str2 = getImgBase64(str1);
        if ((localJMeterVariables != null) && (this.values.length > 1)) {
          String str3 = ((CompoundVariable)this.values[1]).execute().trim();
          localJMeterVariables.put(str3, str2);
        }
        return str2;
    public String getReferenceKey() {
        //提供jmeter函数助手显示的名称
        return KEY;
    public synchronized void setParameters(Collection<CompoundVariable> arg0) throws InvalidVariableException {
        //检查参数的个数，支持的方法有2个，具体用法参加api：
        
        //-----------------
         * 
                                   int min,
                                   int max)
            min - minimum number of parameters allowed
            max - maximum number of parameters allowed
        //checkParameterCount(arg0, 1);
        checkParameterCount(arg0, 1, 2);
        //将参数值存入变量中
        this.values = arg0.toArray();
    public String getImgBase64(String filePath) {
        InputStream in = null;
        byte[] data = null;
        String result = null;
        try {
            in = new FileInputStream(filePath);
            data = new byte[in.available()];
            in.read(data);
            in.close();
            BASE64Encoder encoder = new BASE64Encoder();
            result = encoder.encode(data);
        } catch (Exception e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
            
        
        return result;
}