Make it easier to register configuration extension ... (#9781)

... options closes #9529
apache · Mar 25, 2024 · 0b95577 · 0b95577
1 parent ad89ff8
commit 0b95577
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/datafusion/execution/src/config.rs b/datafusion/execution/src/config.rs
@@ -22,7 +22,10 @@ use std::{
     sync::Arc,
 };
 
-use datafusion_common::{config::ConfigOptions, Result, ScalarValue};
+use datafusion_common::{
+    config::{ConfigExtension, ConfigOptions},
+    Result, ScalarValue,
+};
 
 /// Configuration options for [`SessionContext`].
 ///
@@ -198,6 +201,12 @@ impl SessionConfig {
         self
     }
 
+    /// Insert new [ConfigExtension]
+    pub fn with_option_extension<T: ConfigExtension>(mut self, extension: T) -> Self {
+        self.options_mut().extensions.insert(extension);
+        self
+    }
+
     /// Get [`target_partitions`]
     ///
     /// [`target_partitions`]: datafusion_common::config::ExecutionOptions::target_partitions