mirror of
https://github.com/VCMP-SqMod/SqMod.git
synced 2024-11-08 08:47:17 +01:00
105 lines
3.4 KiB
C++
105 lines
3.4 KiB
C++
// ------------------------------------------------------------------------------------------------
|
|
#include "Transaction.hpp"
|
|
#include "Connection.hpp"
|
|
|
|
// ------------------------------------------------------------------------------------------------
|
|
namespace SqMod {
|
|
|
|
// ------------------------------------------------------------------------------------------------
|
|
SQInteger Transaction::Typename(HSQUIRRELVM vm)
|
|
{
|
|
static const SQChar name[] = _SC("SqSQLiteTransaction");
|
|
sq_pushstring(vm, name, sizeof(name));
|
|
return 1;
|
|
}
|
|
|
|
// ------------------------------------------------------------------------------------------------
|
|
Transaction::Transaction(const Connection & db)
|
|
: Transaction(db.GetHandle())
|
|
{
|
|
/* ... */
|
|
}
|
|
|
|
// ------------------------------------------------------------------------------------------------
|
|
Transaction::Transaction(const ConnRef & db)
|
|
: m_Handle(db), m_Committed(false)
|
|
{
|
|
// Was the specified database connection valid?
|
|
if (!m_Handle)
|
|
{
|
|
STHROWF("Invalid connection handle");
|
|
}
|
|
// Attempt to begin transaction
|
|
m_Handle->mStatus = sqlite3_exec(m_Handle->mPtr, "BEGIN", nullptr, nullptr, nullptr);
|
|
// Validate the result
|
|
if (m_Handle->mStatus != SQLITE_OK)
|
|
{
|
|
STHROWF("Unable to begin transaction [%s]", m_Handle->ErrMsg());
|
|
}
|
|
}
|
|
|
|
// ------------------------------------------------------------------------------------------------
|
|
Transaction::~Transaction()
|
|
{
|
|
// Was this transaction successfully committed?
|
|
if (m_Committed)
|
|
{
|
|
return; // We're done here!
|
|
}
|
|
// Attempt to roll back changes because this failed to commit
|
|
m_Handle->mStatus = sqlite3_exec(m_Handle->mPtr, "ROLLBACK", nullptr, nullptr, nullptr);
|
|
// Validate the result
|
|
if (m_Handle->mStatus != SQLITE_OK)
|
|
{
|
|
STHROWF("Unable to rollback transaction [%s]", m_Handle->ErrMsg());
|
|
}
|
|
}
|
|
|
|
// ------------------------------------------------------------------------------------------------
|
|
bool Transaction::Commit()
|
|
{
|
|
// We shouldn't even be here if there wasn't a valid connection but let's be sure
|
|
if (!m_Handle)
|
|
{
|
|
STHROWF("Invalid database connection");
|
|
}
|
|
// Was this transaction already committed?
|
|
else if (m_Committed)
|
|
{
|
|
STHROWF("Transaction was already committed");
|
|
}
|
|
// Attempt to commit the change during this transaction
|
|
m_Handle->mStatus = sqlite3_exec(m_Handle->mPtr, "COMMIT", nullptr, nullptr, nullptr);
|
|
// Validate the result
|
|
if (m_Handle->mStatus != SQLITE_OK)
|
|
{
|
|
STHROWF("Unable to commit transaction [%s]", m_Handle->ErrMsg());
|
|
}
|
|
else
|
|
{
|
|
m_Committed = true; // Everything was committed successfully
|
|
}
|
|
// Return the result
|
|
return m_Committed;
|
|
}
|
|
|
|
// ================================================================================================
|
|
void Register_Transaction(Table & sqlns)
|
|
{
|
|
sqlns.Bind(_SC("Transaction"),
|
|
Class< Transaction, NoCopy< Transaction > >(sqlns.GetVM(), _SC("SqSQLiteTransaction"))
|
|
// Constructors
|
|
.Ctor< const Connection & >()
|
|
// Meta-methods
|
|
.SquirrelFunc(_SC("_typename"), &Transaction::Typename)
|
|
.Func(_SC("_tostring"), &Transaction::ToString)
|
|
// Properties
|
|
.Prop(_SC("IsValid"), &Transaction::IsValid)
|
|
.Prop(_SC("Committed"), &Transaction::Commited)
|
|
// Member Methods
|
|
.Func(_SC("Commit"), &Transaction::Commit)
|
|
);
|
|
}
|
|
|
|
} // Namespace:: SqMod
|