Skip to content

Commit 64955b7

Browse files
committed
no need for -1s
1 parent 2f419f1 commit 64955b7

File tree

1 file changed

+12
-19
lines changed

1 file changed

+12
-19
lines changed

kernel/src/client/arrow_utils.rs

Lines changed: 12 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -25,16 +25,13 @@ pub(crate) fn get_requested_indices(
2525
.fields()
2626
.iter()
2727
.enumerate()
28-
.map(
29-
|(parquet_position, field)| match requested_schema.index_of(field.name()) {
30-
Some(index) => {
31-
found_count += 1;
32-
indicies[index] = parquet_position;
33-
index as i32
34-
}
35-
None => -1,
36-
},
37-
)
28+
.filter_map(|(parquet_position, field)| {
29+
requested_schema.index_of(field.name()).map(|index| {
30+
found_count += 1;
31+
indicies[index] = parquet_position;
32+
index as i32
33+
})
34+
})
3835
.collect();
3936
if found_count != requested_len {
4037
return Err(Error::generic(
@@ -85,15 +82,11 @@ pub(crate) fn reorder_record_batch(
8582
let mut fields = Vec::with_capacity(indicies.len());
8683
let reordered_columns = requested_ordering
8784
.iter()
88-
.filter_map(|index| {
89-
if *index >= 0 {
90-
let idx = *index as usize;
91-
// cheap clones of `Arc`s
92-
fields.push(input_schema.field(idx).clone());
93-
Some(input_data.column(idx).clone())
94-
} else {
95-
None
96-
}
85+
.map(|index| {
86+
let idx = *index as usize;
87+
// cheap clones of `Arc`s
88+
fields.push(input_schema.field(idx).clone());
89+
input_data.column(idx).clone()
9790
})
9891
.collect();
9992
let schema = Arc::new(ArrowSchema::new(fields));

0 commit comments

Comments
 (0)