NGA-TRAN commented on code in PR #4561:
URL: https://github.com/apache/arrow-datafusion/pull/4561#discussion_r1044868371
##########
datafusion/expr/src/logical_plan/plan.rs:
##########
@@ -364,6 +367,42 @@ impl LogicalPlan {
) -> Result<LogicalPlan, DataFusionError> {
from_plan(self, &self.expressions(), inputs)
}
+
+ /// Convert a prepare logical plan into its inner logical plan with all
params replaced with their corresponding values
+ pub fn with_param_values(
+ self,
+ param_values: Vec<ScalarValue>,
+ ) -> Result<LogicalPlan, DataFusionError> {
+ match self {
+ LogicalPlan::Prepare(prepare_lp) => {
+ // Verify if the number of params matches the number of values
+ if prepare_lp.data_types.len() != param_values.len() {
+ return Err(DataFusionError::Internal(format!(
+ "Expected {} parameters, got {}",
+ prepare_lp.data_types.len(),
+ param_values.len()
+ )));
+ }
+
+ // Verify if the types of the params matches the types of the
values
+ for (param_type, value) in
+ prepare_lp.data_types.iter().zip(param_values.iter())
+ {
+ if *param_type != value.get_datatype() {
Review Comment:
Done
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
