TransactionContext.h

/*
 * Copyright (c) 2008-2015, Hazelcast, Inc. All Rights Reserved.
 *
 * 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.
 */
//
// Created by sancar koyunlu on 8/5/13.
#ifndef HAZELCAST_TransactionContext
#define HAZELCAST_TransactionContext

#include "hazelcast/client/TransactionOptions.h"
#include "hazelcast/client/txn/TransactionProxy.h"
#include "hazelcast/client/TransactionalMap.h"
#include "hazelcast/client/exception/IllegalStateException.h"
#include "hazelcast/client/TransactionalQueue.h"
#include "hazelcast/client/TransactionalMultiMap.h"
#include "hazelcast/client/TransactionalList.h"
#include "hazelcast/client/TransactionalSet.h"

#if  defined(WIN32) || defined(_WIN32) || defined(WIN64) || defined(_WIN64)
#pragma warning(push)
#pragma warning(disable: 4251) //for dll export
#endif

namespace hazelcast {
    namespace client {
        namespace spi {
            class ClientContext;
        }

        namespace connection {
            class ConnectionManager;
            class Connection;
        }

        class HAZELCAST_API TransactionContext {
        public:
            TransactionContext(spi::ClientContext &clientContext, const TransactionOptions &);
            std::string getTxnId() const;

            void beginTransaction();

            void commitTransaction();

            void rollbackTransaction();

            template<typename K, typename V>
            TransactionalMap<K, V> getMap(const std::string &name) {
                return getTransactionalObject< TransactionalMap<K, V> >(name);
            }

            template<typename E>
            TransactionalQueue< E > getQueue(const std::string &name) {
                return getTransactionalObject< TransactionalQueue< E > >(name);
            }

            template<typename K, typename V>
            TransactionalMultiMap<K, V> getMultiMap(const std::string &name) {
                return getTransactionalObject< TransactionalMultiMap<K, V> >(name);
            }

            template<typename E>
            TransactionalList< E > getList(const std::string &name) {
                return getTransactionalObject< TransactionalList< E > >(name);
            }

            template<typename E>
            TransactionalSet< E > getSet(const std::string &name) {
                return getTransactionalObject< TransactionalSet< E > >(name);
            }

            template<typename T>
            T getTransactionalObject(const std::string &name) {
                if (transaction.getState() != txn::TxnState::ACTIVE) {
                    std::string message = "No transaction is found while accessing ";
                    message += "transactional object -> [" + name + "]!";
                    throw exception::IllegalStateException("TransactionContext::getMap(const std::string& name)", message);
                }
                T txnObject(name, &transaction);
                return txnObject;
            }

        private :
            const int CONNECTION_TRY_COUNT;
            spi::ClientContext &clientContext;
            TransactionOptions options;
            boost::shared_ptr<connection::Connection> txnConnection;
            txn::TransactionProxy transaction;

            boost::shared_ptr<connection::Connection> connect();

        };

    }
}

#if  defined(WIN32) || defined(_WIN32) || defined(WIN64) || defined(_WIN64)
#pragma warning(pop)
#endif

#endif //HAZELCAST_TransactionContext