change default aggregation key

underdogio · Apr 11, 2016 · 9f447e4 · 9f447e4
1 parent 5d338e8
commit 9f447e4
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/main/java/io/underdog/GraylogPluginDatadog.java b/src/main/java/io/underdog/GraylogPluginDatadog.java
@@ -116,7 +116,12 @@ public void write(Message message) throws Exception {
         if (aggregationKey != "") {
             payload.put("aggregation_key", aggregationKey);
         } else {
-            payload.put("aggregation_key", message.getSource());
+            final StringBuilder sb = new StringBuilder();
+            sb.append(message.getSource());
+            if (message.getField("facility") != null) {
+                sb.append("_").append(message.getField("facility"));
+            }
+            payload.put("aggregation_key", sb.toString());
         }
 
         Response response = eventTarget
@@ -195,7 +200,7 @@ public ConfigurationRequest getRequestedConfiguration() {
 
             configurationRequest.addField(new TextField(
                             CK_DATADOG_AGGREGATION_KEY, "Datadog event aggregation key", "",
-                            "Datadog event custom aggregation key [default: message source]",
+                            "Datadog event custom aggregation key [default: '[source]_[facility]']",
                             ConfigurationField.Optional.OPTIONAL)
             );