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