Update src/db.rs

LGTM Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
MostroP2P · Jan 3, 2025 · 66bbd9c · 66bbd9c
1 parent d29f20e
commit 66bbd9c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/db.rs b/src/db.rs
@@ -32,6 +32,10 @@ pub async fn connect() -> Result<Pool<Sqlite>> {
                 match sqlx::migrate!().run(&pool).await {
                     Ok(_) => (),
                     Err(e) => {
+                        // Clean up the created file on migration failure
+                        if let Err(cleanup_err) = std::fs::remove_file(db_path) {
+                            tracing::warn!("Failed to clean up db file: {}", cleanup_err);
+                        }
                         return Err(anyhow::anyhow!("Failed to run migrations: {}", e));
                     }
                 }