Skip to content

Commit de5c12c

Browse files
committed
Upgrade to the latest delta-rs which has some useful performance improvements
We should see lower memory in many cases now, which will help for large tables Signed-off-by: R Tyler Croy <rtyler@scribd.com>
1 parent 93c8c3b commit de5c12c

File tree

3 files changed

+15
-17
lines changed

3 files changed

+15
-17
lines changed

lambdas/query-metrics/Cargo.toml

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
[package]
22
name = "query-metrics"
3-
version = "0.4.0"
4-
edition = "2021"
3+
version = "0.5.0"
4+
edition = "2024"
55

66
[[bin]]
77
name = "query-metrics"
@@ -18,8 +18,7 @@ aws-sdk-cloudwatch = "1.11.0"
1818
aws-sdk-config = "1.11.0"
1919
aws_lambda_events = { version = "0.12.0" }
2020
base64 = "0.21.7"
21-
deltalake-core = { version = "0.19.0", features = ["datafusion"] }
22-
deltalake-aws = { version = "0.1.4" }
21+
deltalake = { version = "0.29.0", features = ["datafusion", "s3"] }
2322

2423
lambda_runtime = "0.8.3"
2524
serde = { version = "1.0.195", features = ["derive"] }
@@ -28,4 +27,3 @@ tokio = { version = "1", features = ["macros"] }
2827
tracing = { version = "0.1", features = ["log"] }
2928
tracing-subscriber = { version = "0.3", default-features = false, features = ["fmt", "env-filter"] }
3029
url = { version = "2.5.0", features = ["serde"] }
31-

lambdas/query-metrics/src/cli.rs

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -4,17 +4,17 @@
44
use std::collections::HashMap;
55
use std::sync::Arc;
66

7-
use deltalake_core::arrow::util::pretty::print_batches;
8-
use deltalake_core::arrow::{array::PrimitiveArray, datatypes::Int64Type};
9-
use deltalake_core::datafusion::common::*;
10-
use deltalake_core::datafusion::execution::context::SessionContext;
7+
use deltalake::arrow::util::pretty::print_batches;
8+
use deltalake::arrow::{array::PrimitiveArray, datatypes::Int64Type};
9+
use deltalake::datafusion::common::*;
10+
use deltalake::datafusion::execution::context::SessionContext;
1111
use tracing::log::*;
1212

1313
mod config;
1414

1515
#[tokio::main]
1616
async fn main() -> anyhow::Result<()> {
17-
deltalake_aws::register_handlers(None);
17+
deltalake::aws::register_handlers(None);
1818

1919
tracing_subscriber::fmt()
2020
.with_env_filter(tracing_subscriber::EnvFilter::from_default_env())
@@ -27,7 +27,7 @@ async fn main() -> anyhow::Result<()> {
2727
for gauge in gauges.iter() {
2828
println!("Querying the {name} table");
2929
let ctx = SessionContext::new();
30-
let table = deltalake_core::open_table(&gauge.url)
30+
let table = deltalake::open_table(gauge.url.clone())
3131
.await
3232
.expect("Failed to register table");
3333
println!("table opened");

lambdas/query-metrics/src/main.rs

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -7,10 +7,10 @@ use aws_sdk_cloudwatch::{
77
primitives::DateTime,
88
types::{Dimension, MetricDatum, StandardUnit},
99
};
10-
use deltalake_core::arrow::{array::PrimitiveArray, datatypes::Int64Type};
11-
use deltalake_core::datafusion::common::*;
12-
use deltalake_core::datafusion::execution::context::SessionContext;
13-
use lambda_runtime::{run, service_fn, Error, LambdaEvent};
10+
use deltalake::arrow::{array::PrimitiveArray, datatypes::Int64Type};
11+
use deltalake::datafusion::common::*;
12+
use deltalake::datafusion::execution::context::SessionContext;
13+
use lambda_runtime::{Error, LambdaEvent, run, service_fn};
1414
use tracing::log::*;
1515

1616
use std::collections::HashMap;
@@ -20,7 +20,7 @@ use std::time::SystemTime;
2020
mod config;
2121

2222
async fn function_handler(_event: LambdaEvent<CloudWatchEvent>) -> Result<(), Error> {
23-
deltalake_aws::register_handlers(None);
23+
deltalake::aws::register_handlers(None);
2424

2525
let aws_config = aws_config::load_defaults(aws_config::BehaviorVersion::latest()).await;
2626
let cloudwatch = aws_sdk_cloudwatch::Client::new(&aws_config);
@@ -35,7 +35,7 @@ async fn function_handler(_event: LambdaEvent<CloudWatchEvent>) -> Result<(), Er
3535
for gauge in gauges.iter() {
3636
debug!("Querying the {name} table");
3737
let ctx = SessionContext::new();
38-
let table = deltalake_core::open_table(&gauge.url)
38+
let table = deltalake::open_table(gauge.url.clone())
3939
.await
4040
.expect("Failed to register table");
4141
ctx.register_table("source", Arc::new(table))

0 commit comments

Comments
 (0)