diff --git a/rust/datafusion/src/logical_plan/mod.rs b/rust/datafusion/src/logical_plan/mod.rs index d927a61c23d2e..ed21ca8b8d51a 100644 --- a/rust/datafusion/src/logical_plan/mod.rs +++ b/rust/datafusion/src/logical_plan/mod.rs @@ -21,18 +21,13 @@ //! Logical query plans can then be optimized and executed directly, or translated into //! physical query plans and executed. -use std::collections::HashSet; - -use crate::error::Result; -use crate::physical_plan::udaf::AggregateUDF; -use crate::physical_plan::udf::ScalarUDF; - mod builder; mod display; mod expr; mod extension; mod operators; mod plan; +mod registry; pub use builder::LogicalPlanBuilder; pub use display::display_schema; @@ -43,15 +38,4 @@ pub use expr::{ pub use extension::UserDefinedLogicalNode; pub use operators::Operator; pub use plan::{LogicalPlan, PlanType, PlanVisitor, StringifiedPlan, TableSource}; - -/// A registry knows how to build logical expressions out of user-defined function' names -pub trait FunctionRegistry { - /// Set of all available udfs. - fn udfs(&self) -> HashSet; - - /// Returns a reference to the udf named `name`. - fn udf(&self, name: &str) -> Result<&ScalarUDF>; - - /// Returns a reference to the udaf named `name`. - fn udaf(&self, name: &str) -> Result<&AggregateUDF>; -} +pub use registry::FunctionRegistry; diff --git a/rust/datafusion/src/logical_plan/registry.rs b/rust/datafusion/src/logical_plan/registry.rs new file mode 100644 index 0000000000000..c115048782307 --- /dev/null +++ b/rust/datafusion/src/logical_plan/registry.rs @@ -0,0 +1,34 @@ +// Licensed to the Apache Software Foundation (ASF) under one +// or more contributor license agreements. See the NOTICE file +// distributed with this work for additional information +// regarding copyright ownership. The ASF licenses this file +// to you under the Apache License, Version 2.0 (the +// "License"); you may not use this file except in compliance +// with the License. You may obtain a copy of the License at +// +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, +// software distributed under the License is distributed on an +// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY +// KIND, either express or implied. See the License for the +// specific language governing permissions and limitations +// under the License. + +use std::collections::HashSet; + +use crate::error::Result; +use crate::physical_plan::udaf::AggregateUDF; +use crate::physical_plan::udf::ScalarUDF; + +/// A registry knows how to build logical expressions out of user-defined function' names +pub trait FunctionRegistry { + /// Set of all available udfs. + fn udfs(&self) -> HashSet; + + /// Returns a reference to the udf named `name`. + fn udf(&self, name: &str) -> Result<&ScalarUDF>; + + /// Returns a reference to the udaf named `name`. + fn udaf(&self, name: &str) -> Result<&AggregateUDF>; +}