AbstractTransactable.hpp

Go to the documentation of this file.

#ifndef ABSTRACTTRANSACTABLE_HPP_
#define ABSTRACTTRANSACTABLE_HPP_

#include <map>
#include "kaa/KaaThread.hpp"
#include "kaa/logging/Log.hpp"
#include "kaa/transact/ITransactable.hpp"

namespace kaa {

template<class Container>
class AbstractTransactable : public ITransactable {
public:
    virtual TransactionIdPtr beginTransaction(IKaaClientContext &context_) {
        TransactionIdPtr trxId(new TransactionId);
        KAA_MUTEX_LOCKING("transactionsGuard_")
        KAA_MUTEX_UNIQUE_DECLARE(transactionsLock, transactionsGuard_);
        KAA_MUTEX_LOCKED("transactionsGuard_")

        transactions_.insert(std::make_pair(trxId, Container()));

        return trxId;
    }

    virtual void rollback(TransactionIdPtr trxId, IKaaClientContext &context_) {
        KAA_MUTEX_LOCKING("transactionsGuard_")
        KAA_MUTEX_UNIQUE_DECLARE(transactionsLock, transactionsGuard_);
        KAA_MUTEX_LOCKED("transactionsGuard_")

        auto it = transactions_.find(trxId);
        if (it != transactions_.end()) {
            transactions_.erase(it);
        }
    }

    Container & getContainerByTrxId(TransactionIdPtr trxId, IKaaClientContext &context_) {
        KAA_MUTEX_LOCKING("transactionsGuard_")
        KAA_MUTEX_UNIQUE_DECLARE(transactionsLock, transactionsGuard_);
        KAA_MUTEX_LOCKED("transactionsGuard_")

        auto it = transactions_.find(trxId);

        if (it != transactions_.end()) {
            return it->second;
        } else {
            KAA_LOG_DEBUG(boost::format("Transaction with id %1% was not found. Creating new instance") % trxId->getId());
            return transactions_[trxId];
        }
    }

    virtual ~AbstractTransactable() {}
protected:
    std::map<TransactionIdPtr, Container> transactions_;
    KAA_MUTEX_DECLARE(transactionsGuard_);
};

}

#endif /* ABSTRACTTRANSACTABLE_HPP_ */