From 6a6555e2d5febea49b357298345318c8d023e086 Mon Sep 17 00:00:00 2001
From: Chris Eager <chris@signal.org>
Date: Wed, 8 Sep 2021 18:04:58 -0700
Subject: [PATCH] Add metrics for AuthEnablementRequestEventListener
 displacements

---
 .../auth/AuthEnablementRequestEventListener.java    | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/service/src/main/java/org/whispersystems/textsecuregcm/auth/AuthEnablementRequestEventListener.java b/service/src/main/java/org/whispersystems/textsecuregcm/auth/AuthEnablementRequestEventListener.java
index 51c5d458d..e4f16054f 100644
--- a/service/src/main/java/org/whispersystems/textsecuregcm/auth/AuthEnablementRequestEventListener.java
+++ b/service/src/main/java/org/whispersystems/textsecuregcm/auth/AuthEnablementRequestEventListener.java
@@ -5,7 +5,11 @@
 
 package org.whispersystems.textsecuregcm.auth;
 
+import static org.whispersystems.textsecuregcm.metrics.MetricsUtil.name;
+
 import com.google.common.annotations.VisibleForTesting;
+import io.micrometer.core.instrument.Counter;
+import io.micrometer.core.instrument.Metrics;
 import java.util.Collections;
 import java.util.HashMap;
 import java.util.HashSet;
@@ -45,6 +49,11 @@ public class AuthEnablementRequestEventListener implements RequestEventListener
   private static final String ACCOUNT_ENABLED = AuthEnablementRequestEventListener.class.getName() + ".accountEnabled";
   private static final String DEVICES_ENABLED = AuthEnablementRequestEventListener.class.getName() + ".devicesEnabled";
 
+  private static final Counter DISPLACED_ACCOUNTS = Metrics.counter(
+      name(AuthEnablementRequestEventListener.class, "displacedAccounts"));
+  private static final Counter DISPLACED_DEVICES = Metrics.counter(
+      name(AuthEnablementRequestEventListener.class, "displacedDevices"));
+
   private final ClientPresenceManager clientPresenceManager;
 
   public AuthEnablementRequestEventListener(final ClientPresenceManager clientPresenceManager) {
@@ -86,6 +95,8 @@ public class AuthEnablementRequestEventListener implements RequestEventListener
 
                 deviceIdsToDisplace.addAll(initialDevicesEnabled.keySet());
 
+                DISPLACED_ACCOUNTS.increment();
+
               } else if (!initialDevicesEnabled.isEmpty()) {
 
                 deviceIdsToDisplace = new HashSet<>();
@@ -98,6 +109,8 @@ public class AuthEnablementRequestEventListener implements RequestEventListener
 
                   if (!enabledMatches) {
                     deviceIdsToDisplace.add(deviceId);
+
+                    DISPLACED_DEVICES.increment();
                   }
                 });
               } else {