Merge pull request #42 from JoeHitHard/master

add req headers to send req
ApptuitAI · Oct 26, 2022 · 7eefd21 · 7eefd21
2 parents d3e42d6 + 84a2fb6
commit 7eefd21
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/send-client/src/main/java/ai/apptuit/metrics/client/ApptuitPutClient.java b/send-client/src/main/java/ai/apptuit/metrics/client/ApptuitPutClient.java
@@ -31,10 +31,7 @@
 import java.net.URL;
 import java.nio.charset.Charset;
 import java.nio.charset.StandardCharsets;
-import java.util.Base64;
-import java.util.Collection;
-import java.util.Iterator;
-import java.util.Map;
+import java.util.*;
 import java.util.logging.Level;
 import java.util.logging.Logger;
 import java.util.zip.GZIPOutputStream;
@@ -96,6 +93,9 @@ public void send(Collection<DataPoint> dataPoints) throws ConnectException, Resp
   }
 
   public void send(Collection<DataPoint> dataPoints, Sanitizer sanitizer) throws ConnectException, ResponseStatusException, IOException {
+    send(dataPoints, sanitizer, null);
+  }
+  public void send(Collection<DataPoint> dataPoints, Sanitizer sanitizer, Map<String, String> reqHeaders) throws ConnectException, ResponseStatusException, IOException {
 
 
     if (dataPoints.isEmpty()) {
@@ -118,6 +118,9 @@ public void send(Collection<DataPoint> dataPoints, Sanitizer sanitizer) throws C
         urlConnection.setRequestProperty(CONTENT_ENCODING, CONTENT_ENCODING_GZIP);
       }
       urlConnection.setRequestProperty("Authorization", generateAuthHeader());
+      if (reqHeaders != null && !reqHeaders.isEmpty()) {
+        reqHeaders.forEach(urlConnection::setRequestProperty);
+      }
       urlConnection.setRequestMethod("POST");
       urlConnection.setDoInput(true);
       urlConnection.setDoOutput(true);