From 999fb18b2ea092f1f7d831c66d7fbcd9da46c97c Mon Sep 17 00:00:00 2001
From: Jianghao Lu <jianghaolu@users.noreply.github.com>
Date: Fri, 17 Jun 2016 11:19:27 -0700
Subject: [PATCH] Add retry for 404 GET

---
 ...rceGetExponentialBackoffRetryStrategy.java | 48 +++++++++++++++++++
 .../java/com/microsoft/azure/RestClient.java  |  1 +
 2 files changed, 49 insertions(+)
 create mode 100644 azure-client-runtime/src/main/java/com/microsoft/azure/ResourceGetExponentialBackoffRetryStrategy.java

diff --git a/azure-client-runtime/src/main/java/com/microsoft/azure/ResourceGetExponentialBackoffRetryStrategy.java b/azure-client-runtime/src/main/java/com/microsoft/azure/ResourceGetExponentialBackoffRetryStrategy.java
new file mode 100644
index 0000000000000..a145fecc6795b
--- /dev/null
+++ b/azure-client-runtime/src/main/java/com/microsoft/azure/ResourceGetExponentialBackoffRetryStrategy.java
@@ -0,0 +1,48 @@
+/**
+ *
+ * Copyright (c) Microsoft Corporation. All rights reserved.
+ * Licensed under the MIT License. See License.txt in the project root for license information.
+ *
+ */
+
+package com.microsoft.azure;
+
+import com.microsoft.rest.retry.RetryStrategy;
+import okhttp3.Response;
+
+/**
+ * A retry strategy with backoff parameters for calculating the exponential
+ * delay between retries for 404s from GET calls.
+ */
+public class ResourceGetExponentialBackoffRetryStrategy extends RetryStrategy {
+    /**
+     * Represents the default number of retries.
+     */
+    private static final int DEFAULT_NUMBER_OF_ATTEMPTS = 3;
+
+    /**
+     * Creates an instance of the retry strategy.
+     */
+    public ResourceGetExponentialBackoffRetryStrategy() {
+        this(null, DEFAULT_FIRST_FAST_RETRY);
+    }
+
+    /**
+     * Initializes a new instance of the {@link RetryStrategy} class.
+     *
+     * @param name           The name of the retry strategy.
+     * @param firstFastRetry true to immediately retry in the first attempt; otherwise, false.
+     */
+    private ResourceGetExponentialBackoffRetryStrategy(String name, boolean firstFastRetry) {
+        super(name, firstFastRetry);
+    }
+
+    @Override
+    public boolean shouldRetry(int retryCount, Response response) {
+        int code = response.code();
+        //CHECKSTYLE IGNORE MagicNumber FOR NEXT 2 LINES
+        return retryCount < DEFAULT_NUMBER_OF_ATTEMPTS
+                && code == 404
+                && response.request().method().equalsIgnoreCase("GET");
+    }
+}
diff --git a/azure-client-runtime/src/main/java/com/microsoft/azure/RestClient.java b/azure-client-runtime/src/main/java/com/microsoft/azure/RestClient.java
index 532ba2e392809..0ace881f0ee51 100644
--- a/azure-client-runtime/src/main/java/com/microsoft/azure/RestClient.java
+++ b/azure-client-runtime/src/main/java/com/microsoft/azure/RestClient.java
@@ -330,6 +330,7 @@ public RestClient build() {
                 OkHttpClient httpClient = httpClientBuilder
                         .addInterceptor(baseUrlHandler)
                         .addInterceptor(customHeadersInterceptor)
+                        .addInterceptor(new RetryHandler(new ResourceGetExponentialBackoffRetryStrategy()))
                         .addInterceptor(new RetryHandler())
                         .build();
                 return new RestClient(httpClient,