ActiveMQTransactionContext.h

Go to the documentation of this file.

/*
 * 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.
 */
#ifndef _ACTIVEMQ_CORE_ACTIVEMQTRANSACTIONCONTEXT_H_
#define _ACTIVEMQ_CORE_ACTIVEMQTRANSACTIONCONTEXT_H_

#include <memory>

#include <cms/Message.h>
#include <cms/XAResource.h>
#include <cms/CMSException.h>
#include <cms/XAException.h>

#include <activemq/util/Config.h>
#include <activemq/exceptions/ActiveMQException.h>
#include <activemq/commands/LocalTransactionId.h>
#include <activemq/core/Synchronization.h>
#include <activemq/util/LongSequenceGenerator.h>

#include <decaf/lang/exceptions/InvalidStateException.h>
#include <decaf/util/StlSet.h>
#include <decaf/util/Properties.h>
#include <decaf/util/concurrent/Mutex.h>

namespace activemq {
namespace core {
namespace kernels {
    class ActiveMQSessionKernel;
}

    using decaf::lang::Pointer;

    class LocalTransactionEventListener;
    class ActiveMQConnection;
    class TxContextData;

    class AMQCPP_API ActiveMQTransactionContext : public cms::XAResource {
    private:

        // Internal structure to hold all class TX data.
        TxContextData* context;

        // Session this Transaction is associated with
        activemq::core::kernels::ActiveMQSessionKernel* session;

        // The Connection that is the parent of the Session.
        ActiveMQConnection* connection;

        // List of Registered Synchronizations
        decaf::util::StlSet< Pointer<Synchronization> > synchronizations;

    private:

        ActiveMQTransactionContext(const ActiveMQTransactionContext&);
        ActiveMQTransactionContext& operator=(const ActiveMQTransactionContext&);

    public:

        ActiveMQTransactionContext(activemq::core::kernels::ActiveMQSessionKernel* session,
                                   const decaf::util::Properties& properties);

        virtual ~ActiveMQTransactionContext();

        virtual void addSynchronization(const Pointer<Synchronization>& sync);

        virtual void removeSynchronization(const Pointer<Synchronization>& sync);

        virtual void begin();

        virtual void commit();

        virtual void rollback();

        virtual const decaf::lang::Pointer<commands::TransactionId>& getTransactionId() const;

        virtual bool isInTransaction() const;

        virtual bool isInLocalTransaction() const;

        virtual bool isInXATransaction() const;

    public:  // XAResource implementation.

        virtual void commit(const cms::Xid* xid, bool onePhase);

        virtual void end(const cms::Xid* xid, int flags);

        virtual void forget(const cms::Xid* xid);

        virtual int getTransactionTimeout() const;

        virtual bool isSameRM(const cms::XAResource* theXAResource);

        virtual int prepare(const cms::Xid* xid);

        virtual int recover(int flag, cms::Xid** recovered);

        virtual void rollback(const cms::Xid* xid);

        virtual bool setTransactionTimeout(int seconds);

        virtual void start(const cms::Xid* xid, int flags);

    private:

        std::string getResourceManagerId() const;
        void setXid(const cms::Xid* xid);
        bool equals(const cms::Xid* local, const cms::Xid* remote);
        cms::XAException toXAException(cms::CMSException& ex);
        cms::XAException toXAException(decaf::lang::Exception& ex);

        void beforeEnd();
        void afterCommit();
        void afterRollback();

    };

}}

#endif /*_ACTIVEMQ_CORE_ACTIVEMQTRANSACTIONCONTEXT_H_*/