[PECO-2153] Support OAuth on databricks.azure.cn (#271)

[PEC-2153] Support OAuth on databricks.azure.cn Signed-off-by: Jacky Hu <[email protected]>
databricks · Dec 16, 2024 · af23aa0 · af23aa0
1 parent 56bd0d9
commit af23aa0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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