// $Header: /home/cvs/jakarta-jmeter/src/protocol/java/org/apache/jmeter/protocol/java/sampler/BeanShellSampler.java,v 1.6.2.4 2004/10/08 23:20:10 sebb Exp $
/*
* Copyright 2003-2004 The Apache Software Foundation.
*
* Licensed 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.protocol.java.sampler;
import java.io.FileNotFoundException;
import java.io.IOException;
//import bsh.EvalError;
import bsh.Interpreter;
import org.apache.jmeter.samplers.AbstractSampler;
import org.apache.jmeter.samplers.Entry;
import org.apache.jmeter.samplers.SampleResult;
import org.apache.jmeter.util.JMeterUtils;
import org.apache.jorphan.logging.LoggingManager;
import org.apache.jorphan.util.JOrphanUtils;
import org.apache.log.Logger;
/**
* A sampler which understands BeanShell
*
* @version $Revision: 1.6.2.4 $ Updated on: $Date: 2004/10/08 23:20:10 $
*/
public class BeanShellSampler extends AbstractSampler
{
private static final Logger log = LoggingManager.getLoggerForClass();
public static final String FILENAME = "BeanShellSampler.filename"; //$NON-NLS-1$
public static final String SCRIPT = "BeanShellSampler.query"; //$NON-NLS-1$
public static final String PARAMETERS = "BeanShellSampler.parameters"; //$NON-NLS-1$
public static final String INIT_FILE = "beanshell.sampler.init"; //$NON-NLS-1$
private Interpreter bshInterpreter;
public BeanShellSampler()
{
String init="";
try{
bshInterpreter = new Interpreter();
init = JMeterUtils.getPropDefault(INIT_FILE,null);
if (init != null)
{
try
{
bshInterpreter.source(init);
} catch (IOException e){
log.warn("Error processing init file "+init+" "+e);
} catch (Exception e){
log.warn("Error processing init file "+init+" "+e);
}
}
} catch (NoClassDefFoundError e){
bshInterpreter=null;
}
}
/**
* Returns a formatted string label describing this sampler
*
* @return a formatted string label describing this sampler
*/
public String getLabel()
{
return getName();
}
public String getScript()
{
return this.getPropertyAsString(SCRIPT);
}
public String getFilename()
{
return getPropertyAsString(FILENAME);
}
public String getParameters()
{
return getPropertyAsString(PARAMETERS);
}
public SampleResult sample(Entry e)// Entry tends to be ignored ...
{
//log.info(getLabel()+" "+getFilename());
SampleResult res = new SampleResult();
boolean isSuccessful = false;
res.setSampleLabel(getLabel());
res.sampleStart();
try
{
String request=getScript();
String fileName=getFilename();
if (fileName.length() == 0) {
res.setSamplerData(request);
} else {
res.setSamplerData(fileName);
}
// Has to be done after construction, otherwise fails serialisation check
bshInterpreter.set("log",log); //$NON-NLS-1$
bshInterpreter.set("Label",getLabel()); //$NON-NLS-1$
bshInterpreter.set("FileName",getFilename()); //$NON-NLS-1$
bshInterpreter.set("SampleResult",res); //$NON-NLS-1$
bshInterpreter.set("Parameters",getParameters());// as a single line//$NON-NLS-1$
bshInterpreter.set("bsh.args",JOrphanUtils.split(getParameters()," "));
// Set default values
bshInterpreter.set("ResponseCode","200"); //$NON-NLS-1$
bshInterpreter.set("ResponseMessage","OK");//$NON-NLS-1$
bshInterpreter.set("IsSuccess",true);//$NON-NLS-1$
Object bshOut;
if (fileName.length() == 0){
bshOut = bshInterpreter.eval(request);
} else {
bshOut = bshInterpreter.source(fileName);
}
String out;
if (bshOut == null) {// Script did not return anything...
out="";
} else {
out = bshOut.toString();
}
res.setResponseData(out.getBytes());
res.setDataType(SampleResult.TEXT);
res.setResponseCode(bshInterpreter.get("ResponseCode").toString());//$NON-NLS-1$
res.setResponseMessage(bshInterpreter.get("ResponseMessage").toString());//$NON-NLS-1$
isSuccessful = Boolean.valueOf(bshInterpreter.get("IsSuccess") //$NON-NLS-1$
.toString()).booleanValue();
}
/*
* To avoid class loading problems when bsh,jar is missing,
* we don't try to catch this error separately
* catch (bsh.EvalError ex)
{
log.debug("",ex);
res.setResponseCode("500");//$NON-NLS-1$
res.setResponseMessage(ex.toString());
}
*/
// but we do trap this error to make tests work better
catch(NoClassDefFoundError ex){
log.error("BeanShell Jar missing? "+ex.toString());
res.setResponseCode("501");//$NON-NLS-1$
res.setResponseMessage(ex.toString());
res.setStopThread(true); // No point continuing
}
catch (IOException ex)
{
log.warn(ex.toString());
res.setResponseCode("500");//$NON-NLS-1$
res.setResponseMessage(ex.toString());
}
catch (Exception ex) // Mainly for bsh.EvalError
{
log.warn(ex.toString());
res.setResponseCode("500");//$NON-NLS-1$
res.setResponseMessage(ex.toString());
}
res.sampleEnd();
// Set if we were successful or not
res.setSuccessful(isSuccessful);
return res;
}
}