Skip to content

Commit 9d26397

Browse files
authored
chore: public sql_statement_to_plan_with_context() (#7268)
Signed-off-by: Ruihang Xia <[email protected]>
1 parent a514b67 commit 9d26397

File tree

1 file changed

+14
-3
lines changed

1 file changed

+14
-3
lines changed

datafusion/sql/src/statement.rs

+14-3
Original file line numberDiff line numberDiff line change
@@ -98,11 +98,22 @@ impl<'a, S: ContextProvider> SqlToRel<'a, S> {
9898

9999
/// Generate a logical plan from an SQL statement
100100
pub fn sql_statement_to_plan(&self, statement: Statement) -> Result<LogicalPlan> {
101-
self.sql_statement_to_plan_with_context(statement, &mut PlannerContext::new())
101+
self.sql_statement_to_plan_with_context_impl(
102+
statement,
103+
&mut PlannerContext::new(),
104+
)
102105
}
103106

104107
/// Generate a logical plan from an SQL statement
105-
fn sql_statement_to_plan_with_context(
108+
pub fn sql_statement_to_plan_with_context(
109+
&self,
110+
statement: Statement,
111+
planner_context: &mut PlannerContext,
112+
) -> Result<LogicalPlan> {
113+
self.sql_statement_to_plan_with_context_impl(statement, planner_context)
114+
}
115+
116+
fn sql_statement_to_plan_with_context_impl(
106117
&self,
107118
statement: Statement,
108119
planner_context: &mut PlannerContext,
@@ -320,7 +331,7 @@ impl<'a, S: ContextProvider> SqlToRel<'a, S> {
320331
.with_prepare_param_data_types(data_types.clone());
321332

322333
// Build logical plan for inner statement of the prepare statement
323-
let plan = self.sql_statement_to_plan_with_context(
334+
let plan = self.sql_statement_to_plan_with_context_impl(
324335
*statement,
325336
&mut planner_context,
326337
)?;

0 commit comments

Comments
 (0)