array_formatter_factory.rs into datafusion-expr

Author: tobixdev

datafusion/core/src/dataframe/mod.rs

@@ -69,9 +69,9 @@ use datafusion_functions_aggregate::expr_fn::{
     avg, count, max, median, min, stddev, sum,
 };
 
-use crate::extension_types::DFArrayFormatterFactory;
 use async_trait::async_trait;
 use datafusion_catalog::Session;
+use datafusion_expr::extension_types::DFArrayFormatterFactory;
 
 /// Contains options that control how data is
 /// written out from a DataFrame

datafusion/core/src/lib.rs

@@ -894,7 +894,6 @@ pub mod variable {
 #[cfg(not(target_arch = "wasm32"))]
 pub mod test;
 
-mod extension_types;
 mod schema_equivalence;
 pub mod test_util;
 

datafusion/core/tests/extension_types/pretty_printing.rs

@@ -59,7 +59,7 @@ async fn create_test_table() -> Result<DataFrame> {
 }
 
 #[tokio::test]
-async fn test_pretty_print_logical_plan() -> Result<()> {
+async fn test_pretty_print_extension_type_formatter() -> Result<()> {
     let result = create_test_table().await?.to_string().await?;
 
     assert_snapshot!(

…tension_types/array_formatter_factory.rs …tension_types/array_formatter_factory.rsdatafusion/core/src/extension_types/array_formatter_factory.rs renamed to datafusion/expr/src/extension_types/array_formatter_factory.rs datafusion/core/src/extension_types/array_formatter_factory.rs renamed to datafusion/expr/src/extension_types/array_formatter_factory.rs

@@ -15,10 +15,10 @@
 // specific language governing permissions and limitations
 // under the License.
 
+use crate::registry::ExtensionTypeRegistryRef;
 use arrow::array::Array;
 use arrow::util::display::{ArrayFormatter, ArrayFormatterFactory, FormatOptions};
 use arrow_schema::{ArrowError, Field};
-use datafusion_expr::registry::ExtensionTypeRegistryRef;
 
 /// A factory for creating [`ArrayFormatter`]s that checks whether a registered extension type can
 /// format a given array based on its metadata.

…tafusion/core/src/extension_types/mod.rs …tafusion/expr/src/extension_types/mod.rsdatafusion/core/src/extension_types/mod.rs renamed to datafusion/expr/src/extension_types/mod.rs datafusion/core/src/extension_types/mod.rs renamed to datafusion/expr/src/extension_types/mod.rs

@@ -52,6 +52,7 @@ pub mod expr;
 pub mod expr_fn;
 pub mod expr_rewriter;
 pub mod expr_schema;
+pub mod extension_types;
 pub mod function;
 pub mod select_expr;
 pub mod groups_accumulator {