add check for array type

snowflakedb · Jan 9, 2025 · f22d07d · f22d07d
1 parent 624c564
commit f22d07d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/main/java/net/snowflake/client/core/SFArrowResultSet.java b/src/main/java/net/snowflake/client/core/SFArrowResultSet.java
@@ -597,9 +597,11 @@ private StructObjectWrapper getObjectRepresentation(int columnIndex, boolean wit
         if (type == Types.STRUCT) {
           JsonSqlInput jsonSqlInput = createJsonSqlInput(columnIndex, obj);
           return new StructObjectWrapper(jsonSqlInput.getText(), jsonSqlInput);
-        } else {
+        } else if (type == Types.ARRAY) {
           SfSqlArray sfArray = getJsonArray((String) obj, columnIndex);
           return new StructObjectWrapper(sfArray.getText(), sfArray);
+        } else {
+          throw new SFException(queryId, ErrorCode.INVALID_STRUCT_DATA);
         }
       } else if (converter instanceof StructConverter) {
         String jsonString = withString ? converter.toString(index) : null;