org.ops4j.pax.transx.jms.impl.ConnectionImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of pax-transx-jms Show documentation
There is a newer version: 0.5.4
/*
 * Copyright 2021 OPS4J.
 *
 * 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.ops4j.pax.transx.jms.impl;

import java.util.HashSet;
import java.util.Optional;
import java.util.Set;
import javax.jms.ConnectionConsumer;
import javax.jms.ConnectionMetaData;
import javax.jms.Destination;
import javax.jms.ExceptionListener;
import javax.jms.IllegalStateException;
import javax.jms.JMSException;
import javax.jms.Queue;
import javax.jms.QueueConnection;
import javax.jms.QueueSession;
import javax.jms.ServerSessionPool;
import javax.jms.Session;
import javax.jms.TemporaryQueue;
import javax.jms.TemporaryTopic;
import javax.jms.Topic;
import javax.jms.TopicConnection;
import javax.jms.TopicSession;
import javax.resource.ResourceException;
import javax.resource.spi.ConnectionManager;

import static org.ops4j.pax.transx.jms.impl.Utils.unsupported;


public class ConnectionImpl implements TopicConnection, QueueConnection {

    private final ManagedConnectionFactoryImpl mcf;
    private final ConnectionManager cm;
    private final String userName;
    private final String password;
    private final String clientID;

    private final Set tempQueues = new HashSet<>();
    private final Set tempTopics = new HashSet<>();
    private SessionImpl session;

    private boolean closed;
    private boolean started;

    public ConnectionImpl(ManagedConnectionFactoryImpl mcf, ConnectionManager cm, String userName, String password, String clientID) {
        this.mcf = mcf;
        this.cm = cm;
        this.userName = userName;
        this.password = password;
        this.clientID = clientID;
    }

    public synchronized void close() {
        if (!closed) {
            closed = true;
            doClose();
        }
    }

    private void doClose() {
        Optional.ofNullable(session).ifPresent(SessionImpl::close);
        Utils.doClose(tempQueues, TemporaryQueue::delete);
        Utils.doClose(tempTopics, TemporaryTopic::delete);
    }

    synchronized void closeSession(SessionImpl session) {
        assert this.session == session;
        this.session = null;
    }

    public synchronized void start() throws JMSException {
        if (closed) {
            throw new IllegalStateException("The connection is closed");
        }
        if (!started) {
            SessionImpl s = session;
            if (s != null) {
                s.start();
            }
            started = true;
        }
    }

    public void stop() throws JMSException {
        unsupported("stop");
    }

    public String getClientID() throws JMSException {
        return clientID;
    }

    public void setClientID(String clientID) throws JMSException {
        unsupported("setClientID");
    }

    public ConnectionMetaData getMetaData() throws JMSException {
        ConnectionRequestInfoImpl cri = new ConnectionRequestInfoImpl(false, Session.AUTO_ACKNOWLEDGE, userName, password, clientID);
        try (SessionImpl session = (SessionImpl) cm.allocateConnection(mcf, cri)) {
            session.setConnection(this);
            return session.getManagedConnection().getConnectionMetaData();
        } catch (ResourceException e) {
            throw (JMSException) new JMSException("Unable to retrieve metadata").initCause(e);
        }
    }

    public ExceptionListener getExceptionListener() throws JMSException {
        return unsupported("getExceptionListener");
    }

    public void setExceptionListener(ExceptionListener excptLstnr) throws JMSException {
        unsupported("setExceptionListener");
    }

    public ConnectionConsumer createConnectionConsumer(Destination dest, String msgSel, ServerSessionPool ssp, int maxMsgs) throws JMSException {
        return unsupported("createConnectionConsumer");
    }

    public ConnectionConsumer createConnectionConsumer(Topic topic, String subName, ServerSessionPool ssp, int maxMsgs) throws JMSException {
        return unsupported("createConnectionConsumer");
    }

    public ConnectionConsumer createConnectionConsumer(Queue queue, String subName, ServerSessionPool ssp, int maxMsgs) throws JMSException {
        return unsupported("createConnectionConsumer");
    }

    public ConnectionConsumer createDurableConnectionConsumer(Topic topic, String subName, String msgSel, ServerSessionPool ssp, int maxMsgs) throws JMSException {
        return unsupported("createDurableConnectionConsumer");
    }

    @Override
    public ConnectionConsumer createSharedDurableConnectionConsumer(Topic topic, String subscriptionName, String messageSelector, ServerSessionPool sessionPool, int maxMessages) throws JMSException {
        return unsupported("createSharedDurableConnectionConsumer");
    }

    @Override
    public ConnectionConsumer createSharedConnectionConsumer(Topic topic, String subscriptionName, String messageSelector, ServerSessionPool sessionPool, int maxMessages) throws JMSException {
        return unsupported("createSharedDurableConnectionConsumer");
    }

    @Override
    public Session createSession() throws JMSException {
        return createSession(false, Session.AUTO_ACKNOWLEDGE);
    }

    @Override
    public Session createSession(int sessionMode) throws JMSException {
        return createSession(sessionMode == Session.SESSION_TRANSACTED, sessionMode);
    }

    public TopicSession createTopicSession(boolean transacted, int sessionMode) throws JMSException {
        return createSession(transacted, sessionMode);
    }

    public QueueSession createQueueSession(boolean transacted, int sessionMode) throws JMSException {
        return createSession(transacted, sessionMode);
    }

    public synchronized SessionImpl createSession(boolean transacted, int sessionMode) throws JMSException {
        if (closed) {
            throw new IllegalStateException("The connection is closed");
        }
        if (session != null) {
            throw new IllegalStateException("Only one session per connection is allowed");
        }
        try {
            ConnectionRequestInfoImpl cri = new ConnectionRequestInfoImpl(transacted, sessionMode, userName, password, clientID);
            SessionImpl session = (SessionImpl) cm.allocateConnection(mcf, cri);
            try {
                session.setConnection(this);
                if (started) {
                    session.start();
                }
                this.session = session;
                return session;
            } catch (Throwable t) {
                try {
                    session.close();
                } catch (Throwable ignored) {
                }
                throw t;
            }
        } catch (Exception e) {
            throw Utils.newJMSException(e);
        }
    }

    TemporaryQueue wrapTemporaryQueue(TemporaryQueue queue) {
        tempQueues.add(queue);
        return queue;
    }

    TemporaryTopic wrapTemporaryTopic(TemporaryTopic topic) {
        tempTopics.add(topic);
        return topic;
    }

}