Make retry strategy configurable

Author: avijeet95

databricks-sdk-java/src/main/java/com/databricks/sdk/core/commons/CommonsHttpClient.java

@@ -23,6 +23,7 @@
 import java.util.stream.Collectors;
 import org.apache.commons.io.IOUtils;
 import org.apache.http.*;
+import org.apache.http.client.HttpRequestRetryHandler;
 import org.apache.http.client.config.RequestConfig;
 import org.apache.http.client.methods.*;
 import org.apache.http.conn.ssl.SSLConnectionSocketFactory;
@@ -47,6 +48,7 @@ public static class Builder {
     private ProxyConfig proxyConfig;
     private SSLConnectionSocketFactory sslSocketFactory;
     private PoolingHttpClientConnectionManager connectionManager;
+    private HttpRequestRetryHandler requestRetryHandler;
 
     /**
      * @param databricksConfig The DatabricksConfig to use for the HttpClient. If the
@@ -96,6 +98,15 @@ public Builder withConnectionManager(PoolingHttpClientConnectionManager connecti
       return this;
     }
 
+    /**
+     * @param requestRetryHandler the HttpRequestRetryHandler to use for the HttpClient.
+     * @return This builder.
+     */
+    public Builder withRequestRetryHandler(HttpRequestRetryHandler requestRetryHandler) {
+      this.requestRetryHandler = requestRetryHandler;
+      return this;
+    }
+
     /** Builds a new instance of CommonsHttpClient with the configured parameters. */
     public CommonsHttpClient build() {
       return new CommonsHttpClient(this);
@@ -132,6 +143,9 @@ private CommonsHttpClient(Builder builder) {
       connectionManager.setMaxTotal(100);
       httpClientBuilder.setConnectionManager(connectionManager);
     }
+    if (builder.requestRetryHandler != null) {
+      httpClientBuilder.setRetryHandler(builder.requestRetryHandler);
+    }
     hc = httpClientBuilder.build();
   }