reduce AbortHelper

Signed-off-by: zenghua <[email protected]>
Ceng23333 · Feb 18, 2024 · 500ed3a · 500ed3a
1 parent 00e4528
commit 500ed3a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/rust/lakesoul-datafusion/src/datasource/file_format/metadata_format.rs b/rust/lakesoul-datafusion/src/datasource/file_format/metadata_format.rs
@@ -12,7 +12,6 @@ use datafusion::common::{FileType, Statistics};
 use datafusion::error::DataFusionError;
 use datafusion::execution::TaskContext;
 use datafusion::physical_expr::PhysicalSortExpr;
-use datafusion::physical_plan::common::AbortOnDropSingle;
 use datafusion::physical_plan::stream::RecordBatchStreamAdapter;
 use datafusion::physical_plan::{DisplayAs, DisplayFormatType, Distribution, Partitioning, SendableRecordBatchStream};
 use datafusion::scalar::ScalarValue;
@@ -392,12 +391,12 @@ impl ExecutionPlan for LakeSoulHashSinkExec {
             schema: self.table_info().table_namespace.clone().into(),
             table: self.table_info().table_name.clone().into(),
         };
-        let join_handle = AbortOnDropSingle::new(tokio::spawn(Self::wait_for_commit(
+        let join_handle = tokio::spawn(Self::wait_for_commit(
             join_handles,
             self.metadata_client(),
             table_ref.to_string(),
             partitioned_file_path_and_row_count,
-        )));
+        ));
 
         // });
 

diff --git a/rust/lakesoul-datafusion/src/lakesoul_table/mod.rs b/rust/lakesoul-datafusion/src/lakesoul_table/mod.rs
@@ -7,7 +7,6 @@ pub mod helpers;
 use std::{ops::Deref, sync::Arc};
 
 use arrow::datatypes::{SchemaRef, Schema};
-use datafusion::dataframe;
 use datafusion::sql::TableReference;
 use datafusion::{
     dataframe::DataFrame,