src/main/activemq/core/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{

    using decaf::lang::Pointer;

    class LocalTransactionEventListener;
    class ActiveMQSession;
    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
        ActiveMQSession* 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( ActiveMQSession* 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_*/