Implement exception safe sqlite transaction class (rollback)

1 files changed, 20 insertions, 0 deletions

diff --git a/src/sql/Sql.cpp b/src/sql/Sql.cpp
index 754a30d..4e65ddb 100644
--- a/src/sql/Sql.cpp
+++ b/src/sql/Sql.cpp
@@ -1,5 +1,7 @@
 #include "../../include/odhtdb/sql/Sql.hpp"
 #include <sqlite3.h>
+#include <exception>
+#include <cassert>
 
 namespace odhtdb
 {
@@ -18,4 +20,22 @@ namespace odhtdb
         }
         return SQLITE_OK;
     }
+    
+    SqlTransaction::SqlTransaction(sqlite3 *_db) : 
+        db(_db)
+    {
+        assert(db);
+        sqlite3_exec(db, "BEGIN", 0, 0, 0);
+    }
+    
+    SqlTransaction::~SqlTransaction()
+    {
+        if(std::uncaught_exception())
+            sqlite3_exec(db, "ROLLBACK", 0, 0, 0);
+    }
+    
+    void SqlTransaction::commit()
+    {
+        sqlite3_exec(db, "COMMIT", 0, 0, 0);
+    }
 }