Wrap db schema changes in a transacion.

mudcoders · Apr 17, 2018 · b28d669 · b28d669
1 parent 96442c4
commit b28d669
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/db.c b/src/db.c
@@ -225,17 +225,22 @@ void db_migrate()
     }
   }
 
+  db_execute("BEGIN EXCLUSIVE TRANSACTION");
   while(command[i].schema != -1) {
     if (command[i].schema > db_schema) {
       if ( !db_execute(command[i].action) ) {
+        db_execute("ROLLBACK TRANSACTION");  
         abort();
       }
     }
     if( !db_execute("UPDATE DB_SCHEMA SET db_version = %i", command[i].schema) ) {
+      db_execute("ROLLBACK TRANSACTION");  
       abort();
     }
     i++;
   }
+  db_execute("COMMIT TRANSACTION");
+
 
   if(command[i-1].schema > db_schema ) {
     log_string("db_schema updated to version %d", command[i-1].schema);