From 251e1b51c5fc5588d7ebb3b9b6cd7f1e2d62864d Mon Sep 17 00:00:00 2001
From: Jon Chambers <jon@signal.org>
Date: Fri, 13 Nov 2020 14:53:16 -0500
Subject: [PATCH] Make Micrometer batch size configurable.

---
 .../textsecuregcm/WhisperServerService.java              | 5 +++++
 .../configuration/MicrometerConfiguration.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 d339f0a1a..3c7b2c2fb 100644
--- a/service/src/main/java/org/whispersystems/textsecuregcm/WhisperServerService.java
+++ b/service/src/main/java/org/whispersystems/textsecuregcm/WhisperServerService.java
@@ -231,6 +231,11 @@ public class WhisperServerService extends Application<WhisperServerConfiguration
       public String uri() {
         return config.getMicrometerConfiguration().getUri();
       }
+
+      @Override
+      public int batchSize() {
+        return config.getMicrometerConfiguration().getBatchSize();
+      }
     }, Clock.SYSTEM) {
       @Override
       protected DistributionStatisticConfig defaultHistogramConfig() {
diff --git a/service/src/main/java/org/whispersystems/textsecuregcm/configuration/MicrometerConfiguration.java b/service/src/main/java/org/whispersystems/textsecuregcm/configuration/MicrometerConfiguration.java
index fe4589afb..1edbba862 100644
--- a/service/src/main/java/org/whispersystems/textsecuregcm/configuration/MicrometerConfiguration.java
+++ b/service/src/main/java/org/whispersystems/textsecuregcm/configuration/MicrometerConfiguration.java
@@ -8,13 +8,22 @@ package org.whispersystems.textsecuregcm.configuration;
 import com.fasterxml.jackson.annotation.JsonProperty;
 
 import javax.validation.constraints.NotEmpty;
+import javax.validation.constraints.Positive;
 
 public class MicrometerConfiguration {
 
     @JsonProperty
     private String uri;
 
+    @JsonProperty
+    @Positive
+    private int batchSize = 10_000;
+
     public String getUri() {
         return uri;
     }
+
+    public int getBatchSize() {
+        return batchSize;
+    }
 }