mstover1 2003/03/15 06:44:52
Modified: src/protocol/java/org/apache/jmeter/protocol/java/sampler
JavaSampler.java
Log:
Java Sampler updated (Jeremy Arnold)
Revision Changes Path
1.6 +279 -196
jakarta-jmeter/src/protocol/java/org/apache/jmeter/protocol/java/sampler/JavaSampler.java
Index: JavaSampler.java
===================================================================
RCS file:
/home/cvs/jakarta-jmeter/src/protocol/java/org/apache/jmeter/protocol/java/sampler/JavaSampler.java,v
retrieving revision 1.5
retrieving revision 1.6
diff -u -r1.5 -r1.6
--- JavaSampler.java 14 Mar 2003 21:15:37 -0000 1.5
+++ JavaSampler.java 15 Mar 2003 14:44:52 -0000 1.6
@@ -2,7 +2,7 @@
* ====================================================================
* The Apache Software License, Version 1.1
*
- * Copyright (c) 2002 The Apache Software Foundation. All rights
+ * Copyright (c) 2002,2003 The Apache Software Foundation. All rights
* reserved.
*
* Redistribution and use in source and binary forms, with or without
@@ -56,7 +56,9 @@
package org.apache.jmeter.protocol.java.sampler;
import java.util.HashMap;
+import java.util.HashSet;
import java.util.Iterator;
+import java.util.Set;
import org.apache.jmeter.config.Argument;
import org.apache.jmeter.config.Arguments;
@@ -66,204 +68,285 @@
import org.apache.jmeter.samplers.SampleResult;
import org.apache.jmeter.testelement.PerThreadClonable;
import org.apache.jmeter.testelement.TestElement;
+import org.apache.jmeter.testelement.TestListener;
import org.apache.log.Hierarchy;
import org.apache.log.Logger;
-/**
- *
- */
-
-
-public class JavaSampler extends AbstractSampler
- implements JavaSamplerClient, PerThreadClonable {
+public class JavaSampler extends AbstractSampler implements JavaSamplerClient,
PerThreadClonable, TestListener
+{
- /** Handle to Java client. **/
-
- public final static String CLASSNAME = "classname";
-
- public final static String ARGUMENTS = "arguments";
-
- transient private JavaSamplerClient javaClient = null;
-
- /** Logging */
- transient private static Logger log =
Hierarchy.getDefaultHierarchy().getLoggerFor("jmeter.protocol.java");
-
- public JavaSampler() {
- setArguments(new Arguments());
- }
-
- public void setArguments(Arguments args)
- {
- this.setProperty(ARGUMENTS,args);
- }
-
- public Arguments getArguments()
- {
- return (Arguments)getProperty(ARGUMENTS);
- }
-
- public void addCustomTestElement(TestElement el)
- {
- if(el instanceof JavaConfig)
- {
- mergeIn(el);
- }
- }
-
- /**
- * Releases Java Client.
- */
-
- public void releaseJavaClient () {
- if (javaClient != null) {
-
javaClient.teardownTest(createArgumentsHashMap(getArguments()));
- }
- javaClient = null;
- }
-
- /**
- * Sets the Classname attribute of the JavaConfig object
- *
- [EMAIL PROTECTED] classname The new Classname value
- */
- public void setClassname(String classname)
- {
- this.setProperty(CLASSNAME, classname);
- }
-
-
-
- /**
- * Gets the Classname attribute of the JavaConfig object
- *
- [EMAIL PROTECTED] The Classname value
- */
- public String getClassname()
- {
- return this.getPropertyAsString(CLASSNAME);
- }
-
- /**
- * Performs a test sample.
- *
- * The <code>sample()</code> method retrieves the reference to the
- * Java client and calls its <code>runTest()</code> method.
- * @see Sampler#sample()
- * @see JavaSamplerClient#runTest()
- *
- * @return test SampleResult
- */
-
-
- public SampleResult sample(Entry entry) {
- return
createJavaClient().runTest(createArgumentsHashMap(getArguments()));
- }
-
- public HashMap createArgumentsHashMap(Arguments args)
- {
- HashMap newArgs = new HashMap();
- Iterator iter = args.iterator();
- while (iter.hasNext())
- {
- Argument item = (Argument)iter.next();
- newArgs.put(item.getName(),item.getValue());
- }
-
- return newArgs;
- }
-
- /**
- * Returns reference to <code>JavaSamplerClient</code>.
- *
- * The <code>createJavaClient()</code> method uses reflection
- * to create an instance of the specified Java protocol client.
- * If the class can not be found, the method returns a reference
- * to <code>this</code> object.
- *
- * @return JavaSamplerClient reference.
- */
-
- public JavaSamplerClient createJavaClient() {
- if (javaClient == null) {
- try {
- Class javaClass = Class.forName(getClassname().trim(),
-
false,Thread.currentThread().getContextClassLoader());
- java.lang.reflect.Constructor[] constructors =
javaClass.getConstructors();
-
- for (int i = 0; i < constructors.length; i++) {
- Class[] params =
constructors[i].getParameterTypes();
- if (params.length == 0) {
- Object[] args = {};
- javaClient =
(JavaSamplerClient)constructors[i].newInstance(args);
-
javaClient.setupTest(createArgumentsHashMap(getArguments()));
- if (log.isDebugEnabled()) {
- log.debug(whoAmI() +
"\tCreated:\t"+ getClassname()+ "@" + Integer.toHexString(javaClient.hashCode()));
- }
- break;
- }
- }
-
- } catch (Exception e) {
- log.error(whoAmI() + "\tException creating: " +
getClassname(),e);
- javaClient = this;
- }
- }
-
- return javaClient;
-
- }
-
- /**
- * Retrieves reference to JavaSamplerClient.
- *
- * Convience method used to check for null reference without
- * actually creating a JavaSamplerClient
- *
- * @return reference to JavaSamplerClient
- */
-
- public JavaSamplerClient retrieveJavaClient() {
- return javaClient;
- }
-
- /**
- * Provide default setupTest() implementation for error conditions.
- * @see JavaSamplerClient#setupTest()
- */
- public void setupTest(HashMap arguments) {
- log.debug(whoAmI() + "\tsetupTest");
- }
-
- /**
- * Provide default teardownTest() implementation for error conditions.
- * @see JavaSamplerClient#teardownTest()
- */
- public void teardownTest(HashMap arguments) {
- log.debug(whoAmI() + "\tteardownTest");
- javaClient = null;
- }
-
- /**
- * Return SampleResult with data on error.
- * @see JavaSamplerClient#runTest()
- */
- public SampleResult runTest(HashMap arguments) {
- log.debug(whoAmI() + "\trunTest");
- Thread.yield();
- SampleResult results = new SampleResult();
- results.setTime(0);
- results.setSuccessful(false);
- results.setResponseData(new String("Class not found: " +
getClassname()).getBytes());
- results.setSampleLabel("ERROR: " + getClassname());
- return results;
- }
-
- private String whoAmI() {
- StringBuffer sb = new StringBuffer();
- sb.append(Thread.currentThread().getName());
- sb.append("@");
- sb.append(Integer.toHexString(hashCode()));
- return sb.toString();
- }
+ /**
+ * Property key representing the classname of the JavaSamplerClient to use.
+ */
+ public final static String CLASSNAME = "classname";
+
+ /**
+ * Property key representing the arguments for the JavaSamplerClient.
+ */
+ public final static String ARGUMENTS = "arguments";
+
+ /**
+ * The JavaSamplerClient instance used by this sampler to actually perform
+ * the sample.
+ */
+ transient private JavaSamplerClient javaClient = null;
+
+ /**
+ * Logging
+ */
+ transient private static Logger log =
Hierarchy.getDefaultHierarchy().getLoggerFor("jmeter.protocol.java");
+
+ /**
+ * Set used to register all active JavaSamplers. This is used so that the
+ * samplers can be notified when the test ends.
+ */
+ private static Set allSamplers = new HashSet();
+
+ /**
+ * Create a JavaSampler.
+ */
+ public JavaSampler()
+ {
+ setArguments(new Arguments());
+ synchronized (allSamplers)
+ {
+ allSamplers.add(this);
+ }
+ }
+
+ /**
+ * Set the arguments (parameters) for the JavaSamplerClient to be executed
+ * with.
+ *
+ * @param args the new arguments. These replace any existing arguments.
+ */
+ public void setArguments(Arguments args)
+ {
+ this.setProperty(ARGUMENTS, args);
+ }
+
+ /**
+ * Get the arguments (parameters) for the JavaSamplerClient to be executed
+ * with.
+ *
+ * @return the arguments
+ */
+ public Arguments getArguments()
+ {
+ return (Arguments) getProperty(ARGUMENTS);
+ }
+
+ public void addCustomTestElement(TestElement el)
+ {
+ if (el instanceof JavaConfig)
+ {
+ mergeIn(el);
+ }
+ }
+
+ /**
+ * Releases Java Client.
+ */
+ public void releaseJavaClient()
+ {
+ if (javaClient != null)
+ {
+ javaClient.teardownTest(createArgumentsHashMap(getArguments()));
+ }
+ javaClient = null;
+ }
+
+ /**
+ * Sets the Classname attribute of the JavaConfig object
+ *
+ [EMAIL PROTECTED] classname The new Classname value
+ */
+ public void setClassname(String classname)
+ {
+ this.setProperty(CLASSNAME, classname);
+ }
+
+ /**
+ * Gets the Classname attribute of the JavaConfig object
+ *
+ [EMAIL PROTECTED] The Classname value
+ */
+ public String getClassname()
+ {
+ return this.getPropertyAsString(CLASSNAME);
+ }
+
+ /**
+ * Performs a test sample.
+ *
+ * The <code>sample()</code> method retrieves the reference to the
+ * Java client and calls its <code>runTest()</code> method.
+ * @see Sampler#sample()
+ * @see JavaSamplerClient#runTest()
+ *
+ * @return test SampleResult
+ */
+ public SampleResult sample(Entry entry)
+ {
+ if (javaClient == null)
+ {
+ log.debug(whoAmI() + "Creating Java Client");
+ createJavaClient();
+ javaClient.setupTest(createArgumentsHashMap(getArguments()));
+ }
+ return javaClient.runTest(createArgumentsHashMap(getArguments()));
+ }
+
+ /**
+ * Convert the arguments parameter into a HashMap which can be passed
+ * to the JavaSamplerClient. The "name" of each Argument
+ * in args is added to the HashMap as a key, and the "value"
+ * is the value associated with that key. The original Arguments are
+ * not modified.
+ *
+ * @param args the Arguments to convert to a HashMap
+ * @return a HashMap representation of the Arguments
+ */
+ public HashMap createArgumentsHashMap(Arguments args)
+ {
+ HashMap newArgs = new HashMap();
+ Iterator iter = args.iterator();
+ while (iter.hasNext())
+ {
+ Argument item = (Argument) iter.next();
+ newArgs.put(item.getName(), item.getValue());
+ }
+
+ return newArgs;
+ }
+
+ /**
+ * Returns reference to <code>JavaSamplerClient</code>.
+ *
+ * The <code>createJavaClient()</code> method uses reflection
+ * to create an instance of the specified Java protocol client.
+ * If the class can not be found, the method returns a reference
+ * to <code>this</code> object.
+ *
+ * @return JavaSamplerClient reference.
+ */
+ public JavaSamplerClient createJavaClient()
+ {
+ if (javaClient == null)
+ {
+ try
+ {
+ Class javaClass = Class.forName(getClassname().trim(), false,
Thread.currentThread().getContextClassLoader());
+ javaClient = (JavaSamplerClient) javaClass.newInstance();
+
+ if (log.isDebugEnabled())
+ {
+ log.debug(whoAmI() + "\tCreated:\t" + getClassname() + "@" +
Integer.toHexString(javaClient.hashCode()));
+ }
+ }
+ catch (Exception e)
+ {
+ log.error(whoAmI() + "\tException creating: " + getClassname(), e);
+ javaClient = this;
+ }
+ }
+ return javaClient;
+ }
+
+ /**
+ * Retrieves reference to JavaSamplerClient.
+ *
+ * Convience method used to check for null reference without
+ * actually creating a JavaSamplerClient
+ *
+ * @return reference to JavaSamplerClient
+ */
+ public JavaSamplerClient retrieveJavaClient()
+ {
+ return javaClient;
+ }
+
+ /**
+ * Provide default setupTest() implementation for error conditions.
+ * @see JavaSamplerClient#setupTest()
+ */
+ public void setupTest(HashMap arguments)
+ {
+ log.debug(whoAmI() + "\tsetupTest");
+ }
+
+ /**
+ * Provide default teardownTest() implementation for error conditions.
+ * @see JavaSamplerClient#teardownTest()
+ */
+ public void teardownTest(HashMap arguments)
+ {
+ log.debug(whoAmI() + "\tteardownTest");
+ javaClient = null;
+ }
+
+ /**
+ * Return SampleResult with data on error.
+ * @see JavaSamplerClient#runTest()
+ */
+ public SampleResult runTest(HashMap arguments)
+ {
+ log.debug(whoAmI() + "\trunTest");
+ Thread.yield();
+ SampleResult results = new SampleResult();
+ results.setTime(0);
+ results.setSuccessful(false);
+ results.setResponseData(new String("Class not found: " +
getClassname()).getBytes());
+ results.setSampleLabel("ERROR: " + getClassname());
+ return results;
+ }
+
+ private String whoAmI()
+ {
+ StringBuffer sb = new StringBuffer();
+ sb.append(Thread.currentThread().getName());
+ sb.append("@");
+ sb.append(Integer.toHexString(hashCode()));
+ return sb.toString();
+ }
+
+ // TestListener implementation
+
+ public void testStarted()
+ {
+ log.debug(whoAmI() + "\ttestStarted");
+ }
+
+ public void testStarted(String host)
+ {
+ log.debug(whoAmI() + "\ttestStarted(" + host + ")");
+ }
+
+ /**
+ * Method called at the end of the test. This is called only on one
+ * instance of JavaSampler. This method will loop throuh all of the
+ * other JavaSamplers which have been registered (automatically in the
+ * constructor) and notify them that the test has ended, allowing the
+ * JavaSamplerClients to cleanup.
+ */
+ public void testEnded()
+ {
+ log.debug(whoAmI() + "\ttestEnded");
+ synchronized (allSamplers)
+ {
+ Iterator i = allSamplers.iterator();
+ while (i.hasNext())
+ {
+ JavaSampler sampler = (JavaSampler) i.next();
+ sampler.releaseJavaClient();
+ i.remove();
+ }
+ }
+ }
+
+ public void testEnded(String host)
+ {
+ testEnded();
+ }
}
---------------------------------------------------------------------
To unsubscribe, e-mail: [EMAIL PROTECTED]
For additional commands, e-mail: [EMAIL PROTECTED]
