Improve backup error handling

marcoarment · Aug 26, 2024 · 3c5625b · 3c5625b
1 parent 11170b8
commit 3c5625b
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/Sources/Blackbird/BlackbirdDatabase.swift b/Sources/Blackbird/BlackbirdDatabase.swift
@@ -878,7 +878,9 @@ extension Blackbird {
                 guard let targetDbHandle else {
                     throw Error.cannotOpenDatabaseAtPath(path: targetPath, description: "SQLite cannot allocate memory")
                 }
-
+
+                defer { sqlite3_close(targetDbHandle) }
+
                 guard openResult == SQLITE_OK else {
                     let code = sqlite3_errcode(targetDbHandle)
                     let msg = String(cString: sqlite3_errmsg(targetDbHandle), encoding: .utf8) ?? "(unknown)"
@@ -890,6 +892,8 @@ extension Blackbird {
                     throw Blackbird.Database.Error.backupError(description: errorDesc(targetDbHandle))
                 }
 
+                defer { sqlite3_backup_finish(backup) }
+
                 var stepResult = SQLITE_OK
                 while stepResult == SQLITE_OK || stepResult == SQLITE_BUSY || stepResult == SQLITE_LOCKED {
                     stepResult = sqlite3_backup_step(backup, pagesPerStep)
@@ -903,13 +907,10 @@ extension Blackbird {
 
                     await Task.yield()
                 }
-                
+
                 guard stepResult == SQLITE_DONE else {
                     throw Blackbird.Database.Error.backupError(description: errorDesc(targetDbHandle))
                 }
-
-                sqlite3_backup_finish(backup)
-                sqlite3_close(targetDbHandle)
             }
         }
     }