Skip to content

Commit

Permalink
fix comm
Browse files Browse the repository at this point in the history
  • Loading branch information
xxhZs committed Aug 27, 2024
1 parent d82c8ea commit d97cc35
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 5 deletions.
6 changes: 3 additions & 3 deletions src/frontend/src/scheduler/distributed/stage.rs
Original file line number Diff line number Diff line change
Expand Up @@ -1074,22 +1074,22 @@ impl StageRunner {
}
PlanNodeType::BatchIcebergScan => {
let node_body = execution_plan_node.node.clone();
let NodeBody::IcebergScan(mut iceberg_source_node) = node_body else {
let NodeBody::IcebergScan(mut iceberg_scan_node) = node_body else {
unreachable!();
};

let partition = partition
.expect("no partition info for seq scan")
.into_source()
.expect("PartitionInfo should be SourcePartitionInfo");
iceberg_source_node.split = partition
iceberg_scan_node.split = partition
.into_iter()
.map(|split| split.encode_to_bytes().into())
.collect_vec();
PbPlanNode {
children: vec![],
identity,
node_body: Some(NodeBody::IcebergScan(iceberg_source_node)),
node_body: Some(NodeBody::IcebergScan(iceberg_scan_node)),
}
}
_ => {
Expand Down
4 changes: 2 additions & 2 deletions src/frontend/src/scheduler/local.rs
Original file line number Diff line number Diff line change
Expand Up @@ -580,12 +580,12 @@ impl LocalQueryExecution {
PlanNodeType::BatchIcebergScan => {
let mut node_body = execution_plan_node.node.clone();
match &mut node_body {
NodeBody::IcebergScan(ref mut iceberg_source_node) => {
NodeBody::IcebergScan(ref mut iceberg_scan_node) => {
if let Some(partition) = partition {
let partition = partition
.into_source()
.expect("PartitionInfo should be SourcePartitionInfo here");
iceberg_source_node.split = partition
iceberg_scan_node.split = partition
.into_iter()
.map(|split| split.encode_to_bytes().into())
.collect_vec();
Expand Down

0 comments on commit d97cc35

Please sign in to comment.