SNOW-1940294: Convert string CLIENT_REQUEST_MFA_TOKEN to boolean (#2089)

snowflakedb · Feb 20, 2025 · dcb60b2 · dcb60b2
1 parent 087647d
commit dcb60b2
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/main/java/net/snowflake/client/core/SessionUtil.java b/src/main/java/net/snowflake/client/core/SessionUtil.java
@@ -306,6 +306,8 @@ static SFLoginOutput openSession(
       }
     }
 
+    convertSessionParameterStringValueToBooleanIfGiven(loginInput, CLIENT_REQUEST_MFA_TOKEN);
+
     preNewSession(loginInput);
 
     try {
@@ -317,6 +319,16 @@ static SFLoginOutput openSession(
     }
   }
 
+  private static void convertSessionParameterStringValueToBooleanIfGiven(
+      SFLoginInput loginInput, String parameterName) {
+    Object currentClientRequestMfaToken = loginInput.getSessionParameters().get(parameterName);
+    if (currentClientRequestMfaToken instanceof String) {
+      loginInput
+          .getSessionParameters()
+          .put(parameterName, Boolean.parseBoolean((String) currentClientRequestMfaToken));
+    }
+  }
+
   private static void preNewSession(SFLoginInput loginInput) throws SFException {
     if (asBoolean(loginInput.getSessionParameters().get(CLIENT_STORE_TEMPORARY_CREDENTIAL))) {
       CredentialManager.getInstance().fillCachedIdToken(loginInput);