Add setExtraHeader

src/main/java/com/algolia/search/saas/APIClient.java

@@ -12,7 +12,10 @@
 import java.security.NoSuchAlgorithmException;
 import java.util.Arrays;
 import java.util.Collections;
+import java.util.HashMap;
 import java.util.List;
+import java.util.Map;
+import java.util.Map.Entry;
 
 import javax.crypto.Mac;
 import javax.crypto.spec.SecretKeySpec;
@@ -91,6 +94,7 @@ public class APIClient {
     private String forwardRateLimitAPIKey;
     private String forwardEndUserIP;
     private String forwardAdminAPIKey;
+    private HashMap<String, String> headers;
 
     /**
      * Algolia Search initialization
@@ -126,6 +130,7 @@ public APIClient(String applicationID, String apiKey, List<String> hostsArray) {
         Collections.shuffle(hostsArray);
         this.hostsArray = hostsArray;
         httpClient = HttpClientBuilder.create().build();
+        headers = new HashMap<String, String>();
     }
 
     /**
@@ -149,6 +154,13 @@ public void disableRateLimitForward() {
         forwardAdminAPIKey = forwardEndUserIP = forwardRateLimitAPIKey = null;
     }
 
+    /**
+     * Allow to set custom headers
+     */
+    public void setExtraHeader(String key, String value) {
+    	headers.put(key, value);
+    }
+
     /**
      * List all existing indexes
      * return an JSON Object in the form:
@@ -519,6 +531,9 @@ private JSONObject _request(Method m, String url, String json) throws AlgoliaExc
         		req.setHeader("X-Forwarded-For", this.forwardEndUserIP);
         		req.setHeader("X-Forwarded-API-Key", this.forwardRateLimitAPIKey);
         	}
+        	for (Entry<String, String> entry : headers.entrySet()) {
+        		req.setHeader(entry.getKey(), entry.getValue());
+        	}
 
         	// set user agent
         	req.setHeader("User-Agent", "Algolia for Java " + version);