Add User-Agent header (#43)

I realized this wasn't being set. Signed-off-by: Jesse Szwedko <[email protected]>
DataDog · Apr 14, 2024 · 49d3313 · 49d3313
1 parent f49031e
commit 49d3313
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/src/main/java/com/datadoghq/connect/logs/sink/DatadogLogsApiWriter.java b/src/main/java/com/datadoghq/connect/logs/sink/DatadogLogsApiWriter.java
@@ -171,6 +171,7 @@ private void setRequestProperties(HttpURLConnection con) {
         con.setRequestProperty("DD-API-KEY", config.ddApiKey);
         con.setRequestProperty("DD-EVP-ORIGIN", Project.getName());
         con.setRequestProperty("DD-EVP-ORIGIN-VERSION", Project.getVersion());
+        con.setRequestProperty("User-Agent", Project.getName() + "/" + Project.getVersion());
     }
 
     private byte[] compress(String str) throws IOException {
@@ -192,4 +193,4 @@ private String getOutput(InputStream input) throws IOException {
 
         return errorOutput.toString(StandardCharsets.UTF_8.name());
     }
-}
+}
diff --git a/src/test/java/com/datadoghq/connect/logs/sink/DatadogLogsApiWriterTest.java b/src/test/java/com/datadoghq/connect/logs/sink/DatadogLogsApiWriterTest.java
@@ -60,6 +60,7 @@ public void writer_givenConfigs_sendsPOSTToURL() throws IOException {
         Assert.assertTrue(request.getHeaders().contains("DD-API-KEY:" + apiKey));
         Assert.assertTrue(request.getHeaders().contains("DD-EVP-ORIGIN:datadog-kafka-connect-logs"));
         Assert.assertTrue(request.getHeaders().contains("DD-EVP-ORIGIN-VERSION:" + Project.getVersion()));
+        Assert.assertTrue(request.getHeaders().contains("User-Agent:datadog-kafka-connect-logs/" + Project.getVersion()));
     }
 
     @Test