Skip to content

Commit

Permalink
rename to execute streams for insertexec
Browse files Browse the repository at this point in the history
  • Loading branch information
devinjdangelo committed Aug 4, 2023
1 parent 41698bb commit a4157b1
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions datafusion/core/src/physical_plan/insert.rs
Original file line number Diff line number Diff line change
Expand Up @@ -96,7 +96,7 @@ impl InsertExec {
}
}

fn make_input_stream(
fn execute_input_stream(
&self,
partition: usize,
context: Arc<TaskContext>,
Expand Down Expand Up @@ -136,14 +136,14 @@ impl InsertExec {
}
}

fn make_all_input_streams(
fn execute_all_input_streams(
&self,
context: Arc<TaskContext>,
) -> Result<Vec<SendableRecordBatchStream>> {
let n_input_parts = self.input.output_partitioning().partition_count();
let mut streams = Vec::with_capacity(n_input_parts);
for part in 0..n_input_parts {
streams.push(self.make_input_stream(part, context.clone())?);
streams.push(self.execute_input_stream(part, context.clone())?);
}
Ok(streams)
}
Expand Down Expand Up @@ -236,7 +236,7 @@ impl ExecutionPlan for InsertExec {
"InsertExec can only be called on partition 0!".into(),
));
}
let data = self.make_all_input_streams(context.clone())?;
let data = self.execute_all_input_streams(context.clone())?;

let count_schema = self.count_schema.clone();
let sink = self.sink.clone();
Expand Down

0 comments on commit a4157b1

Please sign in to comment.