com.blazemeter.jmeter.controller.JMeterThreadParallel Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jmeter-parallel Show documentation
Parallel Sampling for JMeter
There is a newer version: 0.12
package com.blazemeter.jmeter.controller;

import org.apache.jmeter.threads.*;
import org.apache.jorphan.collections.HashTree;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.lang.reflect.Field;
import java.util.HashMap;

public class JMeterThreadParallel extends JMeterThread {
    private static final Logger log = LoggerFactory.getLogger(ParallelSampler.class);
    private TestCompilerParallel parallelCompiler;
    private boolean generateParent;
    private boolean isStopped = false;
    private final JMeterThread parentThread;

    public JMeterThreadParallel(HashTree test, JMeterThreadMonitor monitor, ListenerNotifier notifier, boolean generateParent) {
        super(test, monitor, notifier);
        this.generateParent = generateParent;
        parentThread = JMeterContextService.getContext().getThread();
        if (parentThread == null) {
            throw new NullPointerException();
        }
        try {
            copyCompilerFromParent();
        } catch (IllegalAccessException | NoSuchFieldException e) {
            throw new RuntimeException(e);
        }
    }

    protected void copyCompilerFromParent() throws IllegalAccessException, NoSuchFieldException {
        Field field = JMeterThread.class.getDeclaredField("compiler");
        field.setAccessible(true);
        TestCompiler parentCompiler = (TestCompiler) field.get(parentThread);
        parallelCompiler = cloneTestCompiler(parentCompiler);
        field.set(this, parallelCompiler);
    }

    private TestCompilerParallel cloneTestCompiler(TestCompiler parent) throws NoSuchFieldException, IllegalAccessException {
        TestCompilerParallel cloned = new TestCompilerParallel(new HashTree(), generateParent);

        Field samplerConfigMap = TestCompiler.class.getDeclaredField("samplerConfigMap");
        samplerConfigMap.setAccessible(true);
        samplerConfigMap.set(cloned, ((HashMap) (samplerConfigMap.get(parent))).clone());

        Field transactionControllerConfigMap = TestCompiler.class.getDeclaredField("transactionControllerConfigMap");
        transactionControllerConfigMap.setAccessible(true);
        transactionControllerConfigMap.set(cloned, ((HashMap) (transactionControllerConfigMap.get(parent))).clone());

        return cloned;
    }

    public TestCompilerParallel getParallelCompiler() {
        return parallelCompiler;
    }

    @Override
    public void run() {
        JMeterContextServiceAccessorParallel.decrNumberOfThreads();
        super.run();
        if (isStopped) {
            log.info("Stopping current thread");
            parentThread.stop();
        }
    }

    @Override
    public void stop() {
        isStopped = true;
        log.debug("Parallel Thread was stopped. Parent thread will be stopped after parallel sampler.");
        super.stop();
    }

    public void softStop() {
        log.debug("Parallel Thread was stopped. Parent thread will NOT be stopped after parallel sampler.");
        super.stop();
    }
}