chore: add function to set prefer_existing_union (apache#10322)

influxdata · May 1, 2024 · 0f0d2a3 · 0f0d2a3
1 parent 0f588e9
commit 0f0d2a3
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/datafusion/execution/src/config.rs b/datafusion/execution/src/config.rs
@@ -331,6 +331,14 @@ impl SessionConfig {
         self
     }
 
+    /// Prefer existing union (true). See [prefer_existing_union] for more details
+    ///
+    /// [prefer_existing_union]: datafusion_common::config::OptimizerOptions::prefer_existing_union
+    pub fn with_prefer_existing_union(mut self, enabled: bool) -> Self {
+        self.options.optimizer.prefer_existing_union = enabled;
+        self
+    }
+
     /// Enables or disables the use of pruning predicate for parquet readers to skip row groups
     pub fn with_parquet_pruning(mut self, enabled: bool) -> Self {
         self.options.execution.parquet.pruning = enabled;