WIP: MetadataCacheFactory::make_session_config

Author: srh

rust/cubestore/cubestore/src/queryplanner/metadata_cache.rs

@@ -4,6 +4,7 @@ use datafusion::datasource::physical_plan::{FileMeta, ParquetFileReaderFactory};
 use datafusion::parquet::arrow::async_reader::AsyncFileReader;
 use datafusion::parquet::file::metadata::ParquetMetaData;
 use datafusion::physical_plan::metrics::ExecutionPlanMetricsSet;
+use datafusion::prelude::SessionConfig;
 use futures_util::future::BoxFuture;
 use futures_util::FutureExt;
 use std::fmt;
@@ -22,6 +23,9 @@ pub trait MetadataCacheFactory: Sync + Send {
         max_capacity: u64,
         time_to_idle: Duration,
     ) -> Arc<dyn ParquetFileReaderFactory>;
+    fn make_session_config(&self) -> SessionConfig {
+        SessionConfig::new()
+    }
 }
 /// Default MetadataCache, does not cache anything
 #[derive(Debug)]

rust/cubestore/cubestore/src/queryplanner/mod.rs

@@ -245,8 +245,7 @@ impl QueryPlannerImpl {
 }
 
 impl QueryPlannerImpl {
-    pub fn make_execution_context() -> SessionContext {
-        let config = SessionConfig::new();
+    pub fn execution_context_helper(config: SessionConfig) -> SessionContext {
         let context = SessionContext::new_with_config(config);
         // TODO upgrade DF: build SessionContexts consistently -- that now means check all appropriate SessionContext constructors use this make_execution_context or execution_context function.
         for udaf in registerable_aggregate_udfs() {
@@ -267,8 +266,12 @@ impl QueryPlannerImpl {
         context
     }
 
+    pub fn make_execution_context() -> SessionContext {
+        Self::execution_context_helper(SessionConfig::new())
+    }
+
     async fn execution_context(&self) -> Result<Arc<SessionContext>, CubeError> {
-        Ok(Arc::new(Self::make_execution_context()))
+        Ok(Arc::new(Self::execution_context_helper(self.metadata_cache_factory.make_session_config())))
     }
 }