Table Scan Performance Tests

.github/workflows/ci.yml

@@ -110,10 +110,10 @@ jobs:
           rust-version: ${{ env.rust_msrv }}
 
       - name: Test
-        run: cargo test --no-fail-fast --all-targets --all-features --workspace
+        run: cargo test --no-fail-fast --lib --examples --tests --all-features --workspace
 
       - name: Async-std Test
-        run: cargo test --no-fail-fast --all-targets --no-default-features  --features "async-std" --features "storage-all" --workspace
+        run: cargo test --no-fail-fast --lib --examples --tests --no-default-features  --features "async-std" --features "storage-all" --workspace
 
       - name: Doc Test
         run: cargo test --no-fail-fast --doc --all-features --workspace

.gitignore

@@ -26,3 +26,6 @@ dist/*
 *.pyc
 *.whl
 *.tar.gz
+
+crates/iceberg/testdata/performance/raw_data/*
+crates/iceberg/testdata/performance/warehouse

Makefile

@@ -50,7 +50,7 @@ check-toml: install-taplo-cli
 check: check-fmt check-clippy cargo-sort check-toml cargo-machete
 
 doc-test:
-	cargo test --no-fail-fast --doc --all-features --workspace
+	cargo test --no-fail-fast --doc --lib --examples '*' --tests '*' --workspace
 
 unit-test: doc-test
 	cargo test --no-fail-fast --lib --all-features --workspace

crates/iceberg/Cargo.toml

@@ -82,10 +82,21 @@ url = { workspace = true }
 uuid = { workspace = true }
 
 [dev-dependencies]
+criterion = { version = "0.5", features = ["async_tokio", "async_futures"] }
 ctor = { workspace = true }
+futures-util = "0.3"
 iceberg-catalog-memory = { workspace = true }
+iceberg-catalog-rest = { path = "../catalog/rest" }
 iceberg_test_utils = { path = "../test_utils", features = ["tests"] }
 pretty_assertions = { workspace = true }
 rand = { workspace = true }
 tempfile = { workspace = true }
 tera = { workspace = true }
+
+[[bench]]
+name = "table_scan_plan_files"
+harness = false
+
+[[bench]]
+name = "table_scan_execute_query"
+harness = false

crates/iceberg/benches/table_scan_execute_query.rs

@@ -0,0 +1,122 @@
+// Licensed to the Apache Software Foundation (ASF) under one
+// or more contributor license agreements.  See the NOTICE file
+// distributed with this work for additional information
+// regarding copyright ownership.  The ASF licenses this file
+// to you under the Apache License, Version 2.0 (the
+// "License"); you may not use this file except in compliance
+// with the License.  You may obtain a copy of the License at
+//
+//   http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing,
+// software distributed under the License is distributed on an
+// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+// KIND, either express or implied.  See the License for the
+// specific language governing permissions and limitations
+// under the License.
+
+use criterion::*;
+use iceberg::expr::Reference;
+use iceberg::spec::Datum;
+use tokio::runtime::Runtime;
+
+mod utils;
+use utils::{create_file_plan, create_task_stream, exec_plan, setup};
+
+pub fn bench_read_all_files_all_rows(c: &mut Criterion) {
+    let runtime = Runtime::new().unwrap();
+    let table = setup(&runtime);
+    let scan = table.scan().build().unwrap();
+    let tasks = create_file_plan(&runtime, scan);
+
+    c.bench_function("scan: read (all files, all rows)", |b| {
+        b.to_async(&runtime).iter_batched(
+            || create_task_stream(tasks.clone()),
+            |plan| exec_plan(table.clone(), plan),
+            BatchSize::SmallInput,
+        )
+    });
+}
+
+pub fn bench_read_all_files_some_rows(c: &mut Criterion) {
+    let runtime = Runtime::new().unwrap();
+    let table = setup(&runtime);
+    let scan = table
+        .scan()
+        .with_filter(Reference::new("fare_amount").greater_than(Datum::double(50.0)))
+        .build()
+        .unwrap();
+    let tasks = create_file_plan(&runtime, scan);
+
+    c.bench_function("scan: read (all files, some rows)", |b| {
+        b.to_async(&runtime).iter_batched(
+            || create_task_stream(tasks.clone()),
+            |plan| exec_plan(table.clone(), plan),
+            BatchSize::SmallInput,
+        )
+    });
+}
+
+pub fn bench_read_some_files_all_rows(c: &mut Criterion) {
+    let runtime = Runtime::new().unwrap();
+    let table = setup(&runtime);
+    let scan = table
+        .scan()
+        .with_filter(
+            Reference::new("tpep_pickup_datetime")
+                .greater_than_or_equal_to(
+                    Datum::timestamptz_from_str("2024-02-01T00:00:00.000 UTC").unwrap(),
+                )
+                .and(Reference::new("tpep_pickup_datetime").less_than(
+                    Datum::timestamptz_from_str("2024-02-02T00:00:00.000 UTC").unwrap(),
+                )),
+        )
+        .build()
+        .unwrap();
+    let tasks = create_file_plan(&runtime, scan);
+
+    c.bench_function("scan: read (some files, all rows)", |b| {
+        b.to_async(&runtime).iter_batched(
+            || create_task_stream(tasks.clone()),
+            |plan| exec_plan(table.clone(), plan),
+            BatchSize::SmallInput,
+        )
+    });
+}
+
+pub fn bench_read_some_files_some_rows(c: &mut Criterion) {
+    let runtime = Runtime::new().unwrap();
+    let table = setup(&runtime);
+    let scan =
+        table
+            .scan()
+            .with_filter(
+                Reference::new("tpep_pickup_datetime")
+                    .greater_than_or_equal_to(
+                        Datum::timestamptz_from_str("2024-02-01T00:00:00.000 UTC").unwrap(),
+                    )
+                    .and(Reference::new("tpep_pickup_datetime").less_than(
+                        Datum::timestamptz_from_str("2024-02-02T00:00:00.000 UTC").unwrap(),
+                    ))
+                    .and(Reference::new("fare_amount").greater_than(Datum::double(50.0))),
+            )
+            .build()
+            .unwrap();
+    let tasks = create_file_plan(&runtime, scan);
+
+    c.bench_function("scan: read (some files, some rows)", |b| {
+        b.to_async(&runtime).iter_batched(
+            || create_task_stream(tasks.clone()),
+            |plan| exec_plan(table.clone(), plan),
+            BatchSize::SmallInput,
+        )
+    });
+}
+
+criterion_group! {
+    name = benches;
+    config = Criterion::default().sample_size(10);
+    targets = bench_read_some_files_some_rows, bench_read_some_files_all_rows, bench_read_all_files_some_rows, bench_read_all_files_all_rows
+}
+
+criterion_main!(benches);

crates/iceberg/benches/table_scan_plan_files.rs

@@ -0,0 +1,139 @@
+// Licensed to the Apache Software Foundation (ASF) under one
+// or more contributor license agreements.  See the NOTICE file
+// distributed with this work for additional information
+// regarding copyright ownership.  The ASF licenses this file
+// to you under the Apache License, Version 2.0 (the
+// "License"); you may not use this file except in compliance
+// with the License.  You may obtain a copy of the License at
+//
+//   http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing,
+// software distributed under the License is distributed on an
+// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+// KIND, either express or implied.  See the License for the
+// specific language governing permissions and limitations
+// under the License.
+
+use criterion::*;
+use futures_util::StreamExt;
+use iceberg::expr::Reference;
+use iceberg::spec::Datum;
+use iceberg::table::Table;
+use tokio::runtime::Runtime;
+mod utils;
+use utils::setup;
+
+async fn all_files_all_rows(table: &Table) {
+    let scan = table.scan().build().unwrap();
+    let mut stream = scan.plan_files().await.unwrap();
+
+    while let Some(item) = stream.next().await {
+        black_box(item.unwrap());
+    }
+}
+
+async fn one_file_all_rows(table: &Table) {
+    let scan = table
+        .scan()
+        .with_filter(
+            Reference::new("tpep_pickup_datetime")
+                .greater_than_or_equal_to(
+                    Datum::timestamptz_from_str("2024-02-01T00:00:00.000 UTC").unwrap(),
+                )
+                .and(Reference::new("tpep_pickup_datetime").less_than(
+                    Datum::timestamptz_from_str("2024-02-02T00:00:00.000 UTC").unwrap(),
+                )),
+        )
+        .build()
+        .unwrap();
+    let mut stream = scan.plan_files().await.unwrap();
+
+    while let Some(item) = stream.next().await {
+        black_box(item.unwrap());
+    }
+}
+
+async fn all_files_some_rows(table: &Table) {
+    let scan = table
+        .scan()
+        .with_filter(Reference::new("fare_amount").equal_to(Datum::double(50.0)))
+        .build()
+        .unwrap();
+    let mut stream = scan.plan_files().await.unwrap();
+
+    while let Some(item) = stream.next().await {
+        black_box(item.unwrap());
+    }
+}
+
+async fn one_file_some_rows(table: &Table) {
+    let scan =
+        table
+            .scan()
+            .with_filter(
+                Reference::new("tpep_pickup_datetime")
+                    .greater_than_or_equal_to(
+                        Datum::timestamptz_from_str("2024-02-01T00:00:00.000 UTC").unwrap(),
+                    )
+                    .and(Reference::new("tpep_pickup_datetime").less_than(
+                        Datum::timestamptz_from_str("2024-02-02T00:00:00.000 UTC").unwrap(),
+                    ))
+                    .and(Reference::new("fare_amount").equal_to(Datum::double(50.0))),
+            )
+            .build()
+            .unwrap();
+    let mut stream = scan.plan_files().await.unwrap();
+
+    while let Some(item) = stream.next().await {
+        black_box(item.unwrap());
+    }
+}
+
+pub fn bench_all_files_all_rows(c: &mut Criterion) {
+    let runtime = Runtime::new().unwrap();
+    let table = setup(&runtime);
+    println!("setup complete");
+
+    c.bench_function("scan: plan (all files, all rows)", |b| {
+        b.to_async(&runtime).iter(|| all_files_all_rows(&table))
+    });
+}
+
+pub fn bench_one_file_all_rows(c: &mut Criterion) {
+    let runtime = Runtime::new().unwrap();
+    let table = setup(&runtime);
+    println!("setup complete");
+
+    c.bench_function("scan: plan (one file, all rows)", |b| {
+        b.to_async(&runtime).iter(|| one_file_all_rows(&table))
+    });
+}
+
+pub fn bench_all_files_some_rows(c: &mut Criterion) {
+    let runtime = Runtime::new().unwrap();
+    let table = setup(&runtime);
+    println!("setup complete");
+
+    c.bench_function("scan: plan (all files, some rows)", |b| {
+        b.to_async(&runtime).iter(|| all_files_some_rows(&table))
+    });
+}
+
+pub fn bench_one_file_some_rows(c: &mut Criterion) {
+    let runtime = Runtime::new().unwrap();
+    let table = setup(&runtime);
+    println!("setup complete");
+
+    c.bench_function("scan: plan (one file, some rows)", |b| {
+        b.to_async(&runtime).iter(|| one_file_some_rows(&table))
+    });
+}
+
+criterion_group! {
+    name = benches;
+    config = Criterion::default().sample_size(10);
+    targets = bench_all_files_all_rows, bench_all_files_some_rows, bench_one_file_all_rows, bench_one_file_some_rows
+}
+
+criterion_main!(benches);