Add middleware options classes to enable middleware custom configurations

Author: deepak2016

src/main/java/com/microsoft/graph/httpcore/RetryHandler.java

@@ -13,41 +13,51 @@
 import org.apache.http.protocol.HttpCoreContext;
 import org.apache.http.util.Args;
 
+import com.microsoft.graph.httpcore.middlewareoption.MiddlewareType;
+import com.microsoft.graph.httpcore.middlewareoption.RetryOptions;
+
 public class RetryHandler implements ServiceUnavailableRetryStrategy{
 
     /**
      * Maximum number of allowed retries if the server responds with a HTTP code
      * in our retry code list. Default value is 1.
      */
-    private final int maxRetries;
+    private final int maxRetries = 2;
 
     /**
      * Retry interval between subsequent requests, in milliseconds. Default
      * value is 1 second.
      */
-    private long retryInterval;
+    private long retryInterval = 1000;
     private final int DELAY_MILLISECONDS = 1000;
     private final String RETRY_AFTER = "Retry-After";
     private final String TRANSFER_ENCODING = "Transfer-Encoding";
 
     private final int MSClientErrorCodeTooManyRequests = 429;
     private final int MSClientErrorCodeServiceUnavailable  = 503;
     private final int MSClientErrorCodeGatewayTimeout = 504;
+    private final RetryOptions mRetryOption;
 
-    public RetryHandler(final int maxRetries, final int retryInterval) {
+    public RetryHandler(RetryOptions option) {
         super();
-        Args.positive(maxRetries, "Max retries");
-        Args.positive(retryInterval, "Retry interval");
-        this.maxRetries = maxRetries;
-        this.retryInterval = retryInterval;
+        this.mRetryOption = option;
     }
 
     public RetryHandler() {
-        this(2, 1000);
+        this(null);
     }
 
 	@Override
 	public boolean retryRequest(HttpResponse response, int executionCount, HttpContext context) {
+		
+		RetryOptions retryOption = (RetryOptions)context.getAttribute(MiddlewareType.RETRY.toString());
+		if(retryOption != null) {
+			return retryOption.shouldRetry().shouldRetry(response, executionCount, context);
+		}
+		if(mRetryOption != null) {
+			return mRetryOption.shouldRetry().shouldRetry(response, executionCount, context);
+		}
+		
 		boolean shouldRetry = false;
 		int statusCode = response.getStatusLine().getStatusCode();
 		shouldRetry = (executionCount < maxRetries) && checkStatus(statusCode) && isBuffered(response, context);

src/main/java/com/microsoft/graph/httpcore/middlewareoption/HttpContextBuilder.java

@@ -0,0 +1,33 @@
+package com.microsoft.graph.httpcore.middlewareoption;
+
+import org.apache.http.client.config.RequestConfig;
+import org.apache.http.client.protocol.HttpClientContext;
+
+public class HttpContextBuilder {
+	
+	private RetryOptions retryoptions;
+	private int maxRedirect = -1;
+	
+	public static HttpContextBuilder create() {
+		return new HttpContextBuilder();
+	}
+	
+	public void setRetryOption(IShouldRetry shouldRetry) {
+		retryoptions = new RetryOptions(shouldRetry);
+	}
+	
+	public void setRedirectOption(int maxRedirect) {
+		this.maxRedirect = maxRedirect;
+	}
+	
+	public HttpClientContext build() {
+		HttpClientContext context = HttpClientContext.create();
+		if(retryoptions != null)
+			context.setAttribute(MiddlewareType.RETRY.toString(), retryoptions);
+		if(maxRedirect != -1) {
+			RequestConfig config = RequestConfig.custom().setMaxRedirects(maxRedirect).build();
+			context.setRequestConfig(config);
+		}
+		return context;
+	}
+}

src/main/java/com/microsoft/graph/httpcore/middlewareoption/IMiddlewareControl.java

@@ -0,0 +1,5 @@
+package com.microsoft.graph.httpcore.middlewareoption;
+
+public interface IMiddlewareControl {
+
+}

src/main/java/com/microsoft/graph/httpcore/middlewareoption/IShouldRetry.java

@@ -0,0 +1,8 @@
+package com.microsoft.graph.httpcore.middlewareoption;
+
+import org.apache.http.HttpResponse;
+import org.apache.http.protocol.HttpContext;
+
+public interface IShouldRetry {
+	boolean shouldRetry(HttpResponse response, int executionCount, HttpContext context);
+}

src/main/java/com/microsoft/graph/httpcore/middlewareoption/MiddlewareType.java

@@ -0,0 +1,16 @@
+package com.microsoft.graph.httpcore.middlewareoption;
+
+public enum MiddlewareType {
+	
+	//Authentication Middleware
+	AUTHENTICATION,
+	
+	//Redirect Middleware
+	REDIRECT,
+	
+	//Retry Middleware
+	RETRY,
+	
+	//Not supported
+	NOT_SUPPORTED
+}

src/main/java/com/microsoft/graph/httpcore/middlewareoption/RetryOptions.java

@@ -0,0 +1,24 @@
+package com.microsoft.graph.httpcore.middlewareoption;
+
+import org.apache.http.HttpResponse;
+import org.apache.http.protocol.HttpContext;
+
+public class RetryOptions implements IMiddlewareControl {
+	private IShouldRetry shouldretry;
+	
+	public RetryOptions(){
+		this(new IShouldRetry() {
+			public boolean shouldRetry(HttpResponse response, int executionCount, HttpContext context) {
+				return true;
+			}
+		});
+	}
+	
+	public RetryOptions(IShouldRetry shouldretry){
+		this.shouldretry = shouldretry;
+	}
+	
+	public IShouldRetry shouldRetry() {
+		return shouldretry;
+	}
+}