chore: safe indexing

apache · Oct 4, 2024 · d1ffc74 · d1ffc74
1 parent 0a18ba4
commit d1ffc74
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/datafusion/physical-plan/src/union.rs b/datafusion/physical-plan/src/union.rs
@@ -475,8 +475,12 @@ fn union_schema(inputs: &[Arc<dyn ExecutionPlan>]) -> SchemaRef {
                 .filter_map(|input| {
                     if input.schema().fields().len() > i {
                         let field = input.schema().field(i).clone();
-                        let right_hand_metdata =
-                            inputs[1].schema().field(i).metadata().clone();
+                        let right_hand_metdata = inputs
+                            .get(1)
+                            .map(|right_input| {
+                                right_input.schema().field(i).metadata().clone()
+                            })
+                            .unwrap_or_default();
                         let mut metadata = field.metadata().clone();
                         metadata.extend(right_hand_metdata);
                         Some(field.with_metadata(metadata))