Add RaiiTransaction helper class for automatic transaction rollback.

diff --git a/localmylist/database.cpp b/localmylist/database.cpp
index 22a8793..d90efc6 100644 (file)
--- a/localmylist/database.cpp
+++ b/localmylist/database.cpp
@@ -1051,4 +1051,23 @@ void Database::handleNotification(const QString &notification)
        }
 }
 
+
+RaiiTransaction::RaiiTransaction(Database *db) : d(db), c(false)
+{
+       d->transaction();
+}
+
+RaiiTransaction::~RaiiTransaction()
+{
+       if (c)
+               d->commit();
+       else
+               d->rollback();
+}
+
+void RaiiTransaction::commit()
+{
+       c = true;
+}
+
 } // namespace LocalMyList

diff --git a/localmylist/database.h b/localmylist/database.h
index 99d8dcf..39435c7 100644 (file)
--- a/localmylist/database.h
+++ b/localmylist/database.h
@@ -274,6 +274,16 @@ private:
        QString connectionName;
 };
 
+class RaiiTransaction
+{
+       Database *d;
+       bool c;
+public:
+       RaiiTransaction(Database *db);
+       ~RaiiTransaction();
+       void commit();
+};
+
 } // namespace LocalMyList
 
 #endif // DATABASE_H