mirror of
https://github.com/VCMP-SqMod/SqMod.git
synced 2025-02-12 15:57:12 +01:00
4a6bfc086c
Switched to POCO library for unified platform/library interface. Deprecated the external module API. It was creating more problems than solving. Removed most built-in libraries in favor of system libraries for easier maintenance. Cleaned and secured code with help from static analyzers.
204 lines
3.8 KiB
C++
204 lines
3.8 KiB
C++
//
|
|
// Notifier.cpp
|
|
//
|
|
// Library: Data/SQLite
|
|
// Package: SQLite
|
|
// Module: Notifier
|
|
//
|
|
// Implementation of Notifier
|
|
//
|
|
// Copyright (c) 2006, Applied Informatics Software Engineering GmbH.
|
|
// and Contributors.
|
|
//
|
|
// SPDX-License-Identifier: BSL-1.0
|
|
//
|
|
|
|
|
|
#include "Poco/Data/SQLite/Notifier.h"
|
|
|
|
|
|
namespace Poco {
|
|
namespace Data {
|
|
namespace SQLite {
|
|
|
|
|
|
Notifier::Notifier(const Session& session, EnabledEventType enabled):
|
|
_session(session)
|
|
{
|
|
if (enabled & SQLITE_NOTIFY_UPDATE) enableUpdate();
|
|
if (enabled & SQLITE_NOTIFY_COMMIT) enableCommit();
|
|
if (enabled & SQLITE_NOTIFY_ROLLBACK) enableRollback();
|
|
}
|
|
|
|
|
|
Notifier::Notifier(const Session& session, const Any& value, EnabledEventType enabled):
|
|
_session(session),
|
|
_value(value)
|
|
{
|
|
if (enabled & SQLITE_NOTIFY_UPDATE) enableUpdate();
|
|
if (enabled & SQLITE_NOTIFY_COMMIT) enableCommit();
|
|
if (enabled & SQLITE_NOTIFY_ROLLBACK) enableRollback();
|
|
}
|
|
|
|
|
|
Notifier::~Notifier()
|
|
{
|
|
try
|
|
{
|
|
disableAll();
|
|
}
|
|
catch (...)
|
|
{
|
|
poco_unexpected();
|
|
}
|
|
}
|
|
|
|
|
|
bool Notifier::enableUpdate()
|
|
{
|
|
Poco::Mutex::ScopedLock l(_mutex);
|
|
|
|
if (Utility::registerUpdateHandler(Utility::dbHandle(_session), &sqliteUpdateCallbackFn, this))
|
|
_enabledEvents |= SQLITE_NOTIFY_UPDATE;
|
|
|
|
return updateEnabled();
|
|
}
|
|
|
|
|
|
bool Notifier::disableUpdate()
|
|
{
|
|
Poco::Mutex::ScopedLock l(_mutex);
|
|
|
|
if (Utility::registerUpdateHandler(Utility::dbHandle(_session), (Utility::UpdateCallbackType) 0, this))
|
|
_enabledEvents &= ~SQLITE_NOTIFY_UPDATE;
|
|
|
|
return !updateEnabled();
|
|
}
|
|
|
|
|
|
bool Notifier::updateEnabled() const
|
|
{
|
|
return 0 != (_enabledEvents & SQLITE_NOTIFY_UPDATE);
|
|
}
|
|
|
|
|
|
bool Notifier::enableCommit()
|
|
{
|
|
Poco::Mutex::ScopedLock l(_mutex);
|
|
|
|
if (Utility::registerUpdateHandler(Utility::dbHandle(_session), &sqliteCommitCallbackFn, this))
|
|
_enabledEvents |= SQLITE_NOTIFY_COMMIT;
|
|
|
|
return commitEnabled();
|
|
}
|
|
|
|
|
|
bool Notifier::disableCommit()
|
|
{
|
|
Poco::Mutex::ScopedLock l(_mutex);
|
|
|
|
if (Utility::registerUpdateHandler(Utility::dbHandle(_session), (Utility::CommitCallbackType) 0, this))
|
|
_enabledEvents &= ~SQLITE_NOTIFY_COMMIT;
|
|
|
|
return !commitEnabled();
|
|
}
|
|
|
|
|
|
bool Notifier::commitEnabled() const
|
|
{
|
|
return 0 != (_enabledEvents & SQLITE_NOTIFY_COMMIT);
|
|
}
|
|
|
|
|
|
bool Notifier::enableRollback()
|
|
{
|
|
Poco::Mutex::ScopedLock l(_mutex);
|
|
|
|
if (Utility::registerUpdateHandler(Utility::dbHandle(_session), &sqliteRollbackCallbackFn, this))
|
|
_enabledEvents |= SQLITE_NOTIFY_ROLLBACK;
|
|
|
|
return rollbackEnabled();
|
|
}
|
|
|
|
|
|
bool Notifier::disableRollback()
|
|
{
|
|
Poco::Mutex::ScopedLock l(_mutex);
|
|
|
|
if (Utility::registerUpdateHandler(Utility::dbHandle(_session), (Utility::RollbackCallbackType) 0, this))
|
|
_enabledEvents &= ~SQLITE_NOTIFY_ROLLBACK;
|
|
|
|
return !rollbackEnabled();
|
|
}
|
|
|
|
|
|
bool Notifier::rollbackEnabled() const
|
|
{
|
|
return 0 != (_enabledEvents & SQLITE_NOTIFY_ROLLBACK);
|
|
}
|
|
|
|
|
|
bool Notifier::enableAll()
|
|
{
|
|
return enableUpdate() && enableCommit() && enableRollback();
|
|
}
|
|
|
|
|
|
bool Notifier::disableAll()
|
|
{
|
|
return disableUpdate() && disableCommit() && disableRollback();
|
|
}
|
|
|
|
|
|
void Notifier::sqliteUpdateCallbackFn(void* pVal, int opCode, const char* pDB, const char* pTable, Poco::Int64 row)
|
|
{
|
|
poco_check_ptr(pVal);
|
|
Notifier* pV = reinterpret_cast<Notifier*>(pVal);
|
|
|
|
if (opCode == Utility::OPERATION_INSERT)
|
|
{
|
|
pV->_table = pTable;
|
|
pV->_row = row;
|
|
pV->insert.notify(pV);
|
|
}
|
|
else if (opCode == Utility::OPERATION_UPDATE)
|
|
{
|
|
pV->_table = pTable;
|
|
pV->_row = row;
|
|
pV->update.notify(pV);
|
|
}
|
|
else if (opCode == Utility::OPERATION_DELETE)
|
|
{
|
|
pV->_table = pTable;
|
|
pV->_row = row;
|
|
pV->erase.notify(pV);
|
|
}
|
|
}
|
|
|
|
|
|
int Notifier::sqliteCommitCallbackFn(void* pVal)
|
|
{
|
|
Notifier* pV = reinterpret_cast<Notifier*>(pVal);
|
|
|
|
try
|
|
{
|
|
pV->commit.notify(pV);
|
|
}
|
|
catch (...)
|
|
{
|
|
return -1;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
void Notifier::sqliteRollbackCallbackFn(void* pVal)
|
|
{
|
|
Notifier* pV = reinterpret_cast<Notifier*>(pVal);
|
|
pV->rollback.notify(pV);
|
|
}
|
|
|
|
|
|
} } } // namespace Poco::Data::SQLite
|