From f33fc1c859fef84adb1f0cda542899d0457c8d98 Mon Sep 17 00:00:00 2001 From: Jacky Hu Date: Mon, 16 Dec 2024 10:48:12 -0800 Subject: [PATCH] [PEC-2153] Support OAuth on databricks.azure.cn Signed-off-by: Jacky Hu --- lib/connection/auth/DatabricksOAuth/OAuthManager.ts | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lib/connection/auth/DatabricksOAuth/OAuthManager.ts b/lib/connection/auth/DatabricksOAuth/OAuthManager.ts index 02bbde15..db7e7c69 100644 --- a/lib/connection/auth/DatabricksOAuth/OAuthManager.ts +++ b/lib/connection/auth/DatabricksOAuth/OAuthManager.ts @@ -219,7 +219,7 @@ export default abstract class OAuthManager { } if (options.useDatabricksOAuthInAzure) { - const azureDomains = ['.azuredatabricks.net']; + const azureDomains = ['.azuredatabricks.net', '.databricks.azure.cn']; const isAzureDomain = azureDomains.some((domain) => host.endsWith(domain)); if (isAzureDomain) { // eslint-disable-next-line @typescript-eslint/no-use-before-define