DistributedTestControl.java

package kg.apc.jmeter.config;

import org.apache.commons.lang3.StringUtils;
import org.apache.jmeter.config.ConfigTestElement;
import org.apache.jmeter.testelement.property.CollectionProperty;
import org.apache.jmeter.util.JMeterUtils;
import org.apache.jorphan.logging.LoggingManager;
import org.apache.log.Logger;

import java.util.ArrayList;
import java.util.LinkedList;

public class DistributedTestControl extends ConfigTestElement {

    public static final String DATA_PROP = "SERVERS";
    public static final String PROP_HOSTS = "remote_hosts";
    public static Logger log = LoggingManager.getLoggerForClass();

    public CollectionProperty getData() {
        CollectionProperty data = (CollectionProperty) getProperty(DATA_PROP);
        LinkedList<String> arr=new LinkedList<String>();

        for (int n = 0; n < data.size(); n++) {
            arr.add(data.get(n).getStringValue());
        }

        String val = StringUtils.join(arr, ",");
        log.debug("Setting hosts 1: " + val);
        JMeterUtils.setProperty(PROP_HOSTS, val);
        return data;
    }

    public void setData(ArrayList<String> data) {
        setProperty(new CollectionProperty(DistributedTestControl.DATA_PROP, data));
        String val = StringUtils.join(data, ",");
        log.debug("Setting hosts 2: " + val);
        JMeterUtils.setProperty(PROP_HOSTS, val);
    }
}