From 4863a03c02017cfbd62b29133063d78facad2f3b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Marko=20Milenkovi=C4=87?=
 <milenkovicm@users.noreply.github.com>
Date: Sun, 24 Mar 2024 13:56:44 +0000
Subject: [PATCH] Make it easier to register configuration extension ... ...
 options

closes #9529
---
 datafusion/execution/src/config.rs | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/datafusion/execution/src/config.rs b/datafusion/execution/src/config.rs
index 360bac71c510c..0a7a87c7d81a6 100644
--- 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