chore: remove scan timer

This commit is contained in:
shuiyisong
2023-06-25 18:44:52 +08:00
parent a57cd604c1
commit 2d16484e46
2 changed files with 8 additions and 8 deletions

View File

@@ -116,7 +116,6 @@ impl Stream for DfRecordBatchStreamAdapter {
pub struct RecordBatchStreamAdapter {
schema: SchemaRef,
stream: DfSendableRecordBatchStream,
#[allow(dead_code)]
metrics: Option<BaselineMetrics>,
}
@@ -155,12 +154,12 @@ impl Stream for RecordBatchStreamAdapter {
type Item = Result<RecordBatch>;
fn poll_next(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Option<Self::Item>> {
// let timer = self
// .metrics
// .as_ref()
// .map(|m| m.elapsed_compute().clone())
// .unwrap_or_default();
// let _guard = timer.timer();
let timer = self
.metrics
.as_ref()
.map(|m| m.elapsed_compute().clone())
.unwrap_or_default();
let _guard = timer.timer();
match Pin::new(&mut self.stream).poll_next(cx) {
Poll::Pending => Poll::Pending,
Poll::Ready(Some(df_record_batch)) => {

View File

@@ -119,7 +119,8 @@ impl Stream for StreamWithMetricWrapper {
fn poll_next(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Option<Self::Item>> {
let this = self.get_mut();
let _timer = this.metric.elapsed_compute().timer();
// todo(shuiyisong): remove scan timer for now
// let _timer = this.metric.elapsed_compute().timer();
let poll = this.stream.poll_next_unpin(cx);
if let Poll::Ready(Option::Some(Result::Ok(record_batch))) = &poll {
this.metric.record_output(record_batch.num_rows());