SNOW-1636262: getDate triggers NPE with 3.18.0 (regression)

src/main/java/net/snowflake/client/jdbc/SnowflakeBaseResultSet.java

@@ -64,6 +64,7 @@ public abstract class SnowflakeBaseResultSet implements ResultSet {
   protected Map<String, Object> parameters = new HashMap<>();
   private int fetchSize = 0;
   protected SFBaseSession session = null;
+  private SnowflakeResultSetSerializableV1 serializable = null;
   private static final ObjectMapper OBJECT_MAPPER = ObjectMapperFactory.getObjectMapper();
 
   SnowflakeBaseResultSet(Statement statement) throws SQLException {
@@ -95,6 +96,7 @@ public SnowflakeBaseResultSet(SnowflakeResultSetSerializableV1 resultSetSerializ
     this.resultSetType = resultSetSerializable.getResultSetType();
     this.resultSetConcurrency = resultSetSerializable.getResultSetConcurrency();
     this.resultSetHoldability = resultSetSerializable.getResultSetHoldability();
+    this.serializable = resultSetSerializable;
   }
 
   /**
@@ -131,12 +133,22 @@ protected void raiseSQLExceptionIfResultSetIsClosed() throws SQLException {
 
   public abstract Date getDate(int columnIndex, TimeZone tz) throws SQLException;
 
+  public boolean getGetDateUseNullTimezone() {
+    if (this.session != null) {
+      return this.session.getGetDateUseNullTimezone();
+    }
+
+    if (this.serializable != null) {
+      return this.serializable.getGetDateUseNullTimezone();
+    }
+
+    return false;
+  }
+
   @Override
   public Date getDate(int columnIndex) throws SQLException {
     raiseSQLExceptionIfResultSetIsClosed();
-    return getDate(
-        columnIndex,
-        this.session.getGetDateUseNullTimezone() ? (TimeZone) null : TimeZone.getDefault());
+    return getDate(columnIndex, getGetDateUseNullTimezone() ? null : TimeZone.getDefault());
   }
 
   @Override

src/main/java/net/snowflake/client/jdbc/SnowflakeResultSetSerializableV1.java

@@ -155,6 +155,10 @@ public String toString() {
   int resultSetType;
   int resultSetConcurrency;
   int resultSetHoldability;
+  boolean treatNTZAsUTC;
+  boolean formatDateWithTimezone;
+  boolean useSessionTimezone;
+  boolean getDateUseNullTimezone;
 
   // Below are some metadata fields parsed from the result JSON node
   String queryId;
@@ -173,9 +177,6 @@ public String toString() {
   long sendResultTime;
   List<MetaDataOfBinds> metaDataOfBinds = new ArrayList<>();
   QueryResultFormat queryResultFormat;
-  boolean treatNTZAsUTC;
-  boolean formatDateWithTimezone;
-  boolean useSessionTimezone;
   int sessionClientMemoryLimit;
 
   // Below fields are transient, they are generated from parameters
@@ -235,6 +236,7 @@ private SnowflakeResultSetSerializableV1(SnowflakeResultSetSerializableV1 toCopy
     this.treatNTZAsUTC = toCopy.treatNTZAsUTC;
     this.formatDateWithTimezone = toCopy.formatDateWithTimezone;
     this.useSessionTimezone = toCopy.useSessionTimezone;
+    this.getDateUseNullTimezone = toCopy.getDateUseNullTimezone;
 
     // Below are some metadata fields parsed from the result JSON node
     this.queryId = toCopy.queryId;
@@ -438,6 +440,7 @@ protected SnowflakeResultSetSerializableV1(
     this.treatNTZAsUTC = sfSession.getTreatNTZAsUTC();
     this.formatDateWithTimezone = sfSession.getFormatDateWithTimezone();
     this.useSessionTimezone = sfSession.getUseSessionTimezone();
+    this.getDateUseNullTimezone = sfSession.getGetDateUseNullTimezone();
 
     // setup transient fields from parameter
     this.setupFieldsFromParameters();
@@ -701,6 +704,10 @@ public boolean getUseSessionTimezone() {
     return useSessionTimezone;
   }
 
+  public boolean getGetDateUseNullTimezone() {
+    return getDateUseNullTimezone;
+  }
+
   public Optional<SFBaseSession> getSession() {
     return possibleSession;
   }

src/test/java/net/snowflake/client/jdbc/ConnectionIT.java

@@ -34,11 +34,14 @@
 import java.sql.SQLFeatureNotSupportedException;
 import java.sql.Statement;
 import java.util.Collections;
+import java.util.Date;
 import java.util.Enumeration;
+import java.util.List;
 import java.util.Map;
 import java.util.Properties;
 import java.util.concurrent.ExecutorService;
 import java.util.concurrent.Executors;
+import java.util.stream.Collectors;
 import net.snowflake.client.ConditionalIgnoreRule.ConditionalIgnore;
 import net.snowflake.client.RunningNotOnTestaccount;
 import net.snowflake.client.RunningOnGithubAction;
@@ -800,6 +803,36 @@ public void testStatementsAndResultSetsClosedByConnection() throws SQLException
     assertTrue(rs3.isClosed());
   }
 
+  @Test
+  public void testReadDateAfterSplittingResultSet() throws Exception {
+    Connection conn = getConnection();
+    try (Statement statement = conn.createStatement()) {
+      statement.execute("create or replace table table_with_date (int_c int, date_c date)");
+      statement.execute("insert into table_with_date values (1, '2015-10-25')");
+
+      try (ResultSet rs = statement.executeQuery("select * from table_with_date")) {
+        final SnowflakeResultSet resultSet = rs.unwrap(SnowflakeResultSet.class);
+        final List<SnowflakeResultSetSerializable> serializables =
+            resultSet.getResultSetSerializables(1 << 20);
+        final List<Date> dates =
+            serializables.stream()
+                .map(
+                    s -> {
+                      try {
+                        ResultSet srs = s.getResultSet();
+                        srs.next();
+                        return srs.getDate(2);
+                      } catch (Exception e) {
+                        throw new RuntimeException(e);
+                      }
+                    })
+                .collect(Collectors.toList());
+        assertEquals(1, dates.size());
+        assertEquals("2015-10-25", dates.get(0).toString());
+      }
+    }
+  }
+
   @Test
   public void testResultSetsClosedByStatement() throws SQLException {
     Connection connection = getConnection();