add header for uk traffic

Mangopay · Mar 5, 2024 · dc4c9d0 · dc4c9d0
1 parent 0eb5ffa
commit dc4c9d0
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 2 deletions.
diff --git a/src/main/java/com/mangopay/core/APIs/implementation/OAuthApiImpl.java b/src/main/java/com/mangopay/core/APIs/implementation/OAuthApiImpl.java
@@ -52,6 +52,9 @@ public OAuthToken createToken() throws Exception {
         }
         rest.addRequestHttpHeader("Content-Type", "application/x-www-form-urlencoded");
         rest.addRequestHttpHeader("User-Agent",  String.format("MangoPay V2 SDK Java %s", getRoot().getConfig().getVersion()));
+        if(root.getConfig().isUkHeaderFlag()){
+            rest.addRequestHttpHeader("x-tenant-id", "uk");
+        }
         OAuthToken response = rest.request(OAuthToken.class, null, urlMethod, requestType, requestData);
 
         return response;

diff --git a/src/main/java/com/mangopay/core/Configuration.java b/src/main/java/com/mangopay/core/Configuration.java
@@ -45,6 +45,11 @@ public class Configuration {
      * Mangopay SDK Version
      */
     private String version;
+
+    /**
+     * Set to true for uk traffic
+     */
+    private boolean ukHeaderFlag = false;
 
 
     public String getClientId() {
@@ -115,7 +120,14 @@ public void setReadTimeout(int readTimeout) {
         this.readTimeout = readTimeout;
     }
 
-
+    public boolean isUkHeaderFlag() {
+        return ukHeaderFlag;
+    }
+
+    public void setUkHeaderFlag(boolean ukHeaderFlag) {
+        this.ukHeaderFlag = ukHeaderFlag;
+    }
+
     /**
      * Get Mangopay SDK Version
      * @return String Mangopay Version

diff --git a/src/main/java/com/mangopay/core/RestTool.java b/src/main/java/com/mangopay/core/RestTool.java
@@ -679,7 +679,9 @@ private Map<String, String> getHttpHeaders(String restUrl) throws Exception {
         }
 
         httpHeaders.put("User-Agent",  String.format("MangoPay V2 SDK Java %s", root.getConfig().getVersion()));
-
+        if(this.root.getConfig().isUkHeaderFlag()) {
+            httpHeaders.put("x-tenant-id", "uk");
+        }
         return httpHeaders;
     }