feat: support reading tables via Unity Catalog provided credentials

Signed-off-by: Stephen Carman <[email protected]>
delta-io · Dec 22, 2024 · 1c3fe85 · 1c3fe85
1 parent f7cc78b
commit 1c3fe85
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git 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
@@ -201,9 +201,10 @@ impl FromStr for UnityCatalogConfigKey {
     #[allow(deprecated)]
     fn from_str(s: &str) -> Result<Self, Self::Err> {
         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)
             }