Snow-802269 Add JsonNode function to read variant column as json object

src/main/java/com/snowflake/snowpark_java/types/Variant.java

@@ -376,6 +376,27 @@ public String asJsonString() {
     }
   }
 
+  /**
+   * Converts the variant as valid JsonNode. This function allows to read the JSON object directly
+   * rather parsing it as String and then extract the desired value
+   *
+   * <pre>{@code - to get the first value from array for key "a"
+   *
+   * Variant jv = new Variant("{\"a\": [1, 2], \"b\": \"c\"}");
+   * JsonNode jNode = jv.asJsonNode();
+   * System.out.println(jNode.get("a").get(0));
+   *
+   * output
+   * 1
+   * }</pre>
+   *
+   * @return A valid JsonNode
+   * @since 1.14.0
+   */
+  public JsonNode asJsonNode() {
+    return objectToJsonNode(value);
+  }
+
   /**
    * Converts the variant as binary value.
    *

src/main/scala/com/snowflake/snowpark/types/Variant.scala

@@ -381,6 +381,23 @@ class Variant private[snowpark] (
     }
   }
 
+  /**
+   * Converts the variant as valid JsonNode
+   * This function allows to read the JSON object directly rather parsing it as String
+   * and then extract the desired value
+   * Example - to get the first value from array for key "a"
+   * {{{
+   *   val sv = new Variant("{\"a\": [1, 2], \"b\": 3, \"c\": \"xyz\"}")
+   *   println(sv.asMap.get("a").get.asJsonNode().get(0))
+   * output
+   * 1
+   * }}}
+   * @since 1.14.0
+   */
+  def asJsonNode(): JsonNode = {
+    objectToJsonNode(value)
+  }
+
   /**
    * Converts the variant as binary value
    * @since 0.2.0

src/test/java/com/snowflake/snowpark_test/JavaVariantSuite.java

@@ -1,5 +1,6 @@
 package com.snowflake.snowpark_test;
 
+import com.fasterxml.jackson.databind.JsonNode;
 import com.snowflake.snowpark_java.types.Geography;
 import com.snowflake.snowpark_java.types.InternalUtils;
 import com.snowflake.snowpark_java.types.Variant;
@@ -365,4 +366,12 @@ public void equalsAndToString() {
 
     assert v1.toString().equals("123");
   }
+
+  @Test
+  public void javaJsonNodeVariantConverter() throws IllegalArgumentException {
+    Variant jv = new Variant("{\"a\": [1, 2], \"b\": \"c\"}");
+    JsonNode jNode = jv.asJsonNode();
+    assert (jNode.get("a").isArray());
+    assert (jNode.get("b").asText().equals("c"));
+  }
 }

src/test/scala/com/snowflake/snowpark_test/ScalaVariantSuite.scala

@@ -311,4 +311,11 @@ class ScalaVariantSuite extends FunSuite {
 
     assert(v1.toString() == "123")
   }
+
+  test("JsonNode") {
+    val sv = new Variant("{\"a\": [1, 2], \"b\": 3, \"c\": \"xyz\"}")
+    assert(sv.asMap.get("a").get.asJsonNode().isArray)
+    assert(sv.asMap.get("b").get.asJsonNode().asInt().equals(3))
+    assert(sv.asMap.get("c").get.asJsonNode().asText().equals("xyz"))
+  }
 }