Java JMeterContext类代码示例

import org.apache.jmeter.threads.JMeterContext; //导入依赖的package包/类
@Override
public void process(){
    final BeanShellInterpreter bshInterpreter = getBeanShellInterpreter();
    if (bshInterpreter == null) {
        log.error("BeanShell not found");
        return;
    }
    JMeterContext jmctx = JMeterContextService.getContext();
    Sampler sam = jmctx.getCurrentSampler();
    try {
        // Add variables for access to context and variables
        bshInterpreter.set("sampler", sam);//$NON-NLS-1$
        processFileOrScript(bshInterpreter);
    } catch (JMeterException e) {
        log.warn("Problem in BeanShell script "+e);
    }
}
 

import org.apache.jmeter.threads.JMeterContext; //导入依赖的package包/类
public void testBug54467() throws Exception {
    JMeterContext jmctx = JMeterContextService.getContext();
    LoopController loop = new LoopController();
    Map<String, String> variables = new HashMap<String, String>();
    ReplaceStringWithFunctions transformer = new ReplaceStringWithFunctions(new CompoundVariable(), variables);
    jmctx.setVariables(new JMeterVariables());

    StringProperty prop = new StringProperty(LoopController.LOOPS,"${__Random(1,12,)}");
    JMeterProperty newProp = transformer.transformValue(prop);
    newProp.setRunningVersion(true);
    
    loop.setProperty(newProp);
    loop.addTestElement(new TestSampler("random run"));
    loop.setRunningVersion(true);
    loop.initialize();
    int loops = loop.getLoops();
    for (int i = 0; i < loops; i++) {
        Sampler s = loop.next();
        assertNotNull(s);
    }
    assertNull(loop.next());    
}
 

import org.apache.jmeter.threads.JMeterContext; //导入依赖的package包/类
@Override
public void send(JMeterContext context, SampleResult result) {
    JMeterVariables variables = context.getVariables();
    ConcurrentHashMap<String, MockLwM2mClient> clients = (ConcurrentHashMap<String, MockLwM2mClient>) variables.getObject("lwm2mClients");
    MockLwM2mClient client = clients.get(endpoint);
    if (client != null) {
        SimpleResource resource = (SimpleResource) client.getObject(objectId).getChild(instanceId).getChild(resourceId);
        result.sampleStart();
        if (resource != null) {
            resource.setResourceValue(Float.toString(rng.nextInt(100)));
            result.setSuccessful(true);
            log.debug("Sent observation for " + objectId + "/" + instanceId + "/" + resourceId);
        } else {
            result.setSuccessful(false);
            log.debug("Could not send observation for " + objectId + "/" + instanceId + "/" + resourceId);
        }
        result.sampleEnd();
    }
}
 

import org.apache.jmeter.threads.JMeterContext; //导入依赖的package包/类
private void changeVariablesMap() {
    try {
        JMeterContext context = this.getThreadContext();
        if (context != null && context.getVariables() != null) {
            JMeterVariables jMeterVariables = context.getVariables();
            Class<JMeterVariables> 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);
    }
}
 

import org.apache.jmeter.threads.JMeterContext; //导入依赖的package包/类
@Before
public void setup() throws IOException {

    zkServer = new EmbeddedZookeeper();

    String zkConnect = ZKHOST + ":" + zkServer.port();
    zkClient = new ZkClient(zkConnect, 30000, 30000, ZKStringSerializer$.MODULE$);
    ZkUtils zkUtils = ZkUtils.apply(zkClient, false);

    Properties brokerProps = new Properties();
    brokerProps.setProperty("zookeeper.connect", zkConnect);
    brokerProps.setProperty("broker.id", "0");
    brokerProps.setProperty("log.dirs", Files.createTempDirectory("kafka-").toAbsolutePath().toString());
    brokerProps.setProperty("listeners", "PLAINTEXT://" + BROKERHOST +":" + BROKERPORT);
    KafkaConfig config = new KafkaConfig(brokerProps);
    Time mock = new MockTime();
    kafkaServer = TestUtils.createServer(config, mock);
    //AdminUtils.createTopic(zkUtils, TOPIC, 1, 1, new Properties(), RackAwareMode.Disabled$.MODULE$);

    JMeterContext jmcx = JMeterContextService.getContext();
    jmcx.setVariables(new JMeterVariables());

}
 

