From 4c99577c08cc7b4c214db84e1e41bbb939c732ea Mon Sep 17 00:00:00 2001
From: Chris Eager <chris@signal.org>
Date: Wed, 30 Jun 2021 15:01:58 -0500
Subject: [PATCH] Add configuration for Datadog batch size

---
 .../textsecuregcm/WhisperServerService.java              | 5 +++++
 .../configuration/DatadogConfiguration.java              | 9 +++++++++
 2 files changed, 14 insertions(+)

diff --git a/service/src/main/java/org/whispersystems/textsecuregcm/WhisperServerService.java b/service/src/main/java/org/whispersystems/textsecuregcm/WhisperServerService.java
index 2e7182363..d81155f0b 100644
--- a/service/src/main/java/org/whispersystems/textsecuregcm/WhisperServerService.java
+++ b/service/src/main/java/org/whispersystems/textsecuregcm/WhisperServerService.java
@@ -297,6 +297,11 @@ public class WhisperServerService extends Application<WhisperServerConfiguration
           return config.getDatadogConfiguration().getStep();
         }
 
+        @Override
+        public int batchSize() {
+          return config.getDatadogConfiguration().getBatchSize();
+        }
+
         @Override
         public String hostTag() {
           return "host";
diff --git a/service/src/main/java/org/whispersystems/textsecuregcm/configuration/DatadogConfiguration.java b/service/src/main/java/org/whispersystems/textsecuregcm/configuration/DatadogConfiguration.java
index 3642f8d6d..f8f760beb 100644
--- a/service/src/main/java/org/whispersystems/textsecuregcm/configuration/DatadogConfiguration.java
+++ b/service/src/main/java/org/whispersystems/textsecuregcm/configuration/DatadogConfiguration.java
@@ -6,6 +6,7 @@
 package org.whispersystems.textsecuregcm.configuration;
 
 import com.fasterxml.jackson.annotation.JsonProperty;
+import javax.validation.constraints.Min;
 import javax.validation.constraints.NotBlank;
 import javax.validation.constraints.NotNull;
 import java.time.Duration;
@@ -24,6 +25,10 @@ public class DatadogConfiguration {
   @NotBlank
   private String environment;
 
+  @JsonProperty
+  @Min(1)
+  private int batchSize = 5_000;
+
   public String getApiKey() {
     return apiKey;
   }
@@ -35,4 +40,8 @@ public class DatadogConfiguration {
   public String getEnvironment() {
     return environment;
   }
+
+  public int getBatchSize() {
+    return batchSize;
+  }
 }