org.jbasics.jms.JMSMessageSender Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2009-2015
 * IT-Consulting Stephan Schloepke (http://www.schloepke.de/)
 * klemm software consulting Mirko Klemm (http://www.klemm-scs.com/)
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */
package org.jbasics.jms;

import java.net.URI;
import java.util.Collections;
import java.util.Enumeration;
import java.util.Optional;
import java.util.UUID;
import java.util.function.BiConsumer;
import java.util.regex.Pattern;

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

import org.jbasics.checker.ContractCheck;
import org.jbasics.exception.DelegatedException;
import org.jbasics.function.ThrowableConsumer;
import org.jbasics.function.ThrowableFunction;
import org.jbasics.pattern.delegation.Delegate;
import org.jbasics.pattern.delegation.ReleasableDelegate;
import org.jbasics.types.delegates.UnmodifiableDelegate;
import org.jbasics.utilities.DataUtilities;

public class JMSMessageSender implements AutoCloseable {
	public static final String JMSX_GROUP_ID = "JMSXGroupID";
	public static final String ENDPOINT_NAMESPACE = "ENDPOINT_NAMESPACE";
	private final MessageHandler messageHandler;
	private final URI defaultEndpoint;
	private final Delegate sessionDelegate;
	private final Delegate producerDelegate;
	private final Delegate defaultReplyDestinationDelegate;
	private final Pattern copyPropertyPattern;

	public interface MessageHandler {
		M createEmptyMessage(Delegate sessionDelegate) throws JMSException;

		void fillMessage(T message, M jmsMessage) throws JMSException;
	}

	public JMSMessageSender(final MessageHandler messageHandler, final Delegate sessionDelegate, final Delegate destinationDelegate,
			final Delegate defaultReplyDestinationDelegate, final URI defaultEndpoint, Pattern copyPropertyPattern) {
		this.messageHandler = ContractCheck.mustNotBeNull(messageHandler, "messageHandler");
		this.sessionDelegate = ContractCheck.mustNotBeNull(sessionDelegate, "sessionDelegate");
		this.producerDelegate = new JMSMessageProducerDelegate(sessionDelegate, destinationDelegate);
		this.defaultReplyDestinationDelegate = defaultReplyDestinationDelegate == null ? UnmodifiableDelegate.nullDelegate() : defaultReplyDestinationDelegate;
		this.defaultEndpoint = defaultEndpoint;
		this.copyPropertyPattern = copyPropertyPattern;
	}

	public String send(final T message) {
		return send(message, null, null, null, null, null);
	}

	public String send(final T message, final String jmsGroupId) {
		return send(message, null, null, null, jmsGroupId, null);
	}

	public String send(final T message, final Destination replyQueue) {
		return send(message, null, null, null, null, replyQueue);
	}

	public String send(final T message, final String jmsGroupId, final Destination replyQueue) {
		return send(message, null, null, null, jmsGroupId, replyQueue);
	}

	public String send(final T message, final URI endpointSelector) {
		return send(message, null, null, endpointSelector, null, null);
	}

	public String send(final T message, final URI endpointSelector, final String jmsGroupId) {
		return send(message, null, null, endpointSelector, jmsGroupId, null);
	}

	public String send(final T message, final URI endpointSelector, final Destination replyQueue) {
		return send(message, null, null, endpointSelector, null, replyQueue);
	}

	public String send(final T message, final Message correlatedMessage, final String correlationId) {
		return send(message, correlatedMessage, correlationId, null, null, null);
	}

	public String send(final T message, final Message correlatedMessage, final String correlationId, final String jmsGroupId) {
		return send(message, correlatedMessage, correlationId, null, jmsGroupId, null);
	}

	public String send(final T message, final Message correlatedMessage, final String correlationId, final Destination replyQueue) {
		return send(message, correlatedMessage, correlationId, null, null, replyQueue);
	}

	public String send(final T message, final Message correlatedMessage, final String correlationId, final String jmsGroupId, final Destination replyQueue) {
		return send(message, correlatedMessage, correlationId, null, jmsGroupId, replyQueue);
	}

	public String send(final T message, final Message correlatedMessage, final String correlationId, final URI endpointSelector) {
		return send(message, correlatedMessage, correlationId, endpointSelector, null, null);
	}

	public String send(final T message, final Message correlatedMessage, final String correlationId, final URI endpointSelector, final String jmsGroupId) {
		return send(message, correlatedMessage, correlationId, endpointSelector, jmsGroupId, null);
	}

	public String send(final T message, final Message correlatedMessage, final String correlationId, final URI endpointSelector, final Destination replyQueue) {
		return send(message, correlatedMessage, correlationId, endpointSelector, null, replyQueue);
	}

	public String send(final T message, final Message correlatedMessage, final String correlationId, final URI endpointSelector, final String jmsGroupId, final Destination replyQueue) {
		return send(message, correlatedMessage, correlationId, endpointSelector, jmsGroupId, replyQueue, null);
	}

	public String send(final T message, final Message correlatedMessage, final String correlationId, final URI endpointSelector, final String jmsGroupId, final Destination replyQueue,
			BiConsumer sendInjection) {
		try {
			final M jmsMessage = this.messageHandler.createEmptyMessage(this.sessionDelegate);
			jmsMessage.setJMSCorrelationID(determineCorrelationId(correlationId, correlatedMessage));
			Destination replyTo = replyQueue == null ? this.defaultReplyDestinationDelegate.delegate() : replyQueue;
			if (replyTo != null) {
				jmsMessage.setJMSReplyTo(replyTo);
			}
			final URI temp = DataUtilities.coalesce(endpointSelector, this.defaultEndpoint);
			if (temp != null) {
				jmsMessage.setStringProperty(JMSMessageSender.ENDPOINT_NAMESPACE, endpointSelector.toASCIIString());
			}
			if (correlatedMessage != null && this.copyPropertyPattern != null) {
				//noinspection unchecked
				for (final String name : Collections.list((Enumeration)correlatedMessage.getPropertyNames())) {
					if (this.copyPropertyPattern.matcher(name).matches()) {
						if (correlatedMessage.getStringProperty(name) != null) {
							jmsMessage.setStringProperty(name, correlatedMessage.getStringProperty(name));
						} else {
							jmsMessage.setObjectProperty(name, correlatedMessage.getObjectProperty(name));
						}
					}
				}
			}
			this.messageHandler.fillMessage(message, jmsMessage);
			Optional.ofNullable(jmsGroupId)
					.ifPresent(ThrowableConsumer.wrap(gid -> jmsMessage.setStringProperty(JMSMessageSender.JMSX_GROUP_ID, gid)));
			Optional.ofNullable(correlatedMessage)
					.map(ThrowableFunction.wrap(Message::getJMSReplyTo))
					.map(x -> (Delegate)new JMSMessageProducerDelegate(this.sessionDelegate, new UnmodifiableDelegate<>(x)))
					.orElse(this.producerDelegate)
					.delegate(p -> {
						Optional.ofNullable(sendInjection).ifPresent(c -> c.accept(jmsMessage, p));
						p.send(jmsMessage);
					});
			return jmsMessage.getJMSMessageID();
		} catch (final JMSException e) {
			throw DelegatedException.delegate(e);
		}
	}

	@Override
	public void close() {
		if (this.sessionDelegate instanceof ReleasableDelegate) {
			((ReleasableDelegate)this.sessionDelegate).release();
		}
	}

	private String determineCorrelationId(String correlationId, Message correlatedMessage) {
		try {
			if (correlatedMessage != null) {
				return correlatedMessage.getJMSCorrelationID() == null ? correlatedMessage.getJMSMessageID() : correlatedMessage.getJMSCorrelationID();
			} else {
				return UUID.randomUUID().toString();
			}
		} catch (final JMSException e) {
			throw DelegatedException.delegate(e);
		}
	}
}