com.blazemeter.jmeter.controller.ParallelSampler 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
Show all versions of jmeter-parallel Show documentation
Parallel Sampling for JMeter
package com.blazemeter.jmeter.controller;
import com.blazemeter.jmeter.controller.traverse.CustomTreeCloner;
import org.apache.jmeter.control.Controller;
import org.apache.jmeter.control.LoopController;
import org.apache.jmeter.engine.StandardJMeterEngine;
import org.apache.jmeter.engine.event.LoopIterationListener;
import org.apache.jmeter.samplers.*;
import org.apache.jmeter.testelement.AbstractTestElement;
import org.apache.jmeter.testelement.TestElement;
import org.apache.jmeter.testelement.TestStateListener;
import org.apache.jmeter.testelement.ThreadListener;
import org.apache.jmeter.threads.*;
import org.apache.jorphan.collections.HashTree;
import org.apache.jorphan.collections.SearchByClass;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.io.Serializable;
import java.lang.reflect.Field;
import java.util.*;
import java.util.concurrent.*;
import java.lang.ThreadGroup;
// we implement Controller only to enable GUI to add child elements into it
public class ParallelSampler extends AbstractSampler implements Controller, ThreadListener, Interruptible, JMeterThreadMonitor, TestStateListener, Serializable {
private static final Logger log = LoggerFactory.getLogger(ParallelSampler.class);
private static final String GENERATE_PARENT = "PARENT_SAMPLE";
private static final String MAX_THREAD_NUMBER = "MAX_THREAD_NUMBER";
private static final String LIMIT_MAX_THREAD_NUMBER = "LIMIT_MAX_THREAD_NUMBER";
protected List controllers = new ArrayList<>();
protected final ParallelListenerNotifier notifier = new ParallelListenerNotifier();
private ExecutorService executorService;
private DummyThreadGroup threadGroup;
@Override
public void addTestElement(TestElement te) {
if (te instanceof Controller || te instanceof Sampler) {
controllers.add(te);
}
log.debug("Added {}, list size: {}", te, controllers.size());
}
@Override
// prevents property reset for wrapper controllers
public void setRunningVersion(boolean runningVersion) {
super.setRunningVersion(runningVersion);
for (TestElement ctl : controllers) {
ctl.setRunningVersion(runningVersion);
}
}
@Override
public SampleResult sample(Entry e) {
SampleResult res = new SampleResult();
res.setResponseCode("200");
res.setResponseMessage("OK");
res.setSuccessful(true);
res.setSampleLabel(getName());
res.setResponseData("".getBytes());
notifier.setContainer(res);
final List jMeterThreads = new LinkedList<>();
threadGroup.reset();
StringBuilder reqText = new StringBuilder("Parallel items:\n");
for (TestElement ctl : controllers) {
reqText.append(ctl.getName()).append("\n");
JMeterThread jmThread = new JMeterThreadParallel(getTestTree(ctl), this, notifier, getGenerateParent());
String name = JMeterContextService.getContext().getThread() + " - " + this.getName() + " - " + ctl.getName();
jmThread.setThreadName(name);
jmThread.setThreadGroup(threadGroup);
jmThread.setEngine(JMeterContextService.getContext().getEngine());
injectVariables(jmThread, this.getThreadContext());
jMeterThreads.add(jmThread);
threadGroup.addThread(jmThread);
}
res.setSamplerData(reqText.toString());
res.sampleStart();
Collection> futures = new LinkedList<>();
for (JMeterThread jmThread : jMeterThreads) {
futures.add(executorService.submit(jmThread));
}
for (Future future : futures) {
try {
future.get();
log.debug("Thread is done {}", future.isDone());
} catch (InterruptedException | ExecutionException e1) {
log.debug("Interrupted {}", future.isCancelled());
}
}
if (res.getEndTime() == 0) {
res.sampleEnd();
}
return getGenerateParent() ? res : null;
}
private HashTree getTestTree(TestElement te) {
// can't use GenericController because of infinite looping
CustomLoopController wrapper = new CustomLoopController(JMeterContextService.getContext());
wrapper.setLoops(1);
wrapper.setContinueForever(false);
wrapper.addTestElement(te);
wrapper.setName("wrapped " + te.getName());
wrapper.setRunningVersion(isRunningVersion());
HashTree tree = new HashTree();
HashTree subTree = getSubTree(te);
if (subTree != null) {
tree.add(wrapper, subTree);
} else {
tree.add(wrapper);
}
return tree;
}
private static class CustomLoopController extends LoopController {
private final JMeterContext context;
private boolean isFinished = false;
/**
* @param context from parent Thread
*/
public CustomLoopController(JMeterContext context) {
this.context = context;
}
@Override
public boolean isDone() {
return isFinished || super.isDone();
}
@Override
public void triggerEndOfLoop() {
isFinished = true;
context.setRestartNextLoop(true);
super.triggerEndOfLoop();
}
}
private HashTree getSubTree(TestElement te) {
try {
Field field = JMeterThread.class.getDeclaredField("testTree");
field.setAccessible(true);
JMeterThread parentThread = JMeterContextService.getContext().getThread();
if (parentThread == null) {
log.error("Current thread is null.");
throw new NullPointerException();
}
HashTree testTree = (HashTree) field.get(parentThread);
SearchByClass searcher = new SearchByClass<>(te.getClass());
testTree.traverse(searcher);
return searcher.getSubTree(te);
} catch (ReflectiveOperationException ex) {
log.warn("Can not get sub tree for Test element " + te.getName(), ex);
return null;
}
}
@Override
public boolean interrupt() {
executorService.shutdown();
return true;
}
@Override
public Sampler next() {
return null;
}
@Override
public boolean isDone() {
return true; // most likely having true here is ruining things
}
@Override
public void initialize() {
log.debug("Initialize");
}
@Override
public void triggerEndOfLoop() {
log.debug("Trigger End of loop");
}
@Override
public void threadFinished(JMeterThread thread) {
JMeterContextServiceAccessorParallel.incrNumberOfThreads();
try {
Field field = AbstractTestElement.class.getDeclaredField("threadContext");
field.setAccessible(true);
if (thread instanceof JMeterThreadParallel) {
JMeterThreadParallel pthr = (JMeterThreadParallel) thread;
for (TestElement te : pthr.getParallelCompiler().getKnownElements()) {
field.set(te, null);
}
}
} catch (IllegalAccessException | NoSuchFieldException e) {
log.warn("Failed to reset context", e);
}
}
@Override
public void addIterationListener(LoopIterationListener listener) {
}
@Override
public void removeIterationListener(LoopIterationListener iterationListener) {
}
public int getMaxThreadNumber() {
return getPropertyAsInt(MAX_THREAD_NUMBER, 6);
}
public void setMaxThreadNumber(int value) {
setProperty(MAX_THREAD_NUMBER, value);
}
public boolean getLimitMaxThreadNumber() {
return getPropertyAsBoolean(LIMIT_MAX_THREAD_NUMBER);
}
public void setLimitMaxThreadNumber(boolean value) {
setProperty(LIMIT_MAX_THREAD_NUMBER, value);
}
public boolean getGenerateParent() {
return getPropertyAsBoolean(GENERATE_PARENT);
}
public void setGenerateParent(boolean value) {
setProperty(GENERATE_PARENT, value);
}
private void injectVariables(JMeterThread jmThread, JMeterContext threadContext) {
if (threadContext != null && threadContext.getVariables() != null) {
try {
Class cls = JMeterThread.class;
Field vars = cls.getDeclaredField("threadVars");
vars.setAccessible(true);
vars.set(jmThread, threadContext.getVariables());
} catch (Throwable ex) {
log.warn("Cannot inject variables into parallel thread ", ex);
}
}
}
private void changeVariablesMap() {
try {
JMeterContext context = this.getThreadContext();
if (context != null && context.getVariables() != null) {
JMeterVariables jMeterVariables = context.getVariables();
Class cls = JMeterVariables.class;
Field variablesField = cls.getDeclaredField("variables");
variablesField.setAccessible(true);
Object obj = variablesField.get(jMeterVariables);
synchronized (obj) {
if (obj instanceof Map) {
Map variables = (Map) obj;
if (!(variables instanceof ConcurrentHashMap)) {
variablesField.set(jMeterVariables, new ConcurrentHashMap(variables));
}
} else {
log.warn("Unexpected variables map type " + obj.getClass().getName());
}
}
}
} catch (Throwable ex) {
log.warn("Cannot change variables map ", ex);
}
}
@Override
public void threadStarted() {
changeVariablesMap();
if (getLimitMaxThreadNumber()) {
log.info("Starting up to {} threads", getMaxThreadNumber());
executorService = Executors.newFixedThreadPool(getMaxThreadNumber(), new ParallelThreadFactory(this.getName()));
} else {
executorService = Executors.newCachedThreadPool(new ParallelThreadFactory(this.getName()));
}
threadGroup = new DummyThreadGroup();
addThreadGroupToEngine(threadGroup);
}
@Override
public void threadFinished() {
executorService.shutdown();
removeThreadGroupFromEngine(threadGroup);
}
private void addThreadGroupToEngine(AbstractThreadGroup group) {
try {
StandardJMeterEngine engine = JMeterContextService.getContext().getEngine();
Field groupsField = StandardJMeterEngine.class.getDeclaredField("groups");
groupsField.setAccessible(true);
List groups = (List) groupsField.get(engine);
groups.add(group);
} catch (ReflectiveOperationException ex) {
log.warn("Can not add DummyThreadGroup to engine", ex);
}
}
private void removeThreadGroupFromEngine(AbstractThreadGroup group) {
try {
StandardJMeterEngine engine = JMeterContextService.getContext().getEngine();
Field groupsField = StandardJMeterEngine.class.getDeclaredField("groups");
groupsField.setAccessible(true);
List groups = (List) groupsField.get(engine);
groups.remove(group);
} catch (ReflectiveOperationException ex) {
log.warn("Can not remove DummyThreadGroup from engine", ex);
}
}
@Override
public void testStarted() {
testStarted("*local*");
}
@Override
public void testStarted(String s) {
changeCookieManager();
}
private void changeCookieManager() {
try {
StandardJMeterEngine engine = getStandardJMeterEngine();
Field field = StandardJMeterEngine.class.getDeclaredField("test");
field.setAccessible(true);
HashTree testTree = (HashTree) field.get(engine);
HashTree newHashTree = makeCookieManagerThreadSafe(testTree);
field.set(engine, newHashTree);
} catch (Throwable ex) {
log.warn("Cannot change cookie manager", ex);
}
}
private HashTree makeCookieManagerThreadSafe(HashTree testTree) {
CustomTreeCloner cloner = new CustomTreeCloner();
testTree.traverse(cloner);
return cloner.getClonedTree();
}
public StandardJMeterEngine getStandardJMeterEngine() throws IllegalAccessException, NoSuchFieldException {
Field engine = StandardJMeterEngine.class.getDeclaredField("engine");
engine.setAccessible(true);
return (StandardJMeterEngine) engine.get(null);
}
public static class ParallelThreadFactory implements ThreadFactory {
private final ThreadGroup group;
private final String namePrefix;
public ParallelThreadFactory(String controllerName) {
SecurityManager s = System.getSecurityManager();
group = (s != null) ? s.getThreadGroup() :
Thread.currentThread().getThreadGroup();
namePrefix = "parallel " + controllerName;
}
public Thread newThread(Runnable r) {
Thread t = new Thread(group, r, namePrefix, 0);
cleanThreadContext(t);
if (t.isDaemon()) {
t.setDaemon(false);
}
if (t.getPriority() != Thread.NORM_PRIORITY) {
t.setPriority(Thread.NORM_PRIORITY);
}
return t;
}
private void cleanThreadContext(Thread thread) {
try {
Field field2 = Thread.class.getDeclaredField("inheritableThreadLocals");
field2.setAccessible(true);
field2.set(thread, null);
} catch (NoSuchFieldException | IllegalAccessException e) {
e.printStackTrace();
}
}
}
@Override
public void testEnded() {
}
@Override
public void testEnded(String s) {
}
}