1 files changed, 15 insertions, 6 deletions
diff --git a/src/lib.rs b/src/lib.rs
index 111ec3e..f151cf1 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -2,20 +2,29 @@ pub mod sqlite;
 pub mod yaml;
 
 
+#[derive(thiserror::Error, Debug)]
+pub enum Error {
+    #[error("SQL error")]
+    Sqlite(#[from] rusqlite::Error),
+}
+
+
 pub fn insert(
     dbconn: &mut rusqlite::Connection,
     table_name: &str,
     data: &mut [yaml_rust::Yaml],
-) {
-    let table_columns = crate::sqlite::get_column_names(&dbconn, table_name).unwrap();
+) -> Result<(), Error> {
+    let table_columns = crate::sqlite::get_column_names(&dbconn, table_name)?;
 
     for mut doc in data {
-        let tx = dbconn.transaction().unwrap();
+        let tx = dbconn.transaction()?;
 
-        crate::yaml::extract(&mut doc, &tx, &table_name, &table_columns).unwrap();
+        crate::yaml::extract(&mut doc, &tx, &table_name, &table_columns)?;
 
-        tx.commit().unwrap();
+        tx.commit()?;
     }
+
+    Ok(())
 }
 
 
@@ -48,7 +57,7 @@ mod tests {
 
         let mut data = yaml_rust::YamlLoader::load_from_str(&yaml_str).unwrap();
 
-        insert(&mut conn, "test", &mut data);
+        insert(&mut conn, "test", &mut data).unwrap();
 
         {
             let mut stmt = conn.prepare(r#"