/*
* Copyright 2009 The Fornax Project Team, including the original
* author or authors.
*
* 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.sculptor.framework.test.ejbtestbean.messaging;
import javax.annotation.PreDestroy;
import javax.annotation.Resource;
import javax.ejb.Stateless;
import javax.ejb.TransactionManagement;
import javax.ejb.TransactionManagementType;
import javax.jms.ConnectionFactory;
import javax.jms.Destination;
import javax.jms.JMSException;
import javax.jms.Message;
import javax.jms.MessageConsumer;
import javax.jms.Session;
import javax.jms.TemporaryQueue;
import javax.jms.TextMessage;
import org.sculptor.framework.consumer.MessageSender;
import org.sculptor.framework.consumer.MessageSenderImpl;
/**
* Used by AbstractOpenEJBTest, but can be used directly from application
* project (test.ejb-jar file) also. The 'openejb.deployments.classpath.include'
* will cause this EJB to be automatically discovered and deployed when OpenEJB
* boots up.
*/
@Stateless(name = "MessagingTestBean")
@TransactionManagement(TransactionManagementType.BEAN)
public class MessagingTestBean implements MessagingTestLocal {
@Resource
private ConnectionFactory connectionFactory;
private javax.jms.Connection connection;
@Override
public Destination sendMessage(Destination destination, String message) {
Session jmsSession = null;
try {
jmsSession = getJmsConnection().createSession(false, Session.AUTO_ACKNOWLEDGE);
TextMessage textMessage = jmsSession.createTextMessage(message);
return sendMessage(destination, textMessage);
} catch (JMSException e) {
throw new RuntimeException(e);
} finally {
if (jmsSession != null) {
try {
jmsSession.close();
} catch (Exception ignore) {
}
}
}
}
@Override
public Destination sendMessage(Destination destination, Message message) {
try {
Destination replyTo = message.getJMSReplyTo();
if (replyTo == null) {
replyTo = createTemporaryQueue();
message.setJMSReplyTo(replyTo);
}
getMessageSender().sendMessage(destination, message);
return replyTo;
} catch (JMSException e) {
throw new RuntimeException(e);
}
}
protected MessageSender getMessageSender() {
if (getJmsConnection() == null) {
throw new IllegalStateException("Need JMS connection to be able to send messages.");
}
return new MessageSenderImpl(getJmsConnection());
}
protected javax.jms.Connection getJmsConnection() {
try {
if (connection == null) {
connection = connectionFactory.createConnection();
connection.start();
}
return connection;
} catch (JMSException e) {
throw new RuntimeException(e);
}
}
protected TemporaryQueue createTemporaryQueue() {
Session jmsSession = null;
try {
jmsSession = getJmsConnection().createSession(false, Session.AUTO_ACKNOWLEDGE);
return jmsSession.createTemporaryQueue();
} catch (JMSException e) {
throw new RuntimeException(e);
} finally {
if (jmsSession != null) {
try {
jmsSession.close();
} catch (Exception ignore) {
}
}
}
}
@Override
public Message waitForReply(Destination replyDestination, int timeout) {
Session jmsSession = null;
try {
jmsSession = getJmsConnection().createSession(false, Session.AUTO_ACKNOWLEDGE);
MessageConsumer consumer = jmsSession.createConsumer(replyDestination);
Message replyMsg = consumer.receive(timeout);
return replyMsg;
} catch (JMSException e) {
throw new RuntimeException(e);
} finally {
if (jmsSession != null) {
try {
jmsSession.close();
} catch (Exception ignore) {
}
}
}
}
@PreDestroy
public void ejbRemove() {
try {
if (connection != null) {
connection.stop();
connection.close();
connection = null;
}
} catch (Exception ignore) {
}
}
}