feat(rust/driver/datafusion): remove unnecessary dependency on arrow

apache · Oct 22, 2024 · c48dcee · c48dcee
1 parent 73d9c09
commit c48dcee
Show file tree

Hide file tree

Showing 6 changed files with 10 additions and 6 deletions.
diff --git a/rust/Cargo.lock b/rust/Cargo.lock
diff --git a/rust/Cargo.toml b/rust/Cargo.toml
@@ -34,10 +34,10 @@ categories = ["database"]
 
 [workspace.dependencies]
 adbc_core = { path = "./core" }
-arrow = { version = "53.1.0", default-features = false }
 arrow-array = { version = "53.1.0", default-features = false, features = [
     "ffi",
 ] }
 arrow-buffer = { version = "53.1.0", default-features = false }
 arrow-schema = { version = "53.1.0", default-features = false }
 arrow-select = { version = "53.1.0", default-features = false }
+arrow-cast = { version = "53.1.0", default-features = false }
diff --git a/rust/drivers/datafusion/Cargo.toml b/rust/drivers/datafusion/Cargo.toml
@@ -25,7 +25,6 @@ license = { workspace = true }
 
 [dependencies]
 adbc_core = { path = "../../core" }
-arrow.workspace = true
 arrow-array.workspace = true
 arrow-buffer.workspace = true
 arrow-schema.workspace = true
@@ -34,5 +33,9 @@ datafusion-substrait = "42.0.0"
 tokio = { version = "1.0", features = ["rt-multi-thread"] }
 prost = "0.13.3"
 
+[dev-dependencies]
+arrow-select.workspace = true
+arrow-cast.workspace = true
+
 [lib]
 crate-type = ["lib", "cdylib"]
diff --git a/rust/drivers/datafusion/README.md b/rust/drivers/datafusion/README.md
@@ -25,7 +25,7 @@
 use adbc_core::driver_manager::ManagedDriver;
 use adbc_core::options::AdbcVersion;
 use adbc_core::{Connection, Database, Driver, Statement};
-use arrow::util::pretty::print_batches;
+use arrow_cast::pretty::print_batches;
 use arrow_array::RecordBatch;
 
 fn main() {

diff --git a/rust/drivers/datafusion/src/lib.rs b/rust/drivers/datafusion/src/lib.rs
@@ -25,7 +25,7 @@ use std::vec::IntoIter;
 use std::{collections::HashMap, fmt::Debug};
 use tokio::runtime::Runtime;
 
-use arrow::array::{
+use arrow_array::builder::{
     BooleanBuilder, Int32Builder, Int64Builder, ListBuilder, MapBuilder, MapFieldNames,
     StringBuilder, UInt32Builder,
 };

diff --git a/rust/drivers/datafusion/tests/test_datafusion.rs b/rust/drivers/datafusion/tests/test_datafusion.rs
@@ -21,7 +21,7 @@ use arrow_array::RecordBatch;
 use datafusion::prelude::*;
 
 use adbc_core::options::AdbcVersion;
-use arrow::compute::concat_batches;
+use arrow_select::concat::concat_batches;
 use datafusion_substrait::logical_plan::producer::to_substrait_plan;
 use datafusion_substrait::substrait::proto::Plan;
 use prost::Message;