Skip to content

Commit f10f15e

Browse files
authored
[PECOBLR-762] Make U2M work independent to that of java sdk (#954)
## Description - We have had issues with making Azure U2M work in the long term, this PR is to fix these issues in the next release. - In this PR we have replicated the simple U2M flow added in our other OSS drivers. ## Testing - Added unit tests to maintain coverage. - Tested manually using Auth_Flow=2 with adb host. - Note that one of our customer also tested this out and says it is working for them too. ## Additional Notes to the Reviewer - This ticket will decide the future fate of placement of the Azure U2M auth class : https://databricks.atlassian.net/browse/PECOBLR-762
1 parent df447ec commit f10f15e

6 files changed

Lines changed: 1338 additions & 4 deletions

File tree

NEXT_CHANGELOG.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
- Databricks SDK dependency upgraded to latest version 0.60.0
1616

1717
### Fixed
18+
- Integrated Azure U2M flow into driver for improved stability.
1819
- Fixed `ResultSet.getString` for Boolean columns in Metadata result set.
1920
---
2021
*Note: When making changes, please add your change under the appropriate section with a brief description.*

src/main/java/com/databricks/jdbc/api/impl/DatabricksConnectionContext.java

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -215,6 +215,11 @@ public String getHostUrl() throws DatabricksParsingException {
215215
}
216216
}
217217

218+
@Override
219+
public String getHost() {
220+
return this.host;
221+
}
222+
218223
@Override
219224
public IDatabricksComputeResource getComputeResource() {
220225
return computeResource;

src/main/java/com/databricks/jdbc/api/internal/IDatabricksConnectionContext.java

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,13 @@ public interface IDatabricksConnectionContext {
1717
*/
1818
String getHostUrl() throws DatabricksParsingException;
1919

20+
/**
21+
* Returns just the host parsed from JDBC connection. Note : this is not the url.
22+
*
23+
* @return Databricks host
24+
*/
25+
String getHost();
26+
2027
/**
2128
* Returns warehouse-Id as parsed from JDBC connection Url
2229
*

0 commit comments

Comments
 (0)