f Make constructor fallible

Author: tnull

lightning-persister/src/sqlite_store/migrations.rs

@@ -1,21 +1,28 @@
 use rusqlite::Connection;
 
-pub(super) fn migrate_schema(connection: &Connection, kv_table_name: &str, from_version: u16, to_version: u16) {
+use lightning::io;
+
+pub(super) fn migrate_schema(connection: &Connection, kv_table_name: &str, from_version: u16, to_version: u16) -> io::Result<()> {
 	assert!(from_version < to_version);
 	if from_version == 1 && to_version == 2 {
 		let sql = format!(
 			"ALTER TABLE {}
 				ADD sub_namespace TEXT DEFAULT \"\" NOT NULL;",
 				kv_table_name);
-		connection
-			.execute(&sql, [])
-			.unwrap_or_else(|e| 
-				panic!("Failed to migrate table {} from user_version {} to {}: {}",
-				kv_table_name, from_version, to_version, e));
-		connection
-			.pragma(Some(rusqlite::DatabaseName::Main), "user_version", to_version, |_| { Ok(()) })
-			.unwrap_or_else(|e| panic!("Failed to upgrade user_version from {} to {}: {}",
+		connection .execute(&sql, []).map_err(|e| {
+				let msg = format!("Failed to migrate table {} from user_version {} to {}: {}",
+				kv_table_name, from_version, to_version, e);
+				io::Error::new(e.kind(), msg)
+			})?;
+
+		connection.pragma(Some(rusqlite::DatabaseName::Main),
+			"user_version", to_version, |_| {
+				Ok(())
+		}).map_err(|e| {
+			let msg = format!("Failed to upgrade user_version from {} to {}: {}",
 				from_version, to_version, e));
+			io::Error::new(e.kind(), msg)
+		})?;
 	}
 }
 

lightning-persister/src/sqlite_store/mod.rs

@@ -68,9 +68,12 @@ impl SqliteStore {
 				io::Error::new(e.kind(), msg)
 			})?;
 		} else if version_res < SCHEMA_USER_VERSION {
-			migrations::migrate_schema(&connection, &kv_table_name, version_res, SCHEMA_USER_VERSION);
+			migrations::migrate_schema(&connection, &kv_table_name, version_res,
+				SCHEMA_USER_VERSION)?;
 		} else if version_res > SCHEMA_USER_VERSION {
-			panic!("Failed to open database: incompatible schema version {}. Expected: {}", version_res, SCHEMA_USER_VERSION);
+			let msg = format!("Failed to open database: incompatible schema version {}. Expected: {}",
+			version_res, SCHEMA_USER_VERSION);
+			return Err(io::Error::new(io::ErrorKind::Other, msg));
 		}
 
 		let sql = format!(