Skip to content

Commit 6c71761

Browse files
1 parent 919ef83 commit 6c71761

File tree

1 file changed

+2
-3
lines changed
  • rust/experimental/query_engine/engine-columnar/src/pipeline

1 file changed

+2
-3
lines changed

rust/experimental/query_engine/engine-columnar/src/pipeline/filter.rs

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
// Copyright The OpenTelemetry Authors
22
// SPDX-License-Identifier: Apache-2.0
33

4-
use std::collections::HashSet;
54
use std::sync::Arc;
65

76
use arrow::array::{
@@ -734,7 +733,7 @@ impl FilterPipelineStage {
734733

735734
// build the selection vector for the child record batch. This uses common code shared
736735
// with the filter processor
737-
let id_mask: HashSet<u16> = id_col.iter().flatten().collect();
736+
let id_mask = id_col.iter().flatten().map(|i| i as u32).collect();
738737
let child_parent_ids =
739738
child_rb
740739
.column_by_name(consts::PARENT_ID)
@@ -743,7 +742,7 @@ impl FilterPipelineStage {
743742
})?;
744743

745744
let child_selection_vec =
746-
build_uint16_id_filter(child_parent_ids, id_mask).map_err(|e| {
745+
build_uint16_id_filter(child_parent_ids, &id_mask).map_err(|e| {
747746
Error::ExecutionError {
748747
cause: format!("error filtering child batch {:?}", e),
749748
}

0 commit comments

Comments
 (0)