SNOW-1991221 retries draft

Author: sfc-gh-pmotacki

src/main/java/net/snowflake/client/core/HttpExecutingContext.java

@@ -43,9 +43,11 @@ public class HttpExecutingContext {
   int injectSocketTimeout;
   boolean retryHTTP403;
   boolean shouldRetry;
+  boolean skipRetriesBecauseOf200; //todo create skip retry reason enum
   boolean withoutCookies;
   boolean includeRetryParameters;
   boolean includeRequestGuid;
+  boolean unpackResponse;
   AtomicBoolean canceling;
 
   public HttpExecutingContext(String requestIdStr, String requestInfoScrubbed) {
@@ -183,7 +185,6 @@ public void resetRetryCount() {
   }
 
   public void incrementRetryCount() {
-    System.out.println("before INCREMENTED " + retryCount);
     this.retryCount++;
   }
 
@@ -228,11 +229,6 @@ public void increaseElapsedMilliForTransientIssues(long elapsedMilliForLastCall)
   }
 
   public boolean elapsedTimeExceeded() {
-    System.out.println(
-        "elapsedTimeExceeded:"
-            + (elapsedMilliForTransientIssues > getRetryTimeoutInMilliseconds()));
-    System.out.println("elapsedTimeExceeded:" + elapsedMilliForTransientIssues);
-    System.out.println("elapsedTimeExceeded:" + getRetryTimeoutInMilliseconds());
     return elapsedMilliForTransientIssues > getRetryTimeoutInMilliseconds();
   }
 
@@ -241,9 +237,6 @@ public boolean moreThanMinRetries() {
   }
 
   public boolean maxRetriesExceeded() {
-    System.out.println("Exceeded maxRetries: " + (maxRetries > 0 && retryCount > maxRetries));
-    System.out.println("Exceeded maxRetries: " + retryCount);
-    System.out.println("Exceeded maxRetries: " + maxRetries);
     return maxRetries > 0 && retryCount >= maxRetries;
   }
 
@@ -293,7 +286,23 @@ public boolean isIncludeRetryParameters() {
     return includeRetryParameters;
   }
 
+  public boolean isUnpackResponse() {
+    return unpackResponse;
+  }
+
+  public void setUnpackResponse(boolean unpackResponse) {
+    this.unpackResponse = unpackResponse;
+  }
+
   public void setIncludeRetryParameters(boolean includeRetryParameters) {
     this.includeRetryParameters = includeRetryParameters;
   }
+
+  public boolean isSkipRetriesBecauseOf200() {
+    return skipRetriesBecauseOf200;
+  }
+
+  public void setSkipRetriesBecauseOf200(boolean skipRetriesBecauseOf200) {
+    this.skipRetriesBecauseOf200 = skipRetriesBecauseOf200;
+  }
 }

src/main/java/net/snowflake/client/core/HttpResponseContextDto.java

@@ -0,0 +1,50 @@
+package net.snowflake.client.core;
+
+import org.apache.http.client.methods.CloseableHttpResponse;
+
+public class HttpResponseContextDto {
+
+    private CloseableHttpResponse httpResponse;
+    private String unpackedCloseableHttpResponse;
+    private Exception savedEx;
+
+    // Constructors
+    public HttpResponseContextDto() {}
+
+    public HttpResponseContextDto(CloseableHttpResponse httpResponse, String unpackedCloseableHttpResponse) {
+        this.httpResponse = httpResponse;
+        this.unpackedCloseableHttpResponse = unpackedCloseableHttpResponse;
+    }
+
+    public CloseableHttpResponse getHttpResponse() {
+        return httpResponse;
+    }
+
+    public void setHttpResponse(CloseableHttpResponse httpResponse) {
+        this.httpResponse = httpResponse;
+    }
+
+    public String getUnpackedCloseableHttpResponse() {
+        return unpackedCloseableHttpResponse;
+    }
+
+    public void setUnpackedCloseableHttpResponse(String unpackedCloseableHttpResponse) {
+        this.unpackedCloseableHttpResponse = unpackedCloseableHttpResponse;
+    }
+
+    public Exception getSavedEx() {
+        return savedEx;
+    }
+
+    public void setSavedEx(Exception savedEx) {
+        this.savedEx = savedEx;
+    }
+
+    @Override
+    public String toString() {
+        return "CloseableHttpResponseContextDto{" +
+                "httpResponse=" + httpResponse +
+                ", unpackedCloseableHttpResponse=" + unpackedCloseableHttpResponse +
+                '}';
+    }
+}