Make the struct function return the correct data type.

apache · Jun 8, 2023 · 5228bb3 · 5228bb3
1 parent 9c13a77
commit 5228bb3
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/datafusion/expr/src/function.rs b/datafusion/expr/src/function.rs
@@ -352,7 +352,14 @@ pub fn return_type(
             _ => Ok(DataType::Float64),
         },
 
-        BuiltinScalarFunction::Struct => Ok(DataType::Struct(Fields::empty())),
+        BuiltinScalarFunction::Struct => {
+            let return_fields = input_expr_types
+                .iter()
+                .enumerate()
+                .map(|(pos, dt)| Field::new(format!("c{pos}"), dt.clone(), true))
+                .collect::<Vec<Field>>();
+            Ok(DataType::Struct(Fields::from(return_fields)))
+        }
 
         BuiltinScalarFunction::Atan2 => match &input_expr_types[0] {
             DataType::Float32 => Ok(DataType::Float32),