org.fusesource.fabric.bridge.internal.SourceDeliveryHandler Maven / Gradle / Ivy

Go to download
/**
 * Copyright (C) FuseSource, Inc.
 * http://fusesource.com
 *
 * 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.fusesource.fabric.bridge.internal;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

import javax.jms.Destination;
import javax.jms.JMSException;
import javax.jms.Message;
import javax.jms.MessageProducer;
import javax.jms.Session;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.jms.core.JmsTemplate;
import org.springframework.jms.core.ProducerCallback;
import org.springframework.jms.support.JmsUtils;

/**
 * @author Dhiraj Bokde
 *
 */
public class SourceDeliveryHandler extends AbstractDeliveryHandler {
	
	static final Logger LOG = LoggerFactory.getLogger(SourceDeliveryHandler.class);
	
	private String destinationName;
	
	private boolean pubSubDomain;
	
	private Destination stagingDestination;

	protected ProducerCallback createProducerCallback(List messages) {
		return new SourceProducerCallback(messages);
	}

	protected MessageProducer createMessageProducer(Session localSession)
			throws JMSException {
		return localSession.createProducer(stagingDestination);
	}

	protected JmsTemplate createJmsTemplate() {
		JmsTemplate jmsTemplate = new JmsTemplate(getTargetConnectionFactory());
		jmsTemplate.setSessionAcknowledgeMode(getDispatchPolicy().getRemoteAcknowledgeMode());
		jmsTemplate.setSessionTransacted(getDispatchPolicy().isRemoteSessionTransacted());
		jmsTemplate.setDefaultDestination(stagingDestination);
		return jmsTemplate;
	}

	public String getDestinationName() {
		return destinationName;
	}

	public void setDestinationName(String destinationName) {
		this.destinationName = destinationName;
	}

	public void setPubSubDomain(boolean pubSubDomain) {
		this.pubSubDomain = pubSubDomain;
	}

	public void setStagingDestination(Destination stagingDestination) {
		this.stagingDestination = stagingDestination;
	}

	private class SourceProducerCallback implements ProducerCallback {
	
		private final List messages;
	
		public SourceProducerCallback(List messages) {
			this.messages = messages;
		}
	
		@Override
		public Message doInJms(Session sendSession, MessageProducer messageProducer)
				throws JMSException {
			
			Message currentMessage = null;
			try {
				Map properties = new HashMap();
				for (final Message message : messages) {
					
					currentMessage = message;
					if (isReuseMessage()) {
	
						// make message properties writable by clearing and resetting them
						getMessageProperties(currentMessage, properties);
						currentMessage.clearProperties();
						setMessageProperties(currentMessage, properties);
						properties.clear();
						
					} else {
	
						currentMessage = copyMessage(currentMessage, sendSession);
					
					}
	
					// allow tunneling messages through bridge
					if (currentMessage.getStringProperty(getDestinationNameHeader()) == null) {
						currentMessage.setStringProperty(getDestinationNameHeader(), destinationName);
						currentMessage.setBooleanProperty(getDestinationTypeHeader(), pubSubDomain);
						// TODO are there other properties to set??
					}
					
					// convert message before send
					if (getDispatchPolicy().getMessageConverter() != null) {
						currentMessage = getDispatchPolicy().getMessageConverter().convert(currentMessage);
					}
	
					// TODO set delivery mode, priority and time to live??
					messageProducer.send(currentMessage);
				}
				
				// commit sendSession if necessary
				if (!isReuseSession()) {
					JmsUtils.commitIfNecessary(sendSession);
				}
				
				if (LOG.isDebugEnabled()) {
					LOG.debug("Sent " + messages.size() + " messages to " + stagingDestination + " in session " + sendSession);
				}
				
				return null;
				
			} catch (JMSException exception) {
				LOG.error ("Error sending message " + currentMessage + ": " + exception.getMessage(), exception);
				throw exception;
			}
		}
	
	}

}