org.projectodd.wunderboss.messaging.hornetq.HQContext Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of wunderboss-messaging Show documentation
There is a newer version: 0.13.1
/*
 * Copyright 2014-2015 Red Hat, Inc, and individual contributors.
 *
 * 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.projectodd.wunderboss.messaging.hornetq;

import org.projectodd.wunderboss.messaging.Messaging;
import org.jboss.logging.Logger;

import javax.jms.JMSContext;
import java.util.ArrayList;
import java.util.List;
import java.util.UUID;

public class HQContext implements HQSpecificContext {

    public static ThreadLocal currentContext = new ThreadLocal<>();

    public HQContext(JMSContext jmsContext,
                     Messaging broker,
                     Mode mode,
                     boolean remote) {
        this(jmsContext, broker, mode, remote, null);
    }

    public HQContext(JMSContext jmsContext,
                     Messaging broker,
                     Mode mode,
                     boolean remote,
                     HQSpecificContext parent) {
        this.jmsContext = jmsContext;
        this.broker = broker;
        this.mode = mode;
        this.remote = remote;
        this.parentContext = parent;
        this.id = UUID.randomUUID().toString();

        if (parentContext != null) {
            parentContext.addCloseable(this);
        }
    }

    public static int modeToJMSMode(Mode mode) {
        int jmsMode = 0;
        switch (mode) {
            case AUTO_ACK:
                jmsMode = JMSContext.AUTO_ACKNOWLEDGE;
                break;
            case CLIENT_ACK:
                jmsMode = JMSContext.CLIENT_ACKNOWLEDGE;
                break;
            case TRANSACTED:
                jmsMode = JMSContext.SESSION_TRANSACTED;
                break;
        }

        return jmsMode;
    }

    @Override
    public String id() {
        return this.id;
    }

    @Override
    public Mode mode() {
        return this.mode;
    }

    @Override
    public void commit() {
        if (this.jmsContext.getTransacted()) {
            this.jmsContext.commit();
        }
    }

    @Override
    public void rollback() {
        if (this.jmsContext.getTransacted()) {
            this.jmsContext.rollback();
        }
    }

    @Override
    public void acknowledge() {
        this.jmsContext.acknowledge();
    }

    @Override
    public boolean enlist() throws Exception {
        if (HQXAContext.isTransactionActive()) {
            log.warn("This non-XA context cannot participate in the active transaction; hijinks may ensue");
        }
        return false;
    }

    @Override
    public void close() throws Exception {
        for(AutoCloseable each : this.closeables) {
            each.close();
        }

        this.closeables.clear();
        this.jmsContext.close();
    }

    @Override
    public void addCloseable(AutoCloseable closeable) {
        this.closeables.add(closeable);
    }

    @Override
    public boolean isRemote() {
        return this.remote
                || (this.parentContext != null && this.parentContext.isRemote());
    }

    @Override
    public JMSContext jmsContext() {
        return this.jmsContext;
    }

    @Override
    public Messaging broker() {
        return this.broker;
    }

    @Override
    public boolean isXAEnabled() {
        return false;
    }

    @Override
    public boolean isChild() {
        return this.parentContext != null;
    }

    @Override
    public HQSpecificContext asNonCloseable() {
        return this.new NonClosing();
    }

    @Override
    public HQSpecificContext createChildContext(Mode mode) {
        JMSContext subContext = this.jmsContext.createContext(modeToJMSMode(mode));

        return new HQContext(subContext, this.broker, this.mode, this.remote, this);
    }

    class NonClosing implements HQSpecificContext {
        @Override
        public String id() {
            return HQContext.this.id();
        }

        @Override
        public Mode mode() {
            return HQContext.this.mode();
        }

        @Override
        public void commit() {
            HQContext.this.commit();
        }

        @Override
        public void rollback() {
            HQContext.this.rollback();
        }

        @Override
        public void acknowledge() {
            HQContext.this.acknowledge();
        }

        @Override
        public boolean enlist() throws Exception {
            return HQContext.this.enlist();
        }

        @Override
        public void addCloseable(AutoCloseable closeable) {
            HQContext.this.addCloseable(closeable);
        }

        @Override
        public boolean isRemote() {
            return HQContext.this.isRemote();
        }

        @Override
        public void close() throws Exception {
            // Nope
        }

        public JMSContext jmsContext() {
            return HQContext.this.jmsContext();
        }

        @Override
        public Messaging broker() {
            return HQContext.this.broker();
        }

        @Override
        public boolean isXAEnabled() {
            return HQContext.this.isXAEnabled();
        }

        @Override
        public boolean isChild() {
            return HQContext.this.isChild();
        }

        @Override
        public HQSpecificContext asNonCloseable() {
            return this;
        }

        @Override
        public HQSpecificContext createChildContext(Mode mode) {
            return HQContext.this.createChildContext(mode);
        }
    }

    private final String id;
    private final Mode mode;
    private final boolean remote;
    private final JMSContext jmsContext;
    private final HQSpecificContext parentContext;
    private final Messaging broker;
    private final List closeables = new ArrayList<>();
    private final static Logger log = Logger.getLogger(HQContext.class);
}