org.apache.openejb.resource.activemq.SessionWrapper Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of openejb-lite
The newest version!
/*
 * 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.openejb.resource.activemq;

import javax.jms.BytesMessage;
import javax.jms.Destination;
import javax.jms.JMSException;
import javax.jms.MapMessage;
import javax.jms.Message;
import javax.jms.MessageConsumer;
import javax.jms.MessageListener;
import javax.jms.MessageProducer;
import javax.jms.ObjectMessage;
import javax.jms.Queue;
import javax.jms.QueueBrowser;
import javax.jms.QueueReceiver;
import javax.jms.QueueSender;
import javax.jms.QueueSession;
import javax.jms.Session;
import javax.jms.StreamMessage;
import javax.jms.TemporaryQueue;
import javax.jms.TemporaryTopic;
import javax.jms.TextMessage;
import javax.jms.Topic;
import javax.jms.TopicPublisher;
import javax.jms.TopicSession;
import javax.jms.TopicSubscriber;
import java.io.Serializable;

public class SessionWrapper implements Session, TopicSession, QueueSession {

    private final ConnectionWrapper connectionWrapper;
    private final Session session;

    public SessionWrapper(final ConnectionWrapper connectionWrapper, final Session session) {
        this.connectionWrapper = connectionWrapper;
        this.session = session;
    }

    @Override
    public BytesMessage createBytesMessage() throws JMSException {
        return session.createBytesMessage();
    }

    @Override
    public TextMessage createTextMessage() throws JMSException {
        return session.createTextMessage();
    }

    @Override
    public Message createMessage() throws JMSException {
        return session.createMessage();
    }

    @Override
    public boolean getTransacted() throws JMSException {
        return session.getTransacted();
    }

    @Override
    public void rollback() throws JMSException {
        session.rollback();
    }

    @Override
    public MessageConsumer createConsumer(final Destination destination, final String messageSelector, final boolean noLocal) throws JMSException {
        return session.createConsumer(destination, messageSelector, noLocal);
    }

    @Override
    public QueueBrowser createBrowser(final Queue queue) throws JMSException {
        return session.createBrowser(queue);
    }

    @Override
    public TemporaryQueue createTemporaryQueue() throws JMSException {
        return session.createTemporaryQueue();
    }

    @Override
    public MapMessage createMapMessage() throws JMSException {
        return session.createMapMessage();
    }

    @Override
    public MessageConsumer createConsumer(final Destination destination) throws JMSException {
        return session.createConsumer(destination);
    }

    @Override
    public void close() throws JMSException {
        try {
            session.close();
        } finally {
            this.connectionWrapper.remove(this);
        }
    }

    @Override
    public void unsubscribe(final String name) throws JMSException {
        session.unsubscribe(name);
    }

    @Override
    public ObjectMessage createObjectMessage(final Serializable object) throws JMSException {
        return session.createObjectMessage(object);
    }

    @Override
    public void run() {
        session.run();
    }

    @Override
    public void recover() throws JMSException {
        session.recover();
    }

    @Override
    public void commit() throws JMSException {
        session.commit();
    }

    @Override
    public int getAcknowledgeMode() throws JMSException {
        return session.getAcknowledgeMode();
    }

    @Override
    public TextMessage createTextMessage(final String text) throws JMSException {
        return session.createTextMessage(text);
    }

    @Override
    public TopicSubscriber createDurableSubscriber(final Topic topic, final String name, final String messageSelector, final boolean noLocal) throws JMSException {
        return session.createDurableSubscriber(topic, name, messageSelector, noLocal);
    }

    @Override
    public TopicPublisher createPublisher(final Topic topic) throws JMSException {
        return TopicSession.class.cast(this.session).createPublisher(topic);
    }

    @Override
    public ObjectMessage createObjectMessage() throws JMSException {
        return session.createObjectMessage();
    }

    @Override
    public Topic createTopic(final String topicName) throws JMSException {
        return session.createTopic(topicName);
    }

    @Override
    public TopicSubscriber createSubscriber(final Topic topic) throws JMSException {
        return TopicSession.class.cast(this.session).createSubscriber(topic);
    }

    @Override
    public TopicSubscriber createSubscriber(final Topic topic, final String messageSelector, final boolean noLocal) throws JMSException {
        return TopicSession.class.cast(this.session).createSubscriber(topic, messageSelector, noLocal);
    }

    @Override
    public void setMessageListener(final MessageListener listener) throws JMSException {
        session.setMessageListener(listener);
    }

    @Override
    public QueueBrowser createBrowser(final Queue queue, final String messageSelector) throws JMSException {
        return session.createBrowser(queue, messageSelector);
    }

    @Override
    public MessageProducer createProducer(final Destination destination) throws JMSException {
        return session.createProducer(destination);
    }

    @Override
    public Queue createQueue(final String queueName) throws JMSException {
        return session.createQueue(queueName);
    }

    @Override
    public QueueReceiver createReceiver(final Queue queue) throws JMSException {
        return QueueSession.class.cast(this.session).createReceiver(queue);
    }

    @Override
    public QueueReceiver createReceiver(final Queue queue, final String messageSelector) throws JMSException {
        return QueueSession.class.cast(this.session).createReceiver(queue, messageSelector);
    }

    @Override
    public QueueSender createSender(final Queue queue) throws JMSException {
        return QueueSession.class.cast(this.session).createSender(queue);
    }

    @Override
    public TopicSubscriber createDurableSubscriber(final Topic topic, final String name) throws JMSException {
        return session.createDurableSubscriber(topic, name);
    }

    @Override
    public MessageConsumer createConsumer(final Destination destination, final String messageSelector) throws JMSException {
        return session.createConsumer(destination, messageSelector);
    }

    @Override
    public StreamMessage createStreamMessage() throws JMSException {
        return session.createStreamMessage();
    }

    @Override
    public MessageListener getMessageListener() throws JMSException {
        return session.getMessageListener();
    }

    @Override
    public TemporaryTopic createTemporaryTopic() throws JMSException {
        return session.createTemporaryTopic();
    }

    @Override
    public boolean equals(final Object o) {
        if (this == o) {
            return true;
        }
        if (o == null || getClass() != o.getClass()) {
            return false;
        }

        final SessionWrapper that = (SessionWrapper) o;

        return connectionWrapper.equals(that.connectionWrapper) && session.equals(that.session);

    }

    @Override
    public int hashCode() {
        int result = connectionWrapper.hashCode();
        result = 31 * result + session.hashCode();
        return result;
    }
}