mirror of
https://github.com/saitohirga/WSJT-X.git
synced 2024-11-25 13:48:42 -05:00
76 lines
1.6 KiB
C++
76 lines
1.6 KiB
C++
#ifndef QT_DB_HELPERS_HPP_
|
|
#define QT_DB_HELPERS_HPP_
|
|
|
|
#include <utility>
|
|
#include <stdexcept>
|
|
#include <QString>
|
|
#include <QSqlError>
|
|
#include <QSqlTableModel>
|
|
|
|
template<typename T, typename Func, typename... Args>
|
|
void SQL_error_check (T&& object, Func func, Args&&... args)
|
|
{
|
|
if (!(std::forward<T> (object).*func) (std::forward<Args> (args)...))
|
|
{
|
|
auto error = object.lastError ();
|
|
if (QSqlError::NoError != error.type ())
|
|
{
|
|
throw std::runtime_error {("Database Error: " + error.text ()).toStdString ()};
|
|
}
|
|
}
|
|
}
|
|
|
|
class ConditionalTransaction final
|
|
: private boost::noncopyable
|
|
{
|
|
public:
|
|
explicit ConditionalTransaction (QSqlTableModel& model)
|
|
: model_ {model}
|
|
, submitted_ {false}
|
|
{
|
|
model_.database ().transaction ();
|
|
}
|
|
bool submit (bool throw_on_error = true)
|
|
{
|
|
Q_ASSERT (model_.isDirty ());
|
|
bool ok {true};
|
|
if (throw_on_error)
|
|
{
|
|
SQL_error_check (model_, &QSqlTableModel::submitAll);
|
|
}
|
|
else
|
|
{
|
|
ok = model_.submitAll ();
|
|
}
|
|
submitted_ = submitted_ || ok;
|
|
return ok;
|
|
}
|
|
void revert ()
|
|
{
|
|
Q_ASSERT (model_.isDirty ());
|
|
model_.revertAll ();
|
|
}
|
|
~ConditionalTransaction ()
|
|
{
|
|
if (model_.isDirty ())
|
|
{
|
|
// abandon un-submitted changes to the model
|
|
model_.revertAll ();
|
|
}
|
|
auto database = model_.database ();
|
|
if (submitted_)
|
|
{
|
|
SQL_error_check (database, &QSqlDatabase::commit);
|
|
}
|
|
else
|
|
{
|
|
database.rollback ();
|
|
}
|
|
}
|
|
private:
|
|
QSqlTableModel& model_;
|
|
bool submitted_;
|
|
};
|
|
|
|
#endif
|