From 16583300000e8d6bd7e1b97b01efdcf9cdf62dd2 Mon Sep 17 00:00:00 2001 From: Harsh Chaturvedi Date: Tue, 23 Aug 2022 10:38:14 -0700 Subject: [PATCH] SNOW-648958 Add support for new OKTA OIE to JDBC Driver (#1122) * Add support for new OKTA OIE to JDBC Driver --- src/main/java/net/snowflake/client/core/SessionUtil.java | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/main/java/net/snowflake/client/core/SessionUtil.java b/src/main/java/net/snowflake/client/core/SessionUtil.java index ff283549a..006c38d82 100644 --- a/src/main/java/net/snowflake/client/core/SessionUtil.java +++ b/src/main/java/net/snowflake/client/core/SessionUtil.java @@ -1150,7 +1150,10 @@ private static String federatedFlowStep3(SFLoginInput loginInput, String tokenUr // session token is in the data field of the returned json response final JsonNode jsonNode = mapper.readTree(idpResponse); - oneTimeToken = jsonNode.get("cookieToken").asText(); + oneTimeToken = + jsonNode.get("sessionToken") != null + ? jsonNode.get("sessionToken").asText() + : jsonNode.get("cookieToken").asText(); } catch (IOException | URISyntaxException ex) { handleFederatedFlowError(loginInput, ex); }