import org.apache.jmeter.threads.JMeterContext; //导入依赖的package包/类
@Override
public void iterationStart(LoopIterationEvent event) {
	TestElement source = event.getSource();
	JMeterContext threadContext = source.getThreadContext();
	JMeterVariables variables = threadContext.getVariables();
	iterationStart(variables);
}
 

import org.apache.jmeter.threads.JMeterContext; //导入依赖的package包/类
public static ApplicationContext getApplicationContext() {
	JMeterContext context = JMeterContextService.getContext();
	JMeterVariables variables = context.getVariables();
	Object o = variables.getObject(VARIABLE);
	checkNotNull(o, "variable %s not found", VARIABLE);
	return ApplicationContext.class.cast(o);
}
 

import org.apache.jmeter.threads.JMeterContext; //导入依赖的package包/类
/**
 * This is executed once per instantiation of this class. Nothing specific to an
 * individual test run should appear in this method.
 *
 * @param context ignored
 */
@Override
public void setupTest(JavaSamplerContext context) {
	try {
		assertNotInterrupted();
		executing = true;
		super.setupTest(context);

		JMeterContext jmeterContext = JMeterContextService.getContext();
		AbstractThreadGroup threadGroup = jmeterContext.getThreadGroup();
		threadGroupName = threadGroup.getName();

		JMeterThread thread = jmeterContext.getThread();
		threadName = thread.getThreadName();

		String parameter = context.getParameter(PARAMETER, PARAMETER_DEFAULT).trim();
		boolean generatingJson = Boolean.valueOf(parameter);

		assertNotInterrupted();
		if (generatingJson) {
			marshaller = SpringPreProcessor.getBean(MartiniResultMarshaller.class);
		}
		eventManager = SpringPreProcessor.getBean(EventManager.class);
		suiteIdentifier = SpringPreProcessor.getBean(SuiteIdentifier.class);
	}
	finally {
		executing = false;
	}
}
 

import org.apache.jmeter.threads.JMeterContext; //导入依赖的package包/类
private DebuggingThread makeThread(int groupCount, ListenerNotifier notifier, ListedHashTree threadGroupTree, StandardJMeterEngine engine, int i, JMeterContext context) {
    // had to copy whole method because of this line
    DebuggingThread jmeterThread = new DebuggingThread(threadGroupTree, this, notifier, context);

    boolean onErrorStopTest = getOnErrorStopTest();
    boolean onErrorStopTestNow = getOnErrorStopTestNow();
    boolean onErrorStopThread = getOnErrorStopThread();
    boolean onErrorStartNextLoop = getOnErrorStartNextLoop();
    String groupName = getName();

    jmeterThread.setThreadNum(i);
    jmeterThread.setThreadGroup(this);
    jmeterThread.setInitialContext(context);
    String threadName = groupName + " " + (groupCount) + "-" + (i + 1);
    jmeterThread.setThreadName(threadName);
    jmeterThread.setEngine(engine);
    jmeterThread.setOnErrorStopTest(onErrorStopTest);
    jmeterThread.setOnErrorStopTestNow(onErrorStopTestNow);
    jmeterThread.setOnErrorStopThread(onErrorStopThread);
    jmeterThread.setOnErrorStartNextLoop(onErrorStartNextLoop);
    return jmeterThread;
}
 

import org.apache.jmeter.threads.JMeterContext; //导入依赖的package包/类
@Override
public void statusRefresh(JMeterContext context) {
    try {
        refreshVars(context);
        refreshProperties();
    } catch (Throwable e) {
        log.warn("Problem refreshing status pane", e);
    }
    evaluatePanel.refresh(context, debugger.isContinuing());
    SwingUtilities.invokeLater(new Runnable() {
        @Override
        public void run() {
            tree.repaint();
        }
    });
}
 

import org.apache.jmeter.threads.JMeterContext; //导入依赖的package包/类
/**
 * {@inheritDoc}
 */
