jmeter添加自定義擴展函數之圖片base64

原文連接:---------https://www.cnblogs.com/qiaoyeye/p/7218770.html-----------

打開eclipse,新建maven工程，在pom中引用jmeter核心jar包：

<!-- 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>

1，新建一個包com.mytest.functions，包名要包含functions，因為jmeter.properties對這塊有配置，可見該文件的classfinder.functions.contain=.functions

2，在該包下新建一個類並繼承AbstractFunction，重寫該類的所有方法，具體如下：

 

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("圖片路徑");
    }
    static {
        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 {
        // TODO Auto-generated method stub
        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;
    }

    @Override
    public String getReferenceKey() {
        // TODO Auto-generated method stub
        //提供jmeter函數助手顯示的名稱
        return KEY;
    }

    @Override
    public synchronized void setParameters(Collection<CompoundVariable> arg0) throws InvalidVariableException {
        // TODO Auto-generated method stub
        //檢查參數的個數，支持的方法有2個，具體用法參加api：
        /**
         * protected void checkParameterCount(Collection<CompoundVariable> parameters,
                                   int count)
                            throws InvalidVariableException
            Utility method to check parameter counts.
            Parameters:
            parameters - collection of parameters
            count - number of parameters expected
         * */
        //-----------------
        /**
         * 
         * protected void checkParameterCount(Collection<CompoundVariable> parameters,
                                   int min,
                                   int max)
                            throws InvalidVariableException
            Utility method to check parameter counts.
            Parameters:
            parameters - collection of parameters
            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;
    }

}

 

3，由於我寫的類沒有依賴第三方jar包，引入的jmeter核心包都是jmeter自帶的，所以直接導出上面的類為一個jar包，

　  並把這個jar放在jmeter安裝目錄的apache-jmeter-3.2\lib\ext下面

4，
重啟jmeter，打開函數助手，可看見如下圖：

5，下面我們測試一下這個函數是否能使用，新建一個用戶定義的變量分別添加${__MyBase64(C:\\Users\\test\\Desktop\\20190416173622.jpg,imgresult)}<實際應用中使用第一個就可以>和${imgresult}如下圖，注意${__MyBase64(C:\\Users\\test\\Desktop\\20190416173622.jpg,imgresult)}一定要在上面

6，運行后可以看到已經成功

 

 

 7，接口報錯分析：

　　發送接口時，提示500，使用網絡生成的base64與之對比，發現得到的字符串有空格問題，<java制動換行引起>，因此要使用jmeter中的BeanShell Sampler對得到的結果進行處理，刪除空格。

String Base64_get = vars.get("Base64_jpg");  // 獲取用戶定義的變量的值
String Base64_get = Base64_get.replaceAll("\\s","");  // /s去除空格
log.info("name:"+Base64_get);  //log打印
vars.put("Base64_get",Base64_get.toString()); //輸出Base64_get，用來調用

 

 如果自己無法生成jar，請留言留下聯系方式，

 

 同時感謝：作者喬葉葉的代碼分享，博主地址：http://www.cnblogs.com/qiaoyeye/