提交 ee56ab27 编写于 作者: C chenjianxing

fix(接口测试): 解决支持jar包后无法加载部分类的问题

上级 133e1b2a
......@@ -150,10 +150,6 @@
<groupId>org.apache.logging.log4j</groupId>
<artifactId>log4j-slf4j-impl</artifactId>
</exclusion>
<exclusion>
<groupId>org.apache-extras.beanshell</groupId>
<artifactId>bsh</artifactId>
</exclusion>
</exclusions>
</dependency>
......@@ -520,15 +516,6 @@
<outputDirectory>src/main/resources/jmeter/lib/ext</outputDirectory>
<destFileName>jython-standalone.jar</destFileName>
</artifactItem>
<artifactItem>
<groupId>org.apache-extras.beanshell</groupId>
<artifactId>bsh</artifactId>
<version>2.0b6</version>
<type>jar</type>
<overWrite>true</overWrite>
<outputDirectory>src/main/resources/jmeter/lib/ext</outputDirectory>
<destFileName>bsh-2.0b6.jar</destFileName>
</artifactItem>
</artifactItems>
<outputDirectory>${project.build.directory}/wars</outputDirectory>
<overWriteReleases>false</overWriteReleases>
......
......@@ -6,13 +6,11 @@ import io.metersphere.commons.utils.LogUtil;
import io.metersphere.config.JmeterProperties;
import io.metersphere.i18n.Translator;
import org.apache.commons.lang3.StringUtils;
import org.apache.jmeter.NewDriver;
import org.apache.jmeter.config.Arguments;
import org.apache.jmeter.save.SaveService;
import org.apache.jmeter.util.JMeterUtils;
import org.apache.jmeter.visualizers.backend.BackendListener;
import org.apache.jorphan.collections.HashTree;
import org.springframework.context.i18n.LocaleContextHolder;
import org.springframework.stereotype.Service;
......@@ -30,10 +28,6 @@ public class JMeterService {
public void run(String testId, String debugReportId, InputStream is) {
String JMETER_HOME = getJmeterHome();
// 将当前类加载器设置为 loader ,解决由系统类加载器加载的 JMeter 无法动态加载 jar 包问题
// 同时隔离在 beanshell 中访问由系统类加载器加载的其他类
NewDriver.setContextClassLoader();
String JMETER_PROPERTIES = JMETER_HOME + "/bin/jmeter.properties";
JMeterUtils.loadJMeterProperties(JMETER_PROPERTIES);
JMeterUtils.setJMeterHome(JMETER_HOME);
......
......@@ -19,7 +19,6 @@
package org.apache.jmeter;
import java.io.File;
import java.io.IOException;
import java.io.PrintWriter;
import java.io.StringWriter;
import java.lang.reflect.Method;
......@@ -28,12 +27,7 @@ import java.net.URL;
import java.security.AccessController;
import java.security.PrivilegedAction;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Date;
import java.util.LinkedList;
import java.util.List;
import java.util.StringTokenizer;
import java.util.*;
/**
* Main class for JMeter - sets up initial classpath and the loader.
......@@ -65,6 +59,14 @@ public final class NewDriver {
Thread.currentThread().setContextClassLoader(loader);
}
public static void loaderClass(String name) {
try {
loader.loadClass(name);
} catch (ClassNotFoundException e) {
e.printStackTrace();
}
}
static {
final List<URL> jars = new LinkedList<>();
final String initiaClasspath = System.getProperty(JAVA_CLASS_PATH);
......@@ -72,28 +74,13 @@ public final class NewDriver {
// Find JMeter home dir from the initial classpath
String tmpDir;
// StringTokenizer tok = new StringTokenizer(initiaClasspath, File.pathSeparator);
// if (tok.countTokens() == 1
// || (tok.countTokens() == 2 // Java on Mac OS can add a second entry to the initial classpath
// && OS_NAME_LC.startsWith("mac os x")// $NON-NLS-1$
// )
// ) {
// File jar = new File(tok.nextToken());
// try {
// tmpDir = jar.getCanonicalFile().getParentFile().getParent();
// System.out.println(tmpDir + "111");
// } catch (IOException e) {
// tmpDir = null;
// }
// } else {// e.g. started from IDE with full classpath
//只从 jmeter.home 加载
tmpDir = System.getProperty("jmeter.home","");// Allow override $NON-NLS-1$ $NON-NLS-2$
if (tmpDir.length() == 0) {
File userDir = new File(System.getProperty("user.dir"));// $NON-NLS-1$
tmpDir = userDir.getAbsoluteFile().getParent();
}
// }
//只从 jmeter.home 加载
tmpDir = System.getProperty("jmeter.home","");// Allow override $NON-NLS-1$ $NON-NLS-2$
if (tmpDir.length() == 0) {
File userDir = new File(System.getProperty("user.dir"));// $NON-NLS-1$
tmpDir = userDir.getAbsoluteFile().getParent();
}
JMETER_INSTALLATION_DIRECTORY=tmpDir;
/*
......
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
*/
package org.apache.jmeter.util;
import java.io.BufferedReader;
import java.io.File;
import java.io.FileReader;
import java.io.IOException;
import java.io.Serializable;
import java.util.Collections;
import java.util.Map;
import java.util.Properties;
import javax.script.Bindings;
import javax.script.Compilable;
import javax.script.CompiledScript;
import javax.script.ScriptEngine;
import javax.script.ScriptEngineManager;
import javax.script.ScriptException;
import org.apache.commons.codec.digest.DigestUtils;
import org.apache.commons.collections.map.LRUMap;
import org.apache.commons.lang3.StringUtils;
import org.apache.jmeter.NewDriver;
import org.apache.jmeter.samplers.SampleResult;
import org.apache.jmeter.samplers.Sampler;
import org.apache.jmeter.testelement.TestStateListener;
import org.apache.jmeter.threads.JMeterContext;
import org.apache.jmeter.threads.JMeterContextService;
import org.apache.jmeter.threads.JMeterVariables;
import org.apache.jorphan.util.JOrphanUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
/**
* Base class for JSR223 Test elements
*
* 将当前类加载器设置为 loader ,解决由系统类加载器加载的 JMeter 无法动态加载 jar 包问题
* 同时隔离在 beanshell 中访问由系统类加载器加载的其他类
*/
public abstract class JSR223TestElement extends ScriptingTestElement
implements Serializable, TestStateListener
{
private static final long serialVersionUID = 232L;
private static final Logger logger = LoggerFactory.getLogger(JSR223TestElement.class);
/**
* Cache of compiled scripts
*/
@SuppressWarnings("unchecked") // LRUMap does not support generics (yet)
private static final Map<String, CompiledScript> compiledScriptsCache =
Collections.synchronizedMap(
new LRUMap(JMeterUtils.getPropDefault("jsr223.compiled_scripts_cache_size", 100)));
/** If not empty then script in ScriptText will be compiled and cached */
private String cacheKey = "";
/** md5 of the script, used as an unique key for the cache */
private String scriptMd5 = null;
/**
* Initialization On Demand Holder pattern
*/
private static class LazyHolder {
private LazyHolder() {
super();
}
public static final ScriptEngineManager INSTANCE = new ScriptEngineManager();
}
/**
* @return ScriptEngineManager singleton
*/
public static ScriptEngineManager getInstance() {
return LazyHolder.INSTANCE;
}
public JSR223TestElement() {
super();
}
/**
* @return {@link ScriptEngine} for language defaulting to groovy if language is not set
* @throws ScriptException when no {@link ScriptEngine} could be found
*/
protected ScriptEngine getScriptEngine() throws ScriptException {
String lang = getScriptLanguageWithDefault();
ScriptEngine scriptEngine = getInstance().getEngineByName(lang);
if (scriptEngine == null) {
throw new ScriptException("Cannot find engine named: '"+lang+"', ensure you set language field in JSR223 Test Element: "+getName());
}
return scriptEngine;
}
/**
* @return script language or DEFAULT_SCRIPT_LANGUAGE if none is set
*/
private String getScriptLanguageWithDefault() {
String lang = getScriptLanguage();
if (StringUtils.isNotEmpty(lang)) {
return lang;
}
return DEFAULT_SCRIPT_LANGUAGE;
}
/**
* Populate variables to be passed to scripts
* @param bindings Bindings
*/
protected void populateBindings(Bindings bindings) {
final String label = getName();
final String fileName = getFilename();
final String scriptParameters = getParameters();
// Use actual class name for log
final Logger elementLogger = LoggerFactory.getLogger(getClass().getName()+"."+getName());
bindings.put("log", elementLogger); // $NON-NLS-1$ (this name is fixed)
bindings.put("Label", label); // $NON-NLS-1$ (this name is fixed)
bindings.put("FileName", fileName); // $NON-NLS-1$ (this name is fixed)
bindings.put("Parameters", scriptParameters); // $NON-NLS-1$ (this name is fixed)
String[] args=JOrphanUtils.split(scriptParameters, " ");//$NON-NLS-1$
bindings.put("args", args); // $NON-NLS-1$ (this name is fixed)
// Add variables for access to context and variables
JMeterContext jmctx = JMeterContextService.getContext();
bindings.put("ctx", jmctx); // $NON-NLS-1$ (this name is fixed)
JMeterVariables vars = jmctx.getVariables();
bindings.put("vars", vars); // $NON-NLS-1$ (this name is fixed)
Properties props = JMeterUtils.getJMeterProperties();
bindings.put("props", props); // $NON-NLS-1$ (this name is fixed)
// For use in debugging:
bindings.put("OUT", System.out); // NOSONAR $NON-NLS-1$ (this name is fixed)
// Most subclasses will need these:
Sampler sampler = jmctx.getCurrentSampler();
bindings.put("sampler", sampler); // $NON-NLS-1$ (this name is fixed)
SampleResult prev = jmctx.getPreviousResult();
bindings.put("prev", prev); // $NON-NLS-1$ (this name is fixed)
}
/**
* This method will run inline script or file script with special behaviour for file script:
* - If ScriptEngine implements Compilable script will be compiled and cached
* - If not if will be run
* @param scriptEngine ScriptEngine
* @param pBindings {@link Bindings} might be null
* @return Object returned by script
* @throws IOException when reading the script fails
* @throws ScriptException when compiling or evaluation of the script fails
*/
protected Object processFileOrScript(ScriptEngine scriptEngine, final Bindings pBindings)
throws IOException, ScriptException {
Bindings bindings = pBindings;
if (bindings == null) {
bindings = scriptEngine.createBindings();
}
populateBindings(bindings);
File scriptFile = new File(getFilename());
// Hack: bsh-2.0b5.jar BshScriptEngine implements Compilable but throws
// "java.lang.Error: unimplemented"
boolean supportsCompilable = scriptEngine instanceof Compilable
&& !("bsh.engine.BshScriptEngine".equals(scriptEngine.getClass().getName())); // NOSONAR // $NON-NLS-1$
ClassLoader contextClassLoader = Thread.currentThread().getContextClassLoader();
// 将当前类加载器设置为 loader ,解决由系统类加载器加载的 JMeter 无法动态加载 jar 包问题
// 同时隔离在 beanshell 中访问由系统类加载器加载的其他类
NewDriver.setContextClassLoader();
try {
if (!StringUtils.isEmpty(getFilename())) {
if (scriptFile.exists() && scriptFile.canRead()) {
if (supportsCompilable) {
String newCacheKey = getScriptLanguage() + "#" + // $NON-NLS-1$
scriptFile.getAbsolutePath() + "#" + // $NON-NLS-1$
scriptFile.lastModified();
CompiledScript compiledScript = compiledScriptsCache.get(newCacheKey);
if (compiledScript == null) {
synchronized (compiledScriptsCache) {
compiledScript = compiledScriptsCache.get(newCacheKey);
if (compiledScript == null) {
// TODO Charset ?
try (BufferedReader fileReader = new BufferedReader(new FileReader(scriptFile),
(int) scriptFile.length())) {
compiledScript = ((Compilable) scriptEngine).compile(fileReader);
compiledScriptsCache.put(newCacheKey, compiledScript);
}
}
}
}
return compiledScript.eval(bindings);
} else {
// TODO Charset ?
try (BufferedReader fileReader = new BufferedReader(new FileReader(scriptFile),
(int) scriptFile.length())) {
return scriptEngine.eval(fileReader, bindings);
}
}
} else {
throw new ScriptException("Script file '" + scriptFile.getAbsolutePath()
+ "' does not exist or is unreadable for element:" + getName());
}
} else if (!StringUtils.isEmpty(getScript())) {
if (supportsCompilable &&
!ScriptingBeanInfoSupport.FALSE_AS_STRING.equals(cacheKey)) {
computeScriptMD5();
CompiledScript compiledScript = compiledScriptsCache.get(this.scriptMd5);
if (compiledScript == null) {
synchronized (compiledScriptsCache) {
compiledScript = compiledScriptsCache.get(this.scriptMd5);
if (compiledScript == null) {
compiledScript = ((Compilable) scriptEngine).compile(getScript());
compiledScriptsCache.put(this.scriptMd5, compiledScript);
}
}
}
return compiledScript.eval(bindings);
} else {
return scriptEngine.eval(getScript(), bindings);
}
} else {
throw new ScriptException("Both script file and script text are empty for element:" + getName());
}
} catch (ScriptException ex) {
Throwable rootCause = ex.getCause();
if(isStopCondition(rootCause)) {
throw (RuntimeException) ex.getCause();
} else {
throw ex;
}
} finally {
// 将当前类加载器设置回来,避免加载无法加载到系统类加载加载类
Thread.currentThread().setContextClassLoader(contextClassLoader);
}
}
/**
* @return boolean true if element is not compilable or if compilation succeeds
* @throws IOException if script is missing
* @throws ScriptException if compilation fails
*/
public boolean compile()
throws ScriptException, IOException {
String lang = getScriptLanguageWithDefault();
ScriptEngine scriptEngine = getInstance().getEngineByName(lang);
boolean supportsCompilable = scriptEngine instanceof Compilable
&& !("bsh.engine.BshScriptEngine".equals(scriptEngine.getClass().getName())); // NOSONAR // $NON-NLS-1$
if(!supportsCompilable) {
return true;
}
if (!StringUtils.isEmpty(getScript())) {
try {
((Compilable) scriptEngine).compile(getScript());
return true;
} catch (ScriptException e) { // NOSONAR
logger.error("Error compiling script for test element {}, error:{}", getName(), e.getMessage());
return false;
}
} else {
File scriptFile = new File(getFilename());
try (BufferedReader fileReader = new BufferedReader(new FileReader(scriptFile),
(int) scriptFile.length())) {
try {
((Compilable) scriptEngine).compile(fileReader);
return true;
} catch (ScriptException e) { // NOSONAR
logger.error("Error compiling script for test element {}, error:{}", getName(), e.getMessage());
return false;
}
}
}
}
/**
* compute MD5 if it is null
*/
private void computeScriptMD5() {
// compute the md5 of the script if needed
if(scriptMd5 == null) {
scriptMd5 = DigestUtils.md5Hex(getScript());
}
}
/**
* @return the cacheKey
*/
public String getCacheKey() {
return cacheKey;
}
/**
* @param cacheKey the cacheKey to set
*/
public void setCacheKey(String cacheKey) {
this.cacheKey = cacheKey;
}
/**
* @see org.apache.jmeter.testelement.TestStateListener#testStarted()
*/
@Override
public void testStarted() {
// NOOP
}
/**
* @see org.apache.jmeter.testelement.TestStateListener#testStarted(java.lang.String)
*/
@Override
public void testStarted(String host) {
// NOOP
}
/**
* @see org.apache.jmeter.testelement.TestStateListener#testEnded()
*/
@Override
public void testEnded() {
testEnded("");
}
/**
* @see org.apache.jmeter.testelement.TestStateListener#testEnded(java.lang.String)
*/
@Override
public void testEnded(String host) {
compiledScriptsCache.clear();
this.scriptMd5 = null;
}
public String getScriptLanguage() {
return scriptLanguage;
}
public void setScriptLanguage(String s) {
scriptLanguage = s;
}
}
\ No newline at end of file
//
// Source code recreated from a .class file by IntelliJ IDEA
// (powered by Fernflower decompiler)
//
package org.apache.jmeter.visualizers.backend;
import java.io.Serializable;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ArrayBlockingQueue;
import java.util.concurrent.BlockingQueue;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.atomic.LongAdder;
import java.util.concurrent.locks.LockSupport;
import org.apache.jmeter.config.Arguments;
import org.apache.jmeter.engine.util.NoThreadClone;
import org.apache.jmeter.samplers.Remoteable;
import org.apache.jmeter.samplers.SampleEvent;
import org.apache.jmeter.samplers.SampleListener;
import org.apache.jmeter.samplers.SampleResult;
import org.apache.jmeter.testelement.AbstractTestElement;
import org.apache.jmeter.testelement.TestStateListener;
import org.apache.jmeter.testelement.property.TestElementProperty;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
// 当前上下文类加载为 NewDriver 的类加载器,这里使用系统类加载器加载 APIBackendListenerClient
public class BackendListener extends AbstractTestElement implements Backend, Serializable, SampleListener, TestStateListener, NoThreadClone, Remoteable {
private static final long serialVersionUID = 1L;
private static final Logger log = LoggerFactory.getLogger(BackendListener.class);
public static final String CLASSNAME = "classname";
public static final String QUEUE_SIZE = "QUEUE_SIZE";
private static final Object LOCK = new Object();
public static final String ARGUMENTS = "arguments";
private Class<?> clientClass;
public static final String DEFAULT_QUEUE_SIZE = "5000";
private static final transient SampleResult FINAL_SAMPLE_RESULT = new SampleResult();
private static final Map<String, BackendListener.ListenerClientData> queuesByTestElementName = new ConcurrentHashMap();
private transient String myName;
private transient BackendListener.ListenerClientData listenerClientData;
public BackendListener() {
this.setArguments(new Arguments());
}
public Object clone() {
BackendListener clone = (BackendListener)super.clone();
clone.clientClass = this.clientClass;
return clone;
}
private Class<?> initClass() {
String name = this.getClassname().trim();
try {
// 当前上下文类加载为 NewDriver 的类加载器,这里使用系统类加载器加载 APIBackendListenerClient
return Class.forName(name, false, this.getClass().getClassLoader());
} catch (Exception var3) {
log.error("{}\tException initialising: {}", new Object[]{this.whoAmI(), name, var3});
return null;
}
}
private String whoAmI() {
return Thread.currentThread().getName() + "@" + Integer.toHexString(this.hashCode()) + "-" + this.getName();
}
public void sampleOccurred(SampleEvent event) {
Arguments args = this.getArguments();
BackendListenerContext context = new BackendListenerContext(args);
SampleResult sr = this.listenerClientData.client.createSampleResult(context, event.getResult());
if (sr == null) {
if (log.isDebugEnabled()) {
log.debug("{} => Dropping SampleResult: {}", this.getName(), event.getResult());
}
} else {
try {
if (!this.listenerClientData.queue.offer(sr)) {
this.listenerClientData.queueWaits.add(1L);
long t1 = System.nanoTime();
this.listenerClientData.queue.put(sr);
long t2 = System.nanoTime();
this.listenerClientData.queueWaitTime.add(t2 - t1);
}
} catch (Exception var9) {
log.error("sampleOccurred, failed to queue the sample", var9);
}
}
}
private static void sendToListener(BackendListenerClient backendListenerClient, BackendListenerContext context, List<SampleResult> sampleResults) {
if (!sampleResults.isEmpty()) {
backendListenerClient.handleSampleResults(sampleResults, context);
sampleResults.clear();
}
}
private static BackendListenerClient createBackendListenerClientImpl(Class<?> clientClass) {
if (clientClass == null) {
return new BackendListener.ErrorBackendListenerClient();
} else {
try {
return (BackendListenerClient)clientClass.getDeclaredConstructor().newInstance();
} catch (Exception var2) {
log.error("Exception creating: {}", clientClass, var2);
return new BackendListener.ErrorBackendListenerClient();
}
}
}
public void testStarted() {
this.testStarted("local");
}
public void testStarted(String host) {
if (log.isDebugEnabled()) {
log.debug("{}\ttestStarted({})", this.whoAmI(), host);
}
String size = this.getQueueSize();
int queueSize;
try {
queueSize = Integer.parseInt(size);
} catch (NumberFormatException var12) {
log.warn("Invalid queue size '{}' defaulting to {}", size, "5000");
queueSize = Integer.parseInt("5000");
}
Object var4 = LOCK;
synchronized(LOCK) {
this.myName = this.getName();
this.listenerClientData = (BackendListener.ListenerClientData)queuesByTestElementName.get(this.myName);
if (this.listenerClientData == null) {
this.clientClass = this.initClass();
BackendListenerClient backendListenerClient = createBackendListenerClientImpl(this.clientClass);
BackendListenerContext context = new BackendListenerContext((Arguments)this.getArguments().clone());
this.listenerClientData = new BackendListener.ListenerClientData();
this.listenerClientData.queue = new ArrayBlockingQueue(queueSize);
this.listenerClientData.queueWaits = new LongAdder();
this.listenerClientData.queueWaitTime = new LongAdder();
this.listenerClientData.latch = new CountDownLatch(1);
this.listenerClientData.client = backendListenerClient;
if (log.isInfoEnabled()) {
log.info("{}: Starting worker with class: {} and queue capacity: {}", new Object[]{this.getName(), this.clientClass, this.getQueueSize()});
}
BackendListener.Worker worker = new BackendListener.Worker(backendListenerClient, (Arguments)this.getArguments().clone(), this.listenerClientData);
worker.setDaemon(true);
worker.start();
if (log.isInfoEnabled()) {
log.info("{}: Started worker with class: {}", this.getName(), this.clientClass);
}
try {
backendListenerClient.setupTest(context);
} catch (Exception var10) {
throw new IllegalStateException("Failed calling setupTest", var10);
}
queuesByTestElementName.put(this.myName, this.listenerClientData);
}
this.listenerClientData.instanceCount++;
}
}
public void testEnded(String host) {
Object var2 = LOCK;
synchronized(LOCK) {
BackendListener.ListenerClientData listenerClientDataForName = (BackendListener.ListenerClientData)queuesByTestElementName.get(this.myName);
if (log.isDebugEnabled()) {
log.debug("testEnded called on instance {}#{}", this.myName, listenerClientDataForName.instanceCount);
}
if (listenerClientDataForName != null) {
listenerClientDataForName.instanceCount--;
if (listenerClientDataForName.instanceCount > 0) {
return;
}
queuesByTestElementName.remove(this.myName);
} else {
log.error("No listener client data found for BackendListener {}", this.myName);
}
}
try {
this.listenerClientData.queue.put(FINAL_SAMPLE_RESULT);
} catch (Exception var6) {
log.warn("testEnded() with exception: {}", var6.getMessage(), var6);
}
if (this.listenerClientData.queueWaits.longValue() > 0L) {
log.warn("QueueWaits: {}; QueueWaitTime: {} (nanoseconds), you may need to increase queue capacity, see property 'backend_queue_capacity'", this.listenerClientData.queueWaits, this.listenerClientData.queueWaitTime);
}
try {
this.listenerClientData.latch.await();
BackendListenerContext context = new BackendListenerContext(this.getArguments());
this.listenerClientData.client.teardownTest(context);
} catch (Exception var5) {
throw new IllegalStateException("Failed calling teardownTest", var5);
}
}
public void testEnded() {
this.testEnded("local");
}
public void sampleStarted(SampleEvent e) {
}
public void sampleStopped(SampleEvent e) {
}
public void setArguments(Arguments args) {
args.removeArgument("useRegexpForSamplersList", "false");
this.setProperty(new TestElementProperty("arguments", args));
}
public Arguments getArguments() {
return (Arguments)this.getProperty("arguments").getObjectValue();
}
public void setClassname(String classname) {
this.setProperty("classname", classname);
}
public String getClassname() {
return this.getPropertyAsString("classname");
}
public void setQueueSize(String queueSize) {
this.setProperty("QUEUE_SIZE", queueSize, "5000");
}
public String getQueueSize() {
return this.getPropertyAsString("QUEUE_SIZE", "5000");
}
static class ErrorBackendListenerClient extends AbstractBackendListenerClient {
ErrorBackendListenerClient() {
}
public void handleSampleResults(List<SampleResult> sampleResults, BackendListenerContext context) {
BackendListener.log.warn("ErrorBackendListenerClient#handleSampleResult called, noop");
Thread.yield();
}
}
private static final class Worker extends Thread {
private final BackendListener.ListenerClientData listenerClientData;
private final BackendListenerContext context;
private final BackendListenerClient backendListenerClient;
private Worker(BackendListenerClient backendListenerClient, Arguments arguments, BackendListener.ListenerClientData listenerClientData) {
this.listenerClientData = listenerClientData;
arguments.addArgument("TestElement.name", this.getName());
this.context = new BackendListenerContext(arguments);
this.backendListenerClient = backendListenerClient;
}
public void run() {
boolean isDebugEnabled = BackendListener.log.isDebugEnabled();
ArrayList sampleResults = new ArrayList(this.listenerClientData.queue.size());
try {
try {
boolean endOfLoop = false;
while(!endOfLoop) {
if (isDebugEnabled) {
BackendListener.log.debug("Thread: {} taking SampleResult from queue: {}", Thread.currentThread().getName(), this.listenerClientData.queue.size());
}
SampleResult sampleResult = (SampleResult)this.listenerClientData.queue.take();
if (isDebugEnabled) {
BackendListener.log.debug("Thread: {} took SampleResult: {}, isFinal: {}", new Object[]{Thread.currentThread().getName(), sampleResult, sampleResult == BackendListener.FINAL_SAMPLE_RESULT});
}
while(!(endOfLoop = sampleResult == BackendListener.FINAL_SAMPLE_RESULT) && sampleResult != null) {
sampleResults.add(sampleResult);
if (isDebugEnabled) {
BackendListener.log.debug("Thread: {} polling from queue: {}", Thread.currentThread().getName(), this.listenerClientData.queue.size());
}
sampleResult = (SampleResult)this.listenerClientData.queue.poll();
if (isDebugEnabled) {
BackendListener.log.debug("Thread: {} took from queue: {}, isFinal: {}", new Object[]{Thread.currentThread().getName(), sampleResult, sampleResult == BackendListener.FINAL_SAMPLE_RESULT});
}
}
if (isDebugEnabled) {
BackendListener.log.debug("Thread: {} exiting with FINAL EVENT: {}, null: {}", new Object[]{Thread.currentThread().getName(), sampleResult == BackendListener.FINAL_SAMPLE_RESULT, sampleResult == null});
}
BackendListener.sendToListener(this.backendListenerClient, this.context, sampleResults);
if (!endOfLoop) {
LockSupport.parkNanos(100L);
}
}
} catch (InterruptedException var8) {
Thread.currentThread().interrupt();
}
BackendListener.sendToListener(this.backendListenerClient, this.context, sampleResults);
BackendListener.log.info("Worker ended");
} finally {
this.listenerClientData.latch.countDown();
}
}
}
private static final class ListenerClientData {
private BackendListenerClient client;
private BlockingQueue<SampleResult> queue;
private LongAdder queueWaits;
private LongAdder queueWaitTime;
private int instanceCount;
private CountDownLatch latch;
private ListenerClientData() {
}
}
}
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册