/*
* 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.protocol.jms.sampler;
import org.apache.jorphan.io.TextFile;
import org.apache.jmeter.samplers.Entry;
import org.apache.jmeter.samplers.SampleResult;
import org.apache.jmeter.services.FileServer;
import org.apache.jmeter.testelement.TestListener;
import org.apache.jmeter.engine.event.LoopIterationEvent;
import org.apache.jmeter.protocol.jms.control.gui.JMSPublisherGui;
import org.apache.jmeter.protocol.jms.client.ClientPool;
import org.apache.jmeter.protocol.jms.client.Publisher;
import org.apache.jorphan.logging.LoggingManager;
import org.apache.log.Logger;
public class PublisherSampler extends BaseJMSSampler implements TestListener {
private static final Logger log = LoggingManager.getLoggerForClass();
//++ These are JMX file names and must not be changed
private static final String INPUT_FILE = "jms.input_file"; //$NON-NLS-1$
private static final String RANDOM_PATH = "jms.random_path"; //$NON-NLS-1$
private static final String TEXT_MSG = "jms.text_message"; //$NON-NLS-1$
private static final String CONFIG_CHOICE = "jms.config_choice"; //$NON-NLS-1$
private static final String MESSAGE_CHOICE = "jms.config_msg_type"; //$NON-NLS-1$
//--
private transient Publisher PUB = null;
private StringBuffer BUFFER = new StringBuffer();
private static final FileServer FSERVER = FileServer.getFileServer();
private String file_contents = null;
public PublisherSampler() {
}
/**
* the implementation calls testStarted() without any parameters.
*/
public void testStarted(String test) {
testStarted();
}
/**
* the implementation calls testEnded() without any parameters.
*/
public void testEnded(String test) {
testEnded();
}
/**
* endTest cleans up the client
*
* @see junit.framework.TestListener#endTest(junit.framework.Test)
*/
public void testEnded() {
log.info("PublisherSampler.testEnded called");
Thread.currentThread().interrupt();
this.PUB = null;
this.BUFFER.setLength(0);
this.BUFFER = null;
ClientPool.clearClient();
}
/**
* the implementation creates a new StringBuffer
*/
public void testStarted() {
this.BUFFER = new StringBuffer();
}
/**
* NO implementation provided for the sampler. It is necessary in this case.
*/
public void testIterationStart(LoopIterationEvent event) {
}
/**
* initialize the Publisher client.
*
*/
public synchronized void initClient() {
this.PUB = new Publisher(this.getUseJNDIPropertiesAsBoolean(), this.getJNDIInitialContextFactory(), this
.getProviderUrl(), this.getConnectionFactory(), this.getTopic(), this.getUseAuth(), this.getUsername(),
this.getPassword());
ClientPool.addClient(this.PUB);
log.info("PublisherSampler.initClient called");
}
/**
* The implementation calls sample() without any parameters
*/
public SampleResult sample(Entry e) {
return this.sample();
}
/**
* The implementation will publish n messages within a for loop. Once n
* messages are published, it sets the attributes of SampleResult.
*
* @return the populated sample result
*/
public SampleResult sample() {
SampleResult result = new SampleResult();
result.setSampleLabel(getName());
if (this.PUB == null) {
this.initClient();
}
int loop = this.getIterationCount();
if (this.PUB != null) {
result.sampleStart();
for (int idx = 0; idx < loop; idx++) {
String tmsg = this.getMessageContent();
this.PUB.publish(tmsg);
this.BUFFER.append(tmsg);
}
result.sampleEnd();
String content = this.BUFFER.toString();
result.setBytes(content.getBytes().length);
result.setResponseCode("message published successfully");
result.setResponseMessage(loop + " messages published");
result.setSuccessful(true);
result.setResponseData(content.getBytes());
result.setSampleCount(loop);
this.BUFFER.setLength(0);
}
return result;
}
/**
* Method will check the setting and get the contents for the message.
*
* @return the contents for the message
*/
public String getMessageContent() {
if (this.getConfigChoice().equals(JMSPublisherGui.use_file)) {
// in the case the test uses a file, we set it locally and
// prevent loading the file repeatedly
if (this.file_contents == null) {
this.file_contents = this.getFileContent(this.getInputFile());
}
return this.file_contents;
} else if (this.getConfigChoice().equals(JMSPublisherGui.use_random)) {
// Maybe we should consider creating a global cache for the
// random files to make JMeter more efficient.
String fname = FSERVER.getRandomFile(this.getRandomPath(), new String[] { ".txt", ".obj" })
.getAbsolutePath();
return this.getFileContent(fname);
} else {
return this.getTextMessage();
}
}
/**
* The implementation uses TextFile to load the contents of the file and
* returns a string.
*
* @param path
* @return the contents of the file
*/
public String getFileContent(String path) {
TextFile tf = new TextFile(path);
return tf.getText();
}
// ------------- get/set properties ----------------------//
/**
* set the config choice
*
* @param choice
*/
public void setConfigChoice(String choice) {
setProperty(CONFIG_CHOICE, choice);
}
/**
* return the config choice
*
*/
public String getConfigChoice() {
return getPropertyAsString(CONFIG_CHOICE);
}
/**
* set the source of the message
*
* @param choice
*/
public void setMessageChoice(String choice) {
setProperty(MESSAGE_CHOICE, choice);
}
/**
* return the source of the message
*
*/
public String getMessageChoice() {
return getPropertyAsString(MESSAGE_CHOICE);
}
/**
* set the input file for the publisher
*
* @param file
*/
public void setInputFile(String file) {
setProperty(INPUT_FILE, file);
}
/**
* return the path of the input file
*
*/
public String getInputFile() {
return getPropertyAsString(INPUT_FILE);
}
/**
* set the random path for the messages
*
* @param path
*/
public void setRandomPath(String path) {
setProperty(RANDOM_PATH, path);
}
/**
* return the random path for messages
*
*/
public String getRandomPath() {
return getPropertyAsString(RANDOM_PATH);
}
/**
* set the text for the message
*
* @param message
*/
public void setTextMessage(String message) {
setProperty(TEXT_MSG, message);
}
/**
* return the text for the message
*
*/
public String getTextMessage() {
return getPropertyAsString(TEXT_MSG);
}
}