chore: bump connector_arrow from 0.3.0 to 0.4.0

lutra/lutra/Cargo.toml

@@ -21,7 +21,7 @@ required-features = ["cli"]
 anyhow = {workspace = true}
 arrow = {version = "49.0", features = ["prettyprint"], default-features = false}
 clap = {version = "4.4.18", features = ["derive"], optional = true}
-connector_arrow = {version = "0.3.0", features = ["src_sqlite"]}
+connector_arrow = {version = "0.4.0", features = ["src_sqlite"]}
 env_logger = "0.10.2"
 itertools = {workspace = true}
 log = {workspace = true}

lutra/lutra/src/connection.rs

@@ -1,10 +1,11 @@
 use std::path::Path;
 
 use anyhow::Result;
+use connector_arrow::sqlite::SQLiteConnection;
 
 use crate::project::DatabaseModule;
 
-pub fn open(db: &DatabaseModule, project_root: &Path) -> Result<rusqlite::Connection> {
+pub fn open(db: &DatabaseModule, project_root: &Path) -> Result<SQLiteConnection> {
     // convert relative to absolute path
     let mut sqlite_file_abs = project_root.to_path_buf();
     sqlite_file_abs.push(&db.connection_params.file_relative);
@@ -13,5 +14,5 @@ pub fn open(db: &DatabaseModule, project_root: &Path) -> Result<rusqlite::Connec
     // init SQLite
     let sqlite_conn = rusqlite::Connection::open(sqlite_file_abs)?;
 
-    Ok(sqlite_conn)
+    Ok(SQLiteConnection::new(sqlite_conn))
 }

lutra/lutra/src/execute.rs

@@ -47,7 +47,7 @@ fn execute_one(project: &ProjectCompiled, pipeline_ident: &Ident) -> Result<Rela
     };
     log::debug!("executing sql: {pipeline}");
 
-    let batches = connector_arrow::query_one(&mut conn, pipeline)?;
+    let batches = connector_arrow::query(&mut conn, pipeline)?;
 
     Ok(batches)
 }