@Override
public boolean isDone() {
    if (loopCount >= getEndIndex()) {
        return true;
    }
    JMeterContext context = getThreadContext();
    StringBuilder builder = new StringBuilder(
            getInputVal().length()+getSeparator().length()+3);
    String inputVariable = 
            builder.append(getInputVal())
            .append(getSeparator())
            .append(Integer.toString(loopCount+1)).toString();
    final JMeterVariables variables = context.getVariables();
    final Object currentVariable = variables.getObject(inputVariable);
    if (currentVariable != null) {
        variables.putObject(getReturnVal(), currentVariable);
        if (log.isDebugEnabled()) {
            log.debug("ForEach resultstring isDone=" + variables.get(getReturnVal()));
        }
        return false;
    }
    return super.isDone();
}
 

import org.apache.jmeter.threads.JMeterContext; //导入依赖的package包/类
/**
 * Check if there are any matching entries
 *
 * @return whether any entries in the list
 */
private boolean emptyList() {
    JMeterContext context = getThreadContext();

    StringBuilder builder = new StringBuilder(
            getInputVal().length()+getSeparator().length()+3);
    String inputVariable = 
            builder.append(getInputVal())
            .append(getSeparator())
            .append(Integer.toString(loopCount+1)).toString();
    if (context.getVariables().getObject(inputVariable) != null) {
        return false;
    }
    if (log.isDebugEnabled()) {
        log.debug("No entries found - null first entry: " + inputVariable);
    }
    return true;
}
 

import org.apache.jmeter.threads.JMeterContext; //导入依赖的package包/类
@Override
public void process() {
    JMeterContext jmctx = JMeterContextService.getContext();

    SampleResult prev = jmctx.getPreviousResult();
    if (prev == null) {
        return; // TODO - should we skip processing here?
    }
    final BeanShellInterpreter bshInterpreter = getBeanShellInterpreter();
    if (bshInterpreter == null) {
        log.error("BeanShell not found");
        return;
    }

    try {
        // Add variables for access to context and variables
        bshInterpreter.set("data", prev.getResponseData());//$NON-NLS-1$
        processFileOrScript(bshInterpreter);
    } catch (JMeterException e) {
        log.warn("Problem in BeanShell script "+e);
    }
}
 

import org.apache.jmeter.threads.JMeterContext; //导入依赖的package包/类
/**
 * Create additional SampleEvent in NON Parent Mode
 */
protected void notifyListeners() {
    // TODO could these be done earlier (or just once?)
    JMeterContext threadContext = getThreadContext();
    JMeterVariables threadVars = threadContext.getVariables();
    SamplePackage pack = (SamplePackage) threadVars.getObject(JMeterThread.PACKAGE_OBJECT);
    if (pack == null) {
        // If child of TransactionController is a ThroughputController and TPC does
        // not sample its children, then we will have this
        // TODO Should this be at warn level ?
        log.warn("Could not fetch SamplePackage");
    } else {
        SampleEvent event = new SampleEvent(res, threadContext.getThreadGroup().getName(),threadVars, true);
        // We must set res to null now, before sending the event for the transaction,
        // so that we can ignore that event in our sampleOccured method
        res = null;
        lnf.notifyListeners(event, pack.getSampleListeners());
    }
}
 

import org.apache.jmeter.threads.JMeterContext; //导入依赖的package包/类
/**
 * Executes the function (and caches the value for the duration of the test
 * iteration) if the property is a running version. Otherwise, the raw
 * string representation of the function is provided.
 *
 * @see JMeterProperty#getStringValue()
 */
@Override
public String getStringValue() {
    JMeterContext ctx = JMeterContextService.getContext();// Expensive, so
                                                            // do
    // once
    if (!isRunningVersion() /*|| !ctx.isSamplingStarted()*/) {
        log.debug("Not running version, return raw function string");
        return function.getRawParameters();
    }
    if(!ctx.isSamplingStarted()) {
        return function.execute();
    }
    log.debug("Running version, executing function");
    int iter = ctx.getVariables() != null ? ctx.getVariables().getIteration() : -1;
    if (iter < testIteration) {
        testIteration = -1;
    }
    if (iter > testIteration || cacheValue == null) {
        testIteration = iter;
        cacheValue = function.execute();
    }
    return cacheValue;

}