/*
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
*
* Copyright (c) 2002-2017 Oracle and/or its affiliates. All rights reserved.
*
* The contents of this file are subject to the terms of either the GNU
* General Public License Version 2 only ("GPL") or the Common Development
* and Distribution License("CDDL") (collectively, the "License"). You
* may not use this file except in compliance with the License. You can
* obtain a copy of the License at
* https://oss.oracle.com/licenses/CDDL+GPL-1.1
* or LICENSE.txt. See the License for the specific
* language governing permissions and limitations under the License.
*
* When distributing the software, include this License Header Notice in each
* file and include the License file at LICENSE.txt.
*
* GPL Classpath Exception:
* Oracle designates this particular file as subject to the "Classpath"
* exception as provided by Oracle in the GPL Version 2 section of the License
* file that accompanied this code.
*
* Modifications:
* If applicable, add the following below the License Header, with the fields
* enclosed by brackets [] replaced by your own identifying information:
* "Portions Copyright [year] [name of copyright owner]"
*
* Contributor(s):
* If you wish your version of this file to be governed by only the CDDL or
* only the GPL Version 2, indicate your decision by adding "[Contributor]
* elects to include this software in this distribution under the [CDDL or GPL
* Version 2] license." If you don't indicate a single choice of license, a
* recipient has the option to distribute your version of this file under
* either the CDDL, the GPL Version 2 or to extend the choice of license to
* its licensees as provided above. However, if you add GPL Version 2 code
* and therefore, elected the GPL Version 2 license, then the option applies
* only if the new code is made subject to such option by the copyright
* holder.
*/
package com.sun.s1asdev.ejb.mdb.msgbean.client;
import java.io.*;
import java.util.*;
import javax.ejb.EJBHome;
import javax.naming.*;
import javax.rmi.PortableRemoteObject;
import org.omg.CORBA.ORB;
import java.sql.*;
import javax.sql.*;
import javax.jms.*;
import com.sun.ejte.ccl.reporter.SimpleReporterAdapter;
public class HelloClient {
private static SimpleReporterAdapter stat =
new SimpleReporterAdapter("appserv-tests");
// in milli-seconds
private static long TIMEOUT = 90000;
private String[] args;
public static void main (String[] args) {
HelloClient client = new HelloClient(args);
stat.addDescription("ejb-mdb-msgbean");
client.doTest();
stat.printSummary("ejb-mdb-msgbeanID");
int exitCode = 0;
System.exit(exitCode);
}
private InitialContext context;
private QueueConnection queueCon;
private QueueSession queueSession;
private QueueSender queueSender;
private QueueReceiver queueReceiver;
private javax.jms.Queue clientQueue;
private TopicConnection topicCon;
private TopicSession topicSession;
private TopicPublisher topicPublisher;
private TopicSubscriber topicSubscriber;
private Topic clientTopic;
public HelloClient(String[] theArgs) {
this.args = theArgs;
if( this.args.length == 0 ) {
this.args = new String[] { "all", "5" };
}
}
private static Boolean BFALSE = new Boolean(false);
private static Boolean BTRUE = new Boolean(true);
Object[][] userSpecifiedTxTest = { {"", // destname
BFALSE, // reply
BFALSE, // jdbc
BFALSE } }; // rollback
Object[][] onePhaseCommitTests = {
{ "jms/queue_cmt", BFALSE, BFALSE, BFALSE },
{ "jms/topic_cmt", BFALSE, BFALSE, BFALSE },
{ "jms/queue_bmt", BTRUE, BFALSE, BFALSE },
{ "jms/topic_bmt", BTRUE, BFALSE, BFALSE },
{ "jms/queue_bmt", BFALSE, BTRUE, BFALSE },
{ "jms/topic_bmt", BFALSE, BTRUE, BFALSE },
};
Object[][] twoPhaseCommitTests = {
{ "jms/queue_cmt", BFALSE, BTRUE, BFALSE },
{ "jms/topic_cmt", BFALSE, BTRUE, BFALSE },
{ "jms/queue_cmt", BTRUE, BFALSE, BFALSE },
{ "jms/topic_cmt", BTRUE, BFALSE, BFALSE },
{ "jms/queue_bmt", BTRUE, BTRUE, BFALSE },
{ "jms/topic_bmt", BTRUE, BTRUE, BFALSE },
{ "jms/queue_cmt", BTRUE, BTRUE, BFALSE },
{ "jms/topic_cmt", BTRUE, BTRUE, BFALSE },
};
Object[][] rollbackTests = {
{ "jms/queue_cmt", BTRUE, BFALSE, BTRUE },
{ "jms/topic_cmt", BTRUE, BFALSE, BTRUE },
{ "jms/queue_cmt", BTRUE, BTRUE, BTRUE },
{ "jms/topic_cmt", BTRUE, BTRUE, BTRUE },
{ "jms/queue_cmt", BFALSE, BFALSE, BTRUE },
{ "jms/topic_cmt", BFALSE, BFALSE, BTRUE },
{ "jms/queue_cmt", BTRUE, BFALSE, BTRUE },
{ "jms/topic_cmt", BTRUE, BFALSE, BTRUE },
};
Object[][] localTxTests = {
{ "jms/queue_cmtns", BFALSE, BFALSE, BFALSE },
{ "jms/queue_cmtns", BFALSE, BTRUE, BFALSE },
{ "jms/queue_cmtns", BTRUE, BFALSE, BFALSE },
{ "jms/queue_cmtns", BTRUE, BTRUE, BFALSE }
};
public void doTest() {
int numIter = 10;
if(args.length == 0 || (args.length > 0 &&
args[0].equals("1pc" )) ) {
System.out.println("Running 1 phase commit msgbean tests");
numIter = (args.length >= 2) ?
Integer.parseInt(args[1]) : numIter;
doTests(onePhaseCommitTests, numIter);
} else if( args[0].equals("2pc") ) {
System.out.println("Running 2 phase commit msgbean tests");
numIter = Integer.parseInt(args[1]);
doTests(twoPhaseCommitTests, numIter);
} else if( args[0].equals("localtx") ) {
System.out.println("Running localtx msgbean tests");
numIter = Integer.parseInt(args[1]);
doTests(localTxTests, numIter);
} else if( args[0].equals("rollback") ) {
System.out.println("Running rollback msgbean tests");
numIter = Integer.parseInt(args[1]);
doTests(rollbackTests, numIter);
} else if( args[0].equals("all") ) {
System.out.println("Running all msgbean tests");
numIter = Integer.parseInt(args[1]);
doTests(onePhaseCommitTests, numIter);
doTests(twoPhaseCommitTests, numIter);
doTests(localTxTests, numIter);
doTests(rollbackTests, numIter);
} else {
userSpecifiedTxTest[0][0] = args[0];
numIter = Integer.parseInt(args[1]);
System.out.println("Running " + userSpecifiedTxTest[0][0]);
System.out.println("Num iterations = " + numIter);
if( args.length == 3 ) {
userSpecifiedTxTest[0][1] =
new Boolean(args[2].indexOf("reply") != -1);
userSpecifiedTxTest[0][2] =
new Boolean((args[2].indexOf("jdbc") != -1));
userSpecifiedTxTest[0][3] =
new Boolean((args[2].indexOf("rollback") != -1));
}
doTests(userSpecifiedTxTest, numIter);
}
return;
}
private void doTests(Object[][] tests, int numIter) {
if (numIter > 20) {
TIMEOUT = TIMEOUT * (int)(numIter / 10);
}
for(int i = 0; i < tests.length; i++) {
String destName = "java:comp/env/" + (String) tests[i][0];
boolean expectReply = ((Boolean)tests[i][1]).booleanValue();
boolean jdbc = ((Boolean)tests[i][2]).booleanValue();
boolean rollback = ((Boolean)tests[i][3]).booleanValue();
String test = "msgbean " + generateTestName(destName,jdbc,
expectReply, rollback);
try {
setup();
doTest(destName, numIter, expectReply, jdbc, rollback);
stat.addStatus(test, stat.PASS);
} catch(Throwable t) {
System.out.println("Caught unexpected exception " + t);
t.printStackTrace();
stat.addStatus(test, stat.FAIL);
break;
} finally {
cleanup();
}
}
return;
}
public void setup() throws Exception {
context = new InitialContext();
QueueConnectionFactory queueConFactory =
(QueueConnectionFactory) context.lookup
("java:comp/env/jms/QCF");
queueCon = queueConFactory.createQueueConnection();
queueSession = queueCon.createQueueSession
(false, Session.AUTO_ACKNOWLEDGE);
// Producer will be specified when actual msg is sent.
queueSender = queueSession.createSender(null);
clientQueue = (javax.jms.Queue)
context.lookup("java:comp/env/jms/client_queue");
queueReceiver = queueSession.createReceiver(clientQueue);
TopicConnectionFactory topicConFactory =
(TopicConnectionFactory) context.lookup
("java:comp/env/jms/TCF");
topicCon = topicConFactory.createTopicConnection();
topicSession =
topicCon.createTopicSession(false, Session.AUTO_ACKNOWLEDGE);
// Producer will be specified when actual msg is published.
topicPublisher = topicSession.createPublisher(null);
clientTopic = (Topic)
context.lookup("java:comp/env/jms/client_topic");
topicSubscriber = topicSession.createSubscriber(clientTopic);
queueCon.start();
topicCon.start();
}
public void cleanup() {
try {
if( queueCon != null ) {
queueCon.close();
}
} catch(Throwable t) {
t.printStackTrace();
}
try {
if( topicCon != null ) {
topicCon.close();
}
} catch(Throwable t) {
t.printStackTrace();
}
}
public void sendMsgs(javax.jms.Queue queue, Message msg, int num)
throws JMSException {
for(int i = 0; i < num; i++) {
System.out.println("Sending message " + i + " to " + queue);
queueSender.send(queue, msg);
}
}
public void publishMsgs(Topic topic, Message msg, int num)
throws JMSException {
for(int i = 0; i < num; i++) {
System.out.println("Publishing message " + i + " to " + topic);
topicPublisher.publish(topic, msg);
}
}
public void recvQueueMsgs(int num) throws JMSException {
for(int i = 0; i < num; i++) {
System.out.println("Waiting for queue message " + (i+1) + " of " + num);
Message recvdmessage = queueReceiver.receive(TIMEOUT);
if( recvdmessage != null ) {
System.out.println("Received message : " +
((TextMessage)recvdmessage).getText());
} else {
System.out.println("timeout after " + TIMEOUT + " seconds");
throw new JMSException("timeout" + TIMEOUT + " seconds");
}
}
}
public void recvTopicMsgs(int num) throws JMSException {
for(int i = 0; i < num; i++) {
System.out.println("Waiting for topic message " + (i+1) + " of " + num);
Message recvdmessage = topicSubscriber.receive(TIMEOUT);
if( recvdmessage != null ) {
System.out.println("Received message : " +
((TextMessage)recvdmessage).getText());
} else {
System.out.println("timeout after " + TIMEOUT + " seconds");
throw new JMSException("timeout" + TIMEOUT + " seconds");
}
}
}
/**
* Generate testname. (Try to keep synched with target names in Makefile.)
*/
private String generateTestName(String destName, boolean doJdbc,
boolean expectReply, boolean rollbackEnabled)
{
String testName = destName.toLowerCase();
if (doJdbc) {
testName = testName + "_jdbc";
} else {
testName = testName + "_nojdbc";
}
if (expectReply) {
testName = testName + "_reply";
} else {
testName = testName + "_noreply";
}
if (rollbackEnabled) {
testName = testName + "_rollback";
}
return testName;
}
public void doTest(String destName, int numIter, boolean expectReply,
boolean doJdbc, boolean rollbackEnabled)
throws Exception
{
String testName =
generateTestName(destName, doJdbc, expectReply, rollbackEnabled);
try {
Destination dest = (Destination) context.lookup(destName);
boolean pointToPoint = dest instanceof javax.jms.Queue;
System.out.println("Beginning test : " + testName);
Timer.start();
if( pointToPoint ) {
Message message =
queueSession.createTextMessage(destName);
message.setBooleanProperty("doJdbc", doJdbc);
message.setBooleanProperty("rollbackEnabled", rollbackEnabled);
/* aid identifying what test message is associated with on JMS server side.
* (JMS RI Message Implementation toString() will print this property if it
* is set.)
*/
message.setStringProperty("COM_SUN_JMS_TESTNAME", testName + "_#" + numIter);
if( expectReply ) {
message.setJMSReplyTo(clientQueue);
}
sendMsgs((javax.jms.Queue) dest, message, numIter);
} else {
Message message =
topicSession.createTextMessage(destName);
message.setBooleanProperty("doJdbc", doJdbc);
message.setBooleanProperty("rollbackEnabled", rollbackEnabled);
/* aid identifying what test message is associated with on JMS server side.
* (JMS RI Message Implementation toString() will print this property if it
* is set.)
*/
message.setStringProperty("COM_SUN_JMS_TESTNAME", testName + "_numIter");
if( expectReply ) {
message.setJMSReplyTo(clientTopic);
}
publishMsgs((Topic) dest, message, numIter);
}
if( expectReply ) {
if( pointToPoint ) {
recvQueueMsgs(numIter);
} else {
recvTopicMsgs(numIter);
}
}
long time = Timer.stop();
System.out.println("End test : " + testName + ". Time = " +
Timer.format(time) + " -> " + (time/numIter) +
" msec/msg (" + numIter + " msgs)");
} catch (Exception e) {
System.out.println("Unexpected exception " + e + " in test " +
testName);
throw e;
}
}
// helper class for timing the tests
private static class Timer {
static long startTime;
public static void start() {
startTime = System.currentTimeMillis();
}
public static long stop() {
return System.currentTimeMillis() - startTime;
}
/**
* Returns a string that outputs time in an
* easily readable format.
*/
public static String format(long timeIn) {
double foo = ((double)timeIn)/1000;
long time = Math.round(foo);
String ret = null;
if (time < 60) {
ret = time + " seconds";
} else {
ret = (time/60) + " minutes, " +
(time%60) + " seconds (" +
time + " seconds)";
}
return ret;
}
}
}