Add RaiiTransaction helper class for automatic transaction rollback.

diff --git a/localmylist/database.cpp b/localmylist/database.cpp
index 22a8793d00578147ee35e41bd7500564c2d4de92..d90efc68940e3cdf0b7dc6dda18dccbddc2736a7 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 99d8dcf2854448c2d90e947b05a0cfafc08f2206..39435c7610e1b104e6473dedd06a3d3cd366890d 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