From 1c3fe85185370fa199996715ab9afd76a4b83104 Mon Sep 17 00:00:00 2001 From: Stephen Carman Date: Sun, 22 Dec 2024 12:34:42 -0500 Subject: [PATCH] feat: support reading tables via Unity Catalog provided credentials Signed-off-by: Stephen Carman --- crates/catalog-unity/Cargo.toml | 2 +- crates/catalog-unity/src/lib.rs | 7 ++++--- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/crates/catalog-unity/Cargo.toml b/crates/catalog-unity/Cargo.toml index aa6ad560e1..8626834562 100644 --- a/crates/catalog-unity/Cargo.toml +++ b/crates/catalog-unity/Cargo.toml @@ -39,7 +39,7 @@ httpmock = { version = "0.8.0-alpha.1" } tracing-subscriber = { version = "0.3", features = ["env-filter"] } [features] -default = [] +default = ["datafusion", "aws"] aws = ["deltalake-aws"] azure = ["deltalake-azure"] datafusion = ["dep:datafusion", "datafusion-common", "deltalake-core/datafusion", "moka"] diff --git a/crates/catalog-unity/src/lib.rs b/crates/catalog-unity/src/lib.rs index 128cda943c..d19ecdcbcd 100644 --- a/crates/catalog-unity/src/lib.rs +++ b/crates/catalog-unity/src/lib.rs @@ -201,9 +201,10 @@ impl FromStr for UnityCatalogConfigKey { #[allow(deprecated)] fn from_str(s: &str) -> Result { match s { - "access_token" | "unity_access_token" | "databricks_access_token" => { - Ok(UnityCatalogConfigKey::AccessToken) - } + "access_token" + | "unity_access_token" + | "databricks_access_token" + | "databricks_token" => Ok(UnityCatalogConfigKey::AccessToken), "authority_host" | "unity_authority_host" | "databricks_authority_host" => { Ok(UnityCatalogConfigKey::AuthorityHost) }