From 5f898a907185611d7f718797940ab709704f7e70 Mon Sep 17 00:00:00 2001
From: Jon Chambers <jon@signal.org>
Date: Fri, 17 Jul 2020 10:30:04 -0400
Subject: [PATCH] Measure inserted message batch size.

---
 .../org/whispersystems/textsecuregcm/storage/Messages.java     | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/service/src/main/java/org/whispersystems/textsecuregcm/storage/Messages.java b/service/src/main/java/org/whispersystems/textsecuregcm/storage/Messages.java
index 3b9b3b49c..2f9e02136 100644
--- a/service/src/main/java/org/whispersystems/textsecuregcm/storage/Messages.java
+++ b/service/src/main/java/org/whispersystems/textsecuregcm/storage/Messages.java
@@ -1,5 +1,6 @@
 package org.whispersystems.textsecuregcm.storage;
 
+import com.codahale.metrics.Histogram;
 import com.codahale.metrics.MetricRegistry;
 import com.codahale.metrics.SharedMetricRegistries;
 import com.codahale.metrics.Timer;
@@ -43,6 +44,7 @@ public class Messages {
   private final Timer          clearDeviceTimer    = metricRegistry.timer(name(Messages.class, "clearDevice"   ));
   private final Timer          clearTimer          = metricRegistry.timer(name(Messages.class, "clear"         ));
   private final Timer          vacuumTimer         = metricRegistry.timer(name(Messages.class, "vacuum"));
+  private final Histogram      storeSizeHistogram  = metricRegistry.histogram(name(Messages.class, "storeBatchSize"));
 
   private final FaultTolerantDatabase database;
 
@@ -74,6 +76,7 @@ public class Messages {
         }
 
         batch.execute();
+        storeSizeHistogram.update(messages.size());
       }
     }));
   }