本文描述了对ActiveMQ进行性能测试的代码。性能测试用源代码共包含3个文件,分别是:
JMS消息发送类:ActiveMQProducer.java
JMS消息接收类:ActiveMQConsumer.java
JMS消息收发测试主类:ActiveMQTest.java
下面分别介绍这三个类。
1. JMS消息发送类 ActiveMQProducer.java 的源码如下:
/**
* <p>Title: </p>
* <p>Description: </p>
* <p>Copyright: Copyright (c) 2007</p>
* <p>Company: </p>
* @author mqboss
* @version 1.0
*/
import javax.jms.*;
import org.apache.activemq.*;
public class ActiveMQProducer {
public final static int MAX_SEND_TIMES = 100;
private String user = ActiveMQConnection.DEFAULT_USER;
private String password = ActiveMQConnection.DEFAULT_PASSWORD;
private String url = ActiveMQConnection.DEFAULT_BROKER_URL;
private String subject = "TOOL.DEFAULT";
private Destination destination = null;
private Connection connection = null;
private Session session = null;
private MessageProducer producer = null;
/**
* 初始化
*
* @throws JMSException
* @throws java.lang.Exception
*/
private void initialize() throws JMSException, Exception {
ActiveMQConnectionFactory connectionFactory = new ActiveMQConnectionFactory(
user, password, url);
connection = connectionFactory.createConnection();
session = connection.createSession(false, Session.AUTO_ACKNOWLEDGE);
destination = session.createQueue(subject);
producer = session.createProducer(destination);
producer.setDeliveryMode(DeliveryMode.NON_PERSISTENT);
connection.start();
}
/**
* 发送消息
*
* @param message
* @throws JMSException
* @throws java.lang.Exception
*/
public void produceMessage(String message) throws JMSException, Exception {
initialize();
TextMessage msg = session.createTextMessage(message);
long beginTime = System.currentTimeMillis();
System.out.println("Producer:->Sending message: ");
for (int i = 0; i < MAX_SEND_TIMES; i++) {
producer.send(msg);
if ((i + 1) % 1000 == 0) {
System.out.println("This is the " + i + " message!");
}
}
System.out.println("Producer:->Message sent complete!");
long endTime = System.currentTimeMillis();
long executeTime = endTime - beginTime;
System.out.println("ActiveMQ send" + MAX_SEND_TIMES + " messages used: "
+ executeTime + " ms");
}
/**
* 关闭连接
*
* @throws JMSException
*/
public void close() throws JMSException {
System.out.println("Producer:->Closing connection");
if (producer != null) {
producer.close();
}
if (session != null) {
session.close();
}
if (connection != null) {
connection.close();
}
}
}
2. JMS消息接收类 ActiveMQConsumer.java 的源码如下
/**
* <p>Title: </p>
* <p>Description: </p>
* <p>Copyright: Copyright (c) 2007</p>
* <p>Company: </p>
* @author mqboss
* @version 1.0
*/
import javax.jms.*;
import org.apache.activemq.*;
public class ActiveMQConsumer implements MessageListener {
public static int RECEIVED_MSG_NUM = 0;
long beginReceiveTime = 0;
long endReceiveTime = 0;
long receiveDuringTime = 0;
private String user = ActiveMQConnection.DEFAULT_USER;
private String password = ActiveMQConnection.DEFAULT_PASSWORD;
private String url = ActiveMQConnection.DEFAULT_BROKER_URL;
private String subject = "TOOL.DEFAULT";
private Destination destination = null;
private Connection connection = null;
private Session session = null;
private MessageConsumer consumer = null;
/**
* 初始化
*
* @throws JMSException
* @throws java.lang.Exception
*/
private void initialize() throws JMSException, Exception {
ActiveMQConnectionFactory connectionFactory = new ActiveMQConnectionFactory(
user, password, url);
connection = connectionFactory.createConnection();
session = connection.createSession(false, Session.AUTO_ACKNOWLEDGE);
destination = session.createQueue(subject);
consumer = session.createConsumer(destination);
}
/**
* 接收消息
*
* @throws JMSException
* @throws java.lang.Exception
*/
public void consumeMessage() throws JMSException, Exception {
initialize();
connection.start();
System.out.println("Consumer:->Begin listening...");
// 设置消息监听
consumer.setMessageListener(this);
}
/**
* 关闭连接
*
* @throws JMSException
*/
public void close() throws JMSException {
System.out.println("Consumer:->Closing connection");
if (consumer != null) {
consumer.close();
}
if (session != null) {
session.close();
}
if (connection != null) {
connection.close();
}
}
/**
* 收到消息的处理
*
* @param message
*/
public void onMessage(Message message) {
try {
if (message instanceof TextMessage) {
TextMessage txtMsg = (TextMessage) message;
String msg = txtMsg.getText();
// receive the first message
if (RECEIVED_MSG_NUM == 0) {
beginReceiveTime = System.currentTimeMillis();
}
RECEIVED_MSG_NUM++;
// print one String when received 1000 message
if ((RECEIVED_MSG_NUM + 1) % 1000 == 0) {
System.out.println("Consumer:->Received: "
+ RECEIVED_MSG_NUM);
}
// receive the last message
if (RECEIVED_MSG_NUM == ActiveMQProducer.MAX_SEND_TIMES - 1) {
endReceiveTime = System.currentTimeMillis();
receiveDuringTime = endReceiveTime - beginReceiveTime;
System.out.println("ActiveMQ Receive "
+ ActiveMQProducer.MAX_SEND_TIMES
+ " messages used: " + receiveDuringTime + " ms");
}
} else {
System.out.println(System.currentTimeMillis()
+ "Consumer:->Received: " + message);
}
} catch (JMSException e) {
e.printStackTrace();
}
}
}
3. JMS消息收发测试主类:ActiveMQTest.java
/**
* <p>Title: </p>
* <p>Description: </p>
* <p>Copyright: Copyright (c) 2007</p>
* <p>Company: </p>
* @author mqboss
* @version 1.0
*/
import javax.jms.*;
public class ActiveMQTest {
/**
*
* @param args
* @throws JMSException
* @throws java.lang.Exception
*/
public static void main(String[] args) throws JMSException, Exception {
ActiveMQConsumer consumer = new ActiveMQConsumer();
ActiveMQProducer producer = new ActiveMQProducer();
char[] tempChars=new char[1024];
for(int i=0;i<1024;i++){
tempChars[i]='a';
}
String tempMsg=String.valueOf(tempChars);
// 开始监听
consumer.consumeMessage();
producer.produceMessage(tempMsg);
producer.close();
// 延时5000ms后关闭连接
Thread.sleep(5000);
consumer.close();
}
}
为了与JBossMQ进行性能对比测试,下面把JBossMQ收发消息的测试代码也一并附上。
为了保证测试代码的一致性,JBossMQ性能测试的代码也包含3个文件,分别是:
JMS消息发送类:JBossMQProducer.java
JMS消息接收类:JBossMQConsumer.java
JMS消息收发测试主类:JBossMQTest.java
下面分别介绍这三个类。
1. JMS消息发送类 JBossMQProducer.java 的源码如下:
/**
* <p>Title: </p>
* <p>Description: </p>
* <p>Copyright: Copyright (c) 2007</p>
* <p>Company: </p>
* @author mqboss
* @version 1.0
*/
import java.util.Properties;
import javax.jms.*;
import javax.naming.Context;
import javax.naming.InitialContext;
public class JbossMQProducer {
public final static int MAX_SEND_TIMES = 100;
private Destination destination = null;
private Connection connection = null;
private Session session = null;
private MessageProducer producer = null;
/**
* 初始化
*
* @throws JMSException
* @throws java.lang.Exception
*/
private void initialize() throws JMSException, Exception {
Properties props = new Properties();
props.put(Context.INITIAL_CONTEXT_FACTORY,
"org.jnp.interfaces.NamingContextFactory");
props.put(Context.URL_PKG_PREFIXES, "org.jboss.naming");
props.put(Context.PROVIDER_URL, "localhost:1099");
InitialContext jmsContext = new InitialContext(props);
ConnectionFactory connectionFactory = (ConnectionFactory) jmsContext
.lookup("ConnectionFactory");
connection = connectionFactory.createConnection();
session = connection.createSession(false, Session.AUTO_ACKNOWLEDGE);
destination = (Queue) jmsContext.lookup("queue/A");
producer = session.createProducer(destination);
producer.setDeliveryMode(DeliveryMode.NON_PERSISTENT);
connection.start();
}
/**
* 发送消息
*
* @param message
* @throws JMSException
* @throws java.lang.Exception
*/
public void produceMessage(String message) throws JMSException, Exception {
initialize();
TextMessage msg = session.createTextMessage(message);
long beginTime = System.currentTimeMillis();
System.out.println("Producer:->Sending message: ");
for (int i = 0; i < MAX_SEND_TIMES; i++) {
if ((i + 1) % 1000 == 0) {
System.out.println("This is the " + i + " message!");
}
producer.send(msg);
}
System.out.println("Producer:->Message sent complete!");
long endTime = System.currentTimeMillis();
long executeTime = endTime - beginTime;
System.out.println("JbossMQ send" + MAX_SEND_TIMES + " messages used: "
+ executeTime + " ms");
}
/**
* 关闭连接
*
* @throws JMSException
*/
public void close() throws JMSException {
System.out.println("Producer:->Closing connection");
if (producer != null) {
producer.close();
}
if (session != null) {
session.close();
}
if (connection != null) {
connection.close();
}
}
}
2. JMS消息接收类 JBossMQConsumer.java 的源码如下
/**
* <p>Title: </p>
* <p>Description: </p>
* <p>Copyright: Copyright (c) 2007</p>
* <p>Company: </p>
* @author mqboss
* @version 1.0
*/
import java.util.Properties;
import javax.jms.*;
import javax.naming.*;
public class JbossMQConsumer implements MessageListener {
public static int RECEIVED_MSG_NUM = 0;
long beginReceiveTime = 0;
long endReceiveTime = 0;
long receiveDuringTime = 0;
private Destination destination = null;
private Connection connection = null;
private Session session = null;
private MessageConsumer consumer = null;
/**
* 初始化
*
* @throws JMSException
* @throws java.lang.Exception
*/
private void initialize() throws JMSException, Exception {
Properties props = new Properties();
props.put(Context.INITIAL_CONTEXT_FACTORY,
"org.jnp.interfaces.NamingContextFactory");
props.put(Context.URL_PKG_PREFIXES, "org.jboss.naming");
props.put(Context.PROVIDER_URL, "localhost:1099");
// get naming context
InitialContext jmsContext = new InitialContext(props);
ConnectionFactory connectionFactory = (ConnectionFactory) jmsContext
.lookup("ConnectionFactory");
connection = connectionFactory.createConnection();
session = connection.createSession(false, Session.AUTO_ACKNOWLEDGE);
destination = (Queue) jmsContext.lookup("queue/A");
consumer = session.createConsumer(destination);
}
/**
* 接收消息
*
* @throws JMSException
* @throws java.lang.Exception
*/
public void consumeMessage() throws JMSException, Exception {
initialize();
connection.start();
System.out.println("Consumer:->Begin listening...");
// 设置监听
consumer.setMessageListener(this);
}
/**
* 关闭连接
*
* @throws JMSException
*/
public void close() throws JMSException {
System.out.println("Consumer:->Closing connection");
if (consumer != null) {
consumer.close();
}
if (session != null) {
session.close();
}
if (connection != null) {
connection.close();
}
}
/**
* 处理消息
*
* @param message
*/
public void onMessage(Message message) {
try {
if (message instanceof TextMessage) {
TextMessage txtMsg = (TextMessage) message;
String msg = txtMsg.getText();
// receive the first message
if (RECEIVED_MSG_NUM == 0) {
beginReceiveTime = System.currentTimeMillis();
}
RECEIVED_MSG_NUM++;
//print one String when received 1000 message
if ((RECEIVED_MSG_NUM + 1) % 1000 == 0) {
System.out.println("Consumer:->Received: "
+ RECEIVED_MSG_NUM);
}
// receive the last message
if (RECEIVED_MSG_NUM == JbossMQProducer.MAX_SEND_TIMES - 1) {
endReceiveTime = System.currentTimeMillis();
receiveDuringTime = endReceiveTime - beginReceiveTime;
System.out.println("JbossMQ receive "
+ JbossMQProducer.MAX_SEND_TIMES
+ " messages used: " + receiveDuringTime + " ms");
}
} else {
System.out.println(System.currentTimeMillis()
+ "Consumer:->Received: " + message);
}
} catch (JMSException e) {
e.printStackTrace();
}
}
}
3. JMS消息收发测试主类:JBossMQTest.java
import javax.jms.JMSException;
/**
* <p>Title: </p>
* <p>Description: </p>
* <p>Copyright: Copyright (c) 2007</p>
* <p>Company: </p>
* @author mqboss
* @version 1.0
*/
public class JbossMQTest {
/**
*
* @param args
* @throws JMSException
* @throws java.lang.Exception
*/
public static void main(String[] args) throws JMSException, Exception {
JbossMQConsumer consumer = new JbossMQConsumer();
JbossMQProducer producer = new JbossMQProducer();
char[] tempChars = new char[1024];
for (int i = 0; i < 1024; i++) {
tempChars[i] = 'a';
}
String tempMsg = String.valueOf(tempChars);
// 启动消息监听
consumer.consumeMessage();
producer.produceMessage(tempMsg);
producer.close();
// 5000ms后关闭连接
Thread.sleep(5000);
consumer.close();
}
}
以上是把ActiveMQ集成到JBoss以后,对ActiveMQ和JBossMQ进行性能对比测试的源代码。
很显然,以上源代码基本一致,所以性能测试结果可以很好的说明ActiveMQ和JBossMQ的性能对比情况。
参考文献:
[1]ActiveMQ的一个简单示例