io.metersphere.jmeter.LocalRunner Maven / Gradle / Ivy

package io.metersphere.jmeter;

import io.metersphere.cache.JMeterEngineCache;
import io.metersphere.utils.LoggerUtil;
import org.apache.commons.collections.CollectionUtils;
import org.apache.commons.lang3.StringUtils;
import org.apache.jmeter.engine.JMeterEngineException;
import org.apache.jmeter.engine.StandardJMeterEngine;
import org.apache.jorphan.collections.HashTree;

import java.util.List;

public class LocalRunner {
    private HashTree jmxTree;

    public LocalRunner(HashTree jmxTree) {
        this.jmxTree = jmxTree;
    }

    public LocalRunner() {
    }

    public void run(String report) {
        StandardJMeterEngine engine = new StandardJMeterEngine();
        engine.configure(jmxTree);
        try {
            LoggerUtil.info("LocalRunner 开始执行报告",report);
            engine.runTest();
            JMeterEngineCache.runningEngine.put(report, engine);
        } catch (JMeterEngineException e) {
            engine.stopTest(true);
        }
    }

    public void stop(List reports) {
        if (CollectionUtils.isNotEmpty(reports)) {
            for (String report : reports) {
                StandardJMeterEngine engine = JMeterEngineCache.runningEngine.get(report);
                if (engine != null) {
                    engine.stopTest();
                    JMeterEngineCache.runningEngine.remove(report);
                }
            }
        }
    }

    public void stop(String report) {
        if (StringUtils.isNotEmpty(report)) {
            StandardJMeterEngine engine = JMeterEngineCache.runningEngine.get(report);
            if (engine != null) {
                engine.stopTest();
                JMeterEngineCache.runningEngine.remove(report);
            }
        }
    }
}