transaction_context.h

Go to the documentation of this file.

#pragma once
 
#include <mutex>
 
#include "transaction.h"
 
namespace kuzu {
namespace main {
class ClientContext;
}
 
namespace transaction {
 
enum class TransactionMode : uint8_t { AUTO = 0, MANUAL = 1 };
 
class KUZU_API TransactionContext {
public:
    explicit TransactionContext(main::ClientContext& clientContext);
    ~TransactionContext();
 
    bool isAutoTransaction() const { return mode == TransactionMode::AUTO; }
 
    void beginReadTransaction();
    void beginWriteTransaction();
    void beginAutoTransaction(bool readOnlyStatement);
    void beginRecoveryTransaction();
    void validateManualTransaction(bool readOnlyStatement) const;
 
    void commit();
    void rollback();
 
    TransactionMode getTransactionMode() const { return mode; }
    bool hasActiveTransaction() const { return activeTransaction != nullptr; }
    Transaction* getActiveTransaction() const { return activeTransaction.get(); }
 
private:
    void clearTransaction();
 
private:
    void beginTransactionInternal(TransactionType transactionType);
 
private:
    std::mutex mtx;
    main::ClientContext& clientContext;
    TransactionMode mode;
    // TODO(Guodong): Should hold a raw pointer. Move ownership to TransactionManager.
    std::unique_ptr<Transaction> activeTransaction;
};
 
} // namespace transaction
} // namespace kuzu