Set the shutdown gauge earlier in the shutdown process

Co-authored-by: Chris Eager <[email protected]>

service/src/main/java/org/whispersystems/textsecuregcm/metrics/ApplicationShutdownMonitor.java

@@ -8,16 +8,16 @@
 
 import static com.codahale.metrics.MetricRegistry.name;
 
-import io.dropwizard.lifecycle.Managed;
 import io.micrometer.core.instrument.Gauge;
 import io.micrometer.core.instrument.MeterRegistry;
 import java.util.concurrent.atomic.AtomicBoolean;
+import org.eclipse.jetty.util.component.LifeCycle;
 
 /**
  * A managed monitor that reports whether the application is shutting down as a metric. That metric can then be used in
  * conjunction with other indicators to conditionally fire or suppress alerts.
  */
-public class ApplicationShutdownMonitor implements Managed {
+public class ApplicationShutdownMonitor implements LifeCycle.Listener {
 
   private final AtomicBoolean shuttingDown = new AtomicBoolean(false);
 
@@ -30,12 +30,7 @@ public ApplicationShutdownMonitor(final MeterRegistry meterRegistry) {
   }
 
   @Override
-  public void start() throws Exception {
-    shuttingDown.set(false);
-  }
-
-  @Override
-  public void stop() throws Exception {
+  public void lifeCycleStopping(final LifeCycle event) {
     shuttingDown.set(true);
   }
 }

service/src/main/java/org/whispersystems/textsecuregcm/metrics/MetricsUtil.java

@@ -61,9 +61,9 @@ public static void configureRegistries(final WhisperServerConfiguration config,
       Metrics.addRegistry(dogstatsdMeterRegistry);
     }
 
+    environment.lifecycle().addEventListener(new ApplicationShutdownMonitor(Metrics.globalRegistry));
     environment.lifecycle().addEventListener(
         new MicrometerRegistryManager(Metrics.globalRegistry, config.getDatadogConfiguration().pollingFrequency()));
-    environment.lifecycle().manage(new ApplicationShutdownMonitor(Metrics.globalRegistry));
   }
 
   @VisibleForTesting