Package com.databricks.jdbc.dbclient.impl.http

Class DatabricksHttpRetryHandler

java.lang.Object

com.databricks.jdbc.dbclient.impl.http.DatabricksHttpRetryHandler

All Implemented Interfaces:

org.apache.http.client.HttpRequestRetryHandler, org.apache.http.HttpResponseInterceptor

Direct Known Subclasses:

UCVolumeHttpRetryHandler

public class DatabricksHttpRetryHandler
extends Object
implements org.apache.http.HttpResponseInterceptor, org.apache.http.client.HttpRequestRetryHandler

Constructor Summary

Constructors

Constructor	Description
DatabricksHttpRetryHandler(IDatabricksConnectionContext connectionContext)

Method Summary

Modifier and Type	Method	Description
protected void	doSleepForDelay(long delayMillis)
void	process(org.apache.http.HttpResponse httpResponse, org.apache.http.protocol.HttpContext httpContext)	Processes an HTTP response to handle retryable status codes and set up retry logic.
boolean	retryRequest(IOException exception, int executionCount, org.apache.http.protocol.HttpContext context)	Determines whether a request should be retried after encountering an IOException.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

DatabricksHttpRetryHandler

public DatabricksHttpRetryHandler(IDatabricksConnectionContext connectionContext)

Method Detail

process

public void process(org.apache.http.HttpResponse httpResponse,
                    org.apache.http.protocol.HttpContext httpContext)
             throws IOException

Processes an HTTP response to handle retryable status codes and set up retry logic.

This method is responsible for examining the HTTP response, determining if it's retryable, and setting up the necessary context for potential retry attempts.

Specified by:

process in interface org.apache.http.HttpResponseInterceptor

Parameters:

httpResponse - The HTTP response to be processed.

httpContext - The HTTP context associated with the request and response.

Throws:

IOException - If there's an issue processing the response.

DatabricksRetryHandlerException - If the status code is retryable, triggering the retry mechanism.

See Also:

isStatusCodeRetryable(int), initializeRetryAccumulatedTimeIfNotExist(HttpContext), DatabricksRetryHandlerException

retryRequest

public boolean retryRequest(IOException exception,
                            int executionCount,
                            org.apache.http.protocol.HttpContext context)

Determines whether a request should be retried after encountering an IOException.

This method implements a comprehensive retry strategy for HTTP requests, considering various factors such as status codes, retry intervals, and execution counts.

Specified by:

retryRequest in interface org.apache.http.client.HttpRequestRetryHandler

Parameters:

exception - The IOException encountered during the request execution.

executionCount - The number of times this request has been executed.

context - The HttpContext containing attributes related to the request and retry logic.

Returns:

boolean True if the request should be retried, false otherwise.

Throws:

RuntimeException - If an invalid retry interval is found in the context for status codes 503 (Service Unavailable) or 429 (Too Many Requests).

See Also:

isStatusCodeRetryable(int), isRequestMethodRetryable(String), calculateDelayInMillis(int, int, int), doSleepForDelay(long)

doSleepForDelay

protected void doSleepForDelay(long delayMillis)