1

2

3

4

5

6

7

8

9

10

11

12

13

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

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

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：

        /**

         * 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

         * */

        //-----------------

         *

                                   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;

}