jmeter自定义函数学习笔记

源码学习

https://github.com/apache/jmeter/blob/trunk/src/functions/org/apache/jmeter/functions/CSVRead.java

Java代码

import org.apache.commons.io.FileUtils;
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 java.io.File;
import java.util.Collection;
import java.util.LinkedList;
import java.util.List;

/**
 * @author lw
 * @createTime 2018/12/1 20:42
 * @description jmeter插件开发
 */
public class StringToFile extends AbstractFunction {

    private static final List<String> desc = new LinkedList<String>();
    private static final String KEY = "__StringToFile";
    private Object[] values;

    static {
        desc.add("Content to save");
        desc.add("Target file");
        desc.add("Is append (default:false)");
    }

    @Override
    public String execute(SampleResult previousResult, Sampler currentSampler)  {
        String content = ((CompoundVariable) values[0]).execute().trim();
        String file = ((CompoundVariable) values[1]).execute().trim();
        try {
            boolean append = false;
            if (values.length > 2) {
                String appendStr = ((CompoundVariable) values[2]).execute().trim();
                if ("1".equals(appendStr) || "true".equals(appendStr)) {
                    append = true;
                }
            }
            FileUtils.write(new File(file), content, append);
        } catch (Exception e) {
            e.printStackTrace();
        }
        return "";
    }

    @Override
    public void setParameters(Collection<CompoundVariable> parameters) throws InvalidVariableException {
        checkMinParameterCount(parameters, 2);
        values = parameters.toArray();
    }

    @Override
    public String getReferenceKey() {
        return KEY;
    }

    @Override
    public List<String> getArgumentDesc() {
        return desc;
    }
}

IDEAmaven文件

<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0"
         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
    <modelVersion>4.0.0</modelVersion>

    <groupId>jmeter-plugin</groupId>
    <artifactId>jmeter</artifactId>
    <version>1.0-SNAPSHOT</version>
    <properties>
        <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
        <project.reporting.outputEncoding>UTF-8</project.reporting.outputEncoding>
        <java.version>1.8</java.version>
    </properties>

    <dependencies>
        <dependency>
            <groupId>org.apache.jmeter</groupId>
            <artifactId>ApacheJMeter_core</artifactId>
            <version>2.13</version>
            <exclusions>
                <exclusion>
                    <groupId>*</groupId>
                    <artifactId>*</artifactId>
                </exclusion>
            </exclusions>
        </dependency>
        <dependency>
            <groupId>org.apache.jmeter</groupId>
            <artifactId>ApacheJMeter_java</artifactId>
            <version>2.13</version>
            <exclusions>
                <exclusion>
                    <groupId>*</groupId>
                    <artifactId>*</artifactId>
                </exclusion>
            </exclusions>
        </dependency>
        <!-- https://mvnrepository.com/artifact/commons-io/commons-io -->
        <dependency>
            <groupId>commons-io</groupId>
            <artifactId>commons-io</artifactId>
            <version>2.6</version>
        </dependency>
    </dependencies>
    <build>
        <plugins>
            <plugin>
                <artifactId>maven-compiler-plugin</artifactId>
                <version>3.1</version>
                <configuration>
                    <source>1.8</source>
                    <target>1.8</target>
                    <encoding>UTF-8</encoding>
                </configuration>
            </plugin>
        </plugins>
    </build>
</project>

结果：

使用方式：
Debug Sampler ${__StringToFile(${res},C:\Users\lw\Desktop\eeee\login.txt,1)}

文件内容：

---

{status:0,msg:"login success!"}