fix(processor): try fix test failure

databendlabs · Sep 30, 2022 · 6f76776 · 6f76776
1 parent 5acc981
commit 6f76776
Show file tree

Hide file tree

Showing 5 changed files with 9 additions and 3 deletions.
diff --git a/src/query/catalog/src/table_context.rs b/src/query/catalog/src/table_context.rs
@@ -108,4 +108,6 @@ pub trait TableContext: Send + Sync {
     -> Result<Arc<dyn Table>>;
     fn get_processes_info(&self) -> Vec<ProcessInfo>;
     fn get_runtime(&self) -> Result<Arc<Runtime>>;
+
+    fn clone_inner(&self) -> Arc<dyn TableContext>;
 }
diff --git a/src/query/service/src/api/rpc/exchange/exchange_manager.rs b/src/query/service/src/api/rpc/exchange/exchange_manager.rs
@@ -668,7 +668,6 @@ impl FragmentCoordinator {
 
             match &self.payload {
                 FragmentPayload::PlanV2(plan) => {
-                    let ctx = QueryContext::create_from(ctx);
                     let pipeline_builder = PipelineBuilderV2::create(ctx);
                     self.pipeline_build_res = Some(pipeline_builder.finalize(plan)?);
                 }

diff --git a/src/query/service/src/interpreters/interpreter_insert_v2.rs b/src/query/service/src/interpreters/interpreter_insert_v2.rs
@@ -164,7 +164,6 @@ impl Interpreter for InsertInterpreterV2 {
                         _ => unreachable!(),
                     };
 
-                    table1.get_table_info();
                     let catalog = self.plan.catalog.clone();
                     let is_distributed_plan = select_plan.is_distributed_plan();
 

diff --git a/src/query/service/src/sessions/query_ctx.rs b/src/query/service/src/sessions/query_ctx.rs
@@ -386,6 +386,10 @@ impl TableContext for QueryContext {
     fn get_runtime(&self) -> Result<Arc<Runtime>> {
         self.shared.try_get_runtime()
     }
+
+    fn clone_inner(&self) -> Arc<dyn TableContext> {
+        QueryContext::create_from_shared(self.shared.clone())
+    }
 }
 
 impl TrySpawn for QueryContext {

diff --git a/src/query/storages/fuse/src/operations/read.rs b/src/query/storages/fuse/src/operations/read.rs
@@ -74,7 +74,7 @@ impl FuseTable {
     #[inline]
     pub fn do_read2(
         &self,
-        ctx: Arc<dyn TableContext>,
+        mut ctx: Arc<dyn TableContext>,
         plan: &ReadDataSourcePlan,
         pipeline: &mut Pipeline,
     ) -> Result<()> {
@@ -87,6 +87,8 @@ impl FuseTable {
         }
 
         if !lazy_init_segments.is_empty() {
+            ctx = ctx.clone_inner();
+
             let table_info = self.table_info.clone();
             let push_downs = plan.push_downs.clone();
             let query_ctx = ctx.clone();