This is an automated email from the ASF dual-hosted git repository.
alamb pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/datafusion.git
The following commit(s) were added to refs/heads/main by this push:
new ed12f113cb Minor: allow to build RuntimeEnv from RuntimeConfig (#12151)
ed12f113cb is described below
commit ed12f113cb1d33a1eea80c41787a93bc52f196c7
Author: theirix <[email protected]>
AuthorDate: Mon Aug 26 20:13:34 2024 +0100
Minor: allow to build RuntimeEnv from RuntimeConfig (#12151)
* Allow to build RuntimeEnv from RuntimeConfig
* Fix formatting
---
datafusion/core/tests/fuzz_cases/sort_fuzz.rs | 9 +++++----
datafusion/execution/src/runtime_env.rs | 5 +++++
2 files changed, 10 insertions(+), 4 deletions(-)
diff --git a/datafusion/core/tests/fuzz_cases/sort_fuzz.rs
b/datafusion/core/tests/fuzz_cases/sort_fuzz.rs
index f4b4f16aa1..eda306dd3d 100644
--- a/datafusion/core/tests/fuzz_cases/sort_fuzz.rs
+++ b/datafusion/core/tests/fuzz_cases/sort_fuzz.rs
@@ -22,7 +22,7 @@ use arrow::{
compute::SortOptions,
record_batch::RecordBatch,
};
-use datafusion::execution::runtime_env::{RuntimeConfig, RuntimeEnv};
+use datafusion::execution::runtime_env::RuntimeConfig;
use datafusion::physical_plan::expressions::PhysicalSortExpr;
use datafusion::physical_plan::memory::MemoryExec;
use datafusion::physical_plan::sorts::sort::SortExec;
@@ -136,9 +136,10 @@ impl SortTest {
.sort_spill_reservation_bytes,
);
- let runtime_config = RuntimeConfig::new()
- .with_memory_pool(Arc::new(GreedyMemoryPool::new(pool_size)));
- let runtime = Arc::new(RuntimeEnv::new(runtime_config).unwrap());
+ let runtime_env = RuntimeConfig::new()
+ .with_memory_pool(Arc::new(GreedyMemoryPool::new(pool_size)))
+ .build();
+ let runtime = Arc::new(runtime_env.unwrap());
SessionContext::new_with_config_rt(session_config, runtime)
} else {
SessionContext::new_with_config(session_config)
diff --git a/datafusion/execution/src/runtime_env.rs
b/datafusion/execution/src/runtime_env.rs
index 4202465955..bddce81f53 100644
--- a/datafusion/execution/src/runtime_env.rs
+++ b/datafusion/execution/src/runtime_env.rs
@@ -228,4 +228,9 @@ impl RuntimeConfig {
pub fn with_temp_file_path(self, path: impl Into<PathBuf>) -> Self {
self.with_disk_manager(DiskManagerConfig::new_specified(vec![path.into()]))
}
+
+ /// Build a `RuntimeEnv` object from the configuration
+ pub fn build(self) -> Result<RuntimeEnv> {
+ RuntimeEnv::new(self)
+ }
}
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
