/*
* 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.tuscany.sca.binding.jms.transport;
import java.util.logging.Level;
import java.util.logging.Logger;
import javax.jms.DeliveryMode;
import javax.jms.Destination;
import javax.jms.JMSException;
import javax.jms.MessageProducer;
import javax.jms.Session;
import javax.naming.NamingException;
import org.apache.tuscany.sca.assembly.AssemblyFactory;
import org.apache.tuscany.sca.assembly.Endpoint;
import org.apache.tuscany.sca.assembly.EndpointReference;
import org.apache.tuscany.sca.binding.jms.JMSBinding;
import org.apache.tuscany.sca.binding.jms.JMSBindingConstants;
import org.apache.tuscany.sca.binding.jms.JMSBindingException;
import org.apache.tuscany.sca.binding.jms.context.JMSBindingContext;
import org.apache.tuscany.sca.binding.jms.provider.JMSMessageProcessor;
import org.apache.tuscany.sca.binding.jms.provider.JMSMessageProcessorUtil;
import org.apache.tuscany.sca.binding.jms.provider.JMSResourceFactory;
import org.apache.tuscany.sca.core.ExtensionPointRegistry;
import org.apache.tuscany.sca.core.FactoryExtensionPoint;
import org.apache.tuscany.sca.core.invocation.InterceptorAsyncImpl;
import org.apache.tuscany.sca.interfacedef.Operation;
import org.apache.tuscany.sca.invocation.Interceptor;
import org.apache.tuscany.sca.invocation.Invoker;
import org.apache.tuscany.sca.invocation.Message;
import org.apache.tuscany.sca.runtime.RuntimeComponentService;
import org.apache.tuscany.sca.runtime.RuntimeEndpoint;
/**
*
* @version $Rev: 1291715 $ $Date: 2012-02-21 11:31:26 +0000 (Tue, 21 Feb 2012) $
*/
public class TransportServiceInterceptor extends InterceptorAsyncImpl {
private static final Logger logger = Logger.getLogger(TransportServiceInterceptor.class.getName());
private Invoker next;
private JMSResourceFactory jmsResourceFactory;
private JMSBinding jmsBinding;
private JMSMessageProcessor responseMessageProcessor;
private RuntimeComponentService service;
private String correlationScheme;
private AssemblyFactory assemblyFactory;
public TransportServiceInterceptor(ExtensionPointRegistry registry, JMSBinding jmsBinding, JMSResourceFactory jmsResourceFactory, RuntimeEndpoint endpoint) {
super();
this.jmsBinding = jmsBinding;
this.jmsResourceFactory = jmsResourceFactory;
this.responseMessageProcessor = JMSMessageProcessorUtil.getResponseMessageProcessor(registry, jmsBinding);
this.service = (RuntimeComponentService)endpoint.getService();
this.correlationScheme = jmsBinding.getCorrelationScheme();
FactoryExtensionPoint factories = registry.getExtensionPoint(FactoryExtensionPoint.class);
this.assemblyFactory = factories.getFactory(AssemblyFactory.class);
}
public Message invoke(Message msg) {
try {
return invokeResponse(next.invoke(invokeRequest(msg)));
} catch (Throwable e) {
try {
// Normally only runtime exceptions (whether thrown by the application or the runtime)
// reach this catch block. Business exceptions are handled in the normal invokeResponse path.
logger.log(Level.SEVERE, "Exception invoking service '" + service.getName(), e);
Operation operation = msg.getOperation();
if (operation == null || !operation.isNonBlocking()) {
JMSBindingContext context = msg.getBindingContext();
Session session = context.getJmsResponseSession();
javax.jms.Message replyJMSMsg = responseMessageProcessor.createFaultMessage(session, e);
msg.setBody(replyJMSMsg);
invokeResponse(msg);
}
} catch (Throwable e2) {}
// Rethrow a runtime exception so that the JMS resource adapter can rollback
// the message (if delivery is transacted) and increment the failed delivery count.
if (e instanceof Error) {
throw (Error)e;
}
if (e instanceof RuntimeException) {
throw (RuntimeException)e;
}
return msg;
} finally {
try {
try {
((JMSBindingContext)msg.getBindingContext()).closeJmsResponseSession();
} catch (Throwable e) {
}
// Use the resource factory in the binding context to close the response connection,
// to ensure we use same resource factory used to close response session.
JMSResourceFactory rf = ((JMSBindingContext)msg.getBindingContext()).getJmsResourceFactory();
if (rf.isConnectionClosedAfterUse())
rf.closeResponseConnection();
} catch (JMSException e) {
}
}
}
public Message invokeRequest(Message msg) {
EndpointReference from = assemblyFactory.createEndpointReference();
Endpoint fromEndpoint = assemblyFactory.createEndpoint();
from.setTargetEndpoint(fromEndpoint);
from.setStatus(EndpointReference.Status.WIRED_TARGET_FOUND_AND_MATCHED);
msg.setFrom(from);
return msg;
} // end method invokeRequest
public Message invokeResponse(Message msg) {
JMSBindingContext context = msg.getBindingContext();
try {
//if operation is oneway, return back.
Operation operation = msg.getOperation();
if (operation != null && operation.isNonBlocking()) {
return msg;
}
Session session = context.getJmsResponseSession();
javax.jms.Message requestJMSMsg = context.getJmsMsg();
javax.jms.Message responseJMSMsg = msg.getBody();
Destination replyDest = requestJMSMsg.getJMSReplyTo();
if (replyDest == null) {
if (jmsBinding.getResponseDestinationName() != null) {
try {
replyDest = jmsResourceFactory.lookupDestination(jmsBinding.getResponseDestinationName());
} catch (NamingException e) {
throw new JMSBindingException("Exception lookingup response destination", e);
}
}
}
if (replyDest == null) {
// assume no reply is expected
if (msg.getBody() != null) {
logger.log(Level.FINE, "JMS service '" + service.getName() + "' dropped response as request has no replyTo");
}
return msg;
}
if ((msg.getOperation() != null)) {
String operationName = msg.getOperation().getName();
if (jmsBinding.getEffectiveJMSPriority(operationName) != null) {
responseJMSMsg.setJMSPriority(jmsBinding.getEffectiveJMSPriority(operationName));
}
if ( jmsBinding.getEffectiveJMSType(operationName) != null) {
responseJMSMsg.setJMSType(jmsBinding.getEffectiveJMSType(operationName));
}
if ((jmsBinding.getEffectiveJMSDeliveryMode(operationName) != null)) {
responseJMSMsg.setJMSDeliveryMode(jmsBinding.getEffectiveJMSDeliveryMode(operationName) ? DeliveryMode.PERSISTENT : DeliveryMode.NON_PERSISTENT);
}
if ((jmsBinding.getEffectiveJMSTimeToLive(operationName) != null)) {
responseJMSMsg.setJMSExpiration(jmsBinding.getEffectiveJMSTimeToLive(operationName).longValue());
}
}
if (correlationScheme == null ||
JMSBindingConstants.CORRELATE_MSG_ID.equalsIgnoreCase(correlationScheme)) {
responseJMSMsg.setJMSCorrelationID(requestJMSMsg.getJMSMessageID());
} else if (JMSBindingConstants.CORRELATE_CORRELATION_ID.equalsIgnoreCase(correlationScheme)) {
responseJMSMsg.setJMSCorrelationID(requestJMSMsg.getJMSCorrelationID());
}
MessageProducer producer = session.createProducer(replyDest);
// Set jms header attributes in producer, not message.
producer.setDeliveryMode(responseJMSMsg.getJMSDeliveryMode());
producer.setPriority(responseJMSMsg.getJMSPriority());
producer.setTimeToLive(responseJMSMsg.getJMSExpiration());
producer.send((javax.jms.Message)msg.getBody());
producer.close();
return msg;
} catch (JMSException e) {
throw new JMSBindingException(e);
} finally {
context.closeJmsResponseSession();
}
}
public Invoker getNext() {
return next;
}
public void setNext(Invoker next) {
this.next = next;
}
public Message processRequest(Message msg) {
return invokeRequest( msg );
} // end method processRequest
public Message processResponse(Message msg) {
JMSBindingContext context = msg.getBindingContext();
try {
Session session = context.getJmsResponseSession();
//javax.jms.Message requestJMSMsg = context.getJmsMsg();
javax.jms.Message responseJMSMsg = msg.getBody();
//Destination replyDest = requestJMSMsg.getJMSReplyTo();
Destination replyDest = context.getReplyToDestination();
if (replyDest == null) {
if (jmsBinding.getResponseDestinationName() != null) {
try {
replyDest = jmsResourceFactory.lookupDestination(jmsBinding.getResponseDestinationName());
} catch (NamingException e) {
throw new JMSBindingException("Exception lookingup response destination", e);
}
}
} // end if
if (replyDest == null) {
// assume no reply is expected
if (msg.getBody() != null) {
logger.log(Level.FINE, "JMS service '" + service.getName() + "' dropped response as request has no replyTo");
}
return msg;
} // end if
if ((msg.getOperation() != null)) {
String operationName = msg.getOperation().getName();
if (jmsBinding.getEffectiveJMSPriority(operationName) != null) {
responseJMSMsg.setJMSPriority(jmsBinding.getEffectiveJMSPriority(operationName));
}
if ( jmsBinding.getEffectiveJMSType(operationName) != null) {
responseJMSMsg.setJMSType(jmsBinding.getEffectiveJMSType(operationName));
}
if ((jmsBinding.getEffectiveJMSDeliveryMode(operationName) != null)) {
responseJMSMsg.setJMSDeliveryMode(jmsBinding.getEffectiveJMSDeliveryMode(operationName) ? DeliveryMode.PERSISTENT : DeliveryMode.NON_PERSISTENT);
}
if ((jmsBinding.getEffectiveJMSTimeToLive(operationName) != null)) {
responseJMSMsg.setJMSExpiration(jmsBinding.getEffectiveJMSTimeToLive(operationName).longValue());
}
}
/*
if (correlationScheme == null ||
JMSBindingConstants.CORRELATE_MSG_ID.equalsIgnoreCase(correlationScheme)) {
responseJMSMsg.setJMSCorrelationID(requestJMSMsg.getJMSMessageID());
} else if (JMSBindingConstants.CORRELATE_CORRELATION_ID.equalsIgnoreCase(correlationScheme)) {
responseJMSMsg.setJMSCorrelationID(requestJMSMsg.getJMSCorrelationID());
}
*/
MessageProducer producer = session.createProducer(replyDest);
// Set jms header attributes in producer, not message.
//int deliveryMode = requestJMSMsg.getJMSDeliveryMode();
//producer.setDeliveryMode(deliveryMode);
//int deliveryPriority = requestJMSMsg.getJMSPriority();
//producer.setPriority(deliveryPriority);
//long timeToLive = requestJMSMsg.getJMSExpiration();
//producer.setTimeToLive(timeToLive);
producer.send((javax.jms.Message)msg.getBody());
producer.close();
return msg;
} catch (JMSException e) {
throw new JMSBindingException(e);
} finally {
context.closeJmsResponseSession();
} // end try
} // end method processResponse
}