verify after db restore is done

Author: avinassh

bottomless-cli/src/main.rs

@@ -1,3 +1,4 @@
+use std::path::PathBuf;
 use anyhow::Result;
 use aws_sdk_s3::Client;
 use chrono::NaiveDateTime;
@@ -168,6 +169,12 @@ async fn run() -> Result<()> {
         } => {
             tokio::fs::create_dir_all(&database_dir).await?;
             client.restore(generation, utc_time).await?;
+            let db_path = PathBuf::from(&database);
+            let result = verify_db(&db_path).unwrap_or_else(|e| e.to_string());
+            println!("Verification: {result}");
+            if result != "ok" {
+                std::process::exit(1)
+            }
         }
         Commands::Verify {
             generation,
@@ -179,10 +186,7 @@ async fn run() -> Result<()> {
             client.restore(generation, utc_time).await?;
             let size = tokio::fs::metadata(&temp).await?.len();
             println!("Snapshot size: {size}");
-            let conn = rusqlite::Connection::open(&temp)?;
-            let mut stmt = conn.prepare("PRAGMA integrity_check")?;
-            let mut rows = stmt.query(())?;
-            let result: String = rows.next()?.unwrap().get(0)?;
+            let result = verify_db(&temp).unwrap_or_else(|e| e.to_string());
             println!("Verification: {result}");
             let _ = tokio::fs::remove_file(&temp).await;
             if result != "ok" {
@@ -205,6 +209,14 @@ async fn run() -> Result<()> {
     Ok(())
 }
 
+fn verify_db(path: &PathBuf) -> Result<String> {
+    let conn = rusqlite::Connection::open(path)?;
+    let mut stmt = conn.prepare("PRAGMA integrity_check")?;
+    let mut rows = stmt.query(())?;
+    let result: String = rows.next()?.unwrap().get(0)?;
+    Ok(result)
+}
+
 #[tokio::main]
 async fn main() {
     if let Err(e) = run().await {