SNOW-2240640 Set OCSP server for chunk downloader (#2299)

Author: sfc-gh-rkowalski

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

@@ -234,6 +234,15 @@ public SnowflakeChunkDownloader(SnowflakeResultSetSerializableV1 resultSetSerial
         this.prefetchMaxRetry = (int) prefetchMaxRetry;
       }
     }
+
+    if (resultSetSerializable.getServerURL() != null) {
+      try {
+        SessionUtil.resetOCSPUrlIfNecessary(resultSetSerializable.getServerURL());
+      } catch (IOException e) {
+        logger.debug("Exception while resetting OCSP URL", e);
+      }
+    }
+
     this.memoryLimit = resultSetSerializable.getMemoryLimit();
     if (this.session != null
         && session.getMemoryLimitForTesting() != SFBaseSession.MEMORY_LIMIT_UNSET) {

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

@@ -142,6 +142,7 @@ public String toString() {
   // Below fields are from session or statement
   SnowflakeConnectString snowflakeConnectionString;
   OCSPMode ocspMode;
+  String serverUrl;
   HttpClientSettingsKey httpClientKey;
   int networkTimeoutInMilli;
   int authTimeout;
@@ -220,6 +221,7 @@ private SnowflakeResultSetSerializableV1(SnowflakeResultSetSerializableV1 toCopy
     // Below fields are from session or statement
     this.snowflakeConnectionString = toCopy.snowflakeConnectionString;
     this.ocspMode = toCopy.ocspMode;
+    this.serverUrl = toCopy.serverUrl;
     this.httpClientKey = toCopy.httpClientKey;
     this.networkTimeoutInMilli = toCopy.networkTimeoutInMilli;
     this.authTimeout = toCopy.authTimeout;
@@ -428,6 +430,7 @@ protected SnowflakeResultSetSerializableV1(
 
     // setup fields from sessions.
     this.ocspMode = sfSession.getOCSPMode();
+    this.serverUrl = sfSession.getServerUrl();
     this.httpClientKey = sfSession.getHttpClientKey();
     this.snowflakeConnectionString = sfSession.getSnowflakeConnectionString();
     this.networkTimeoutInMilli = sfSession.getNetworkTimeoutInMilli();
@@ -523,6 +526,10 @@ public OCSPMode getOCSPMode() {
     return ocspMode;
   }
 
+  public String getServerURL() {
+    return serverUrl;
+  }
+
   public HttpClientSettingsKey getHttpClientKey() {
     return httpClientKey;
   }