Added io_csv_read and io_csv_write feature (#408)

jorgecarleitao · Sep 17, 2021 · d5cc0db · d5cc0db
1 parent 06c45f8
commit d5cc0db
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 4 deletions.
diff --git a/Cargo.toml b/Cargo.toml
@@ -94,7 +94,9 @@ full = [
     "chrono-tz",
 ]
 merge_sort = ["itertools"]
-io_csv = ["csv", "lazy_static", "regex", "lexical-core", "streaming-iterator"]
+io_csv = ["io_csv_read", "io_csv_write"]
+io_csv_read = ["csv", "lazy_static", "regex", "lexical-core"]
+io_csv_write = ["csv", "streaming-iterator", "lexical-core"]
 io_json = ["serde", "serde_json", "indexmap"]
 io_ipc = ["flatbuffers"]
 io_ipc_compression = ["lz4", "zstd"]

diff --git a/src/io/csv/mod.rs b/src/io/csv/mod.rs
@@ -16,5 +16,9 @@ impl From<chrono::ParseError> for ArrowError {
     }
 }
 
+#[cfg(feature = "io_csv_read")]
+#[cfg_attr(docsrs, doc(cfg(feature = "io_csv_read")))]
 pub mod read;
+#[cfg(feature = "io_csv_write")]
+#[cfg_attr(docsrs, doc(cfg(feature = "io_csv_write")))]
 pub mod write;
diff --git a/src/io/mod.rs b/src/io/mod.rs
@@ -1,5 +1,5 @@
 //! Interact with different formats such as Arrow, CSV, parquet, etc.
-#[cfg(feature = "io_csv")]
+#[cfg(any(feature = "io_csv_read", feature = "io_csv_write"))]
 #[cfg_attr(docsrs, doc(cfg(feature = "io_csv")))]
 pub mod csv;
 

diff --git a/src/util/mod.rs b/src/util/mod.rs
@@ -1,8 +1,8 @@
 //! Misc utilities used in different places in the crate.
 
-#[cfg(any(feature = "compute", feature = "io_csv"))]
+#[cfg(any(feature = "compute", feature = "io_csv_write", feature = "io_csv_read"))]
 mod lexical;
-#[cfg(any(feature = "compute", feature = "io_csv"))]
+#[cfg(any(feature = "compute", feature = "io_csv_write", feature = "io_csv_read"))]
 pub use lexical::*;
 
 #[cfg(feature = "benchmarks")]