Rename the pushdown config into a positive boolean value (#537)

spiraldb · Aug 2, 2024 · 89e918b · 89e918b
1 parent f5a5466
commit 89e918b
Show file tree

Hide file tree

Showing 5 changed files with 15 additions and 15 deletions.
diff --git a/bench-vortex/benches/datafusion_benchmark.rs b/bench-vortex/benches/datafusion_benchmark.rs
@@ -158,13 +158,13 @@ fn bench_arrow<M: Measurement>(mut group: BenchmarkGroup<M>, session: &SessionCo
 fn bench_vortex<M: Measurement>(
     mut group: BenchmarkGroup<M>,
     session: &SessionContext,
-    disable_pushdown: bool,
+    enable_pushdown: bool,
     compress: bool,
 ) {
     let vortex_dataset = toy_dataset_vortex(compress);
     let vortex_table = Arc::new(VortexMemTable::new(
         vortex_dataset,
-        VortexMemTableOptions::default().with_disable_pushdown(disable_pushdown),
+        VortexMemTableOptions::default().with_pushdown(enable_pushdown),
     ));
 
     measure_provider(&mut group, session, vortex_table);

diff --git a/bench-vortex/benches/tpch_benchmark.rs b/bench-vortex/benches/tpch_benchmark.rs
@@ -13,15 +13,15 @@ fn benchmark(c: &mut Criterion) {
         .block_on(load_datasets(
             &data_dir,
             Format::Vortex {
-                disable_pushdown: true,
+                enable_pushdown: false,
             },
         ))
         .unwrap();
     let vortex_ctx = runtime
         .block_on(load_datasets(
             &data_dir,
             Format::Vortex {
-                disable_pushdown: false,
+                enable_pushdown: true,
             },
         ))
         .unwrap();

diff --git a/bench-vortex/src/bin/tpch_benchmark.rs b/bench-vortex/src/bin/tpch_benchmark.rs
@@ -20,7 +20,7 @@ async fn main() {
         Format::Arrow,
         Format::Parquet,
         Format::Vortex {
-            disable_pushdown: false,
+            enable_pushdown: true,
         },
     ];
 

diff --git a/bench-vortex/src/tpch/mod.rs b/bench-vortex/src/tpch/mod.rs
@@ -22,7 +22,7 @@ pub enum Format {
     Csv,
     Arrow,
     Parquet,
-    Vortex { disable_pushdown: bool },
+    Vortex { enable_pushdown: bool },
 }
 
 // Generate table dataset.
@@ -51,14 +51,14 @@ pub async fn load_datasets<P: AsRef<Path>>(
                     register_parquet(&context, stringify!($name), &$name, $schema).await
                 }
                 Format::Vortex {
-                    disable_pushdown, ..
+                    enable_pushdown, ..
                 } => {
                     register_vortex(
                         &context,
                         stringify!($name),
                         &$name,
                         $schema,
-                        disable_pushdown,
+                        enable_pushdown,
                     )
                     .await
                 }
@@ -172,7 +172,7 @@ async fn register_vortex(
     name: &str,
     file: &Path,
     schema: &Schema,
-    disable_pushdown: bool,
+    enable_pushdown: bool,
 ) -> anyhow::Result<()> {
     let record_batches = session
         .read_csv(
@@ -201,7 +201,7 @@ async fn register_vortex(
     session.register_vortex_opts(
         name,
         chunked_array,
-        VortexMemTableOptions::default().with_disable_pushdown(disable_pushdown),
+        VortexMemTableOptions::default().with_pushdown(enable_pushdown),
     )?;
 
     Ok(())

diff --git a/vortex-datafusion/src/lib.rs b/vortex-datafusion/src/lib.rs
@@ -50,12 +50,12 @@ const SUPPORTED_BINARY_OPS: &[Operator] = &[
 /// Optional configurations to pass when loading a [VortexMemTable].
 #[derive(Default, Debug, Clone)]
 pub struct VortexMemTableOptions {
-    pub disable_pushdown: bool,
+    pub enable_pushdown: bool,
 }
 
 impl VortexMemTableOptions {
-    pub fn with_disable_pushdown(mut self, disable_pushdown: bool) -> Self {
-        self.disable_pushdown = disable_pushdown;
+    pub fn with_pushdown(mut self, enable_pushdown: bool) -> Self {
+        self.enable_pushdown = enable_pushdown;
         self
     }
 }
@@ -248,7 +248,7 @@ impl TableProvider for VortexMemTable {
     ) -> DFResult<Vec<TableProviderFilterPushDown>> {
         // In the case the caller has configured this provider with filter pushdown disabled,
         // do not attempt to apply any filters at scan time.
-        if self.options.disable_pushdown {
+        if !self.options.enable_pushdown {
             return Ok(filters
                 .iter()
                 .map(|_| TableProviderFilterPushDown::Unsupported)
@@ -548,7 +548,7 @@ mod test {
                 presidents_array(),
                 // Disable pushdown. We run this test to make sure that the naive codepath also
                 // produces correct results and does not panic anywhere.
-                VortexMemTableOptions::default().with_disable_pushdown(true),
+                VortexMemTableOptions::default().with_pushdown(false),
             )
             .unwrap();