From dd552e8e8f23ce412365d5f97fdf7aa12871aa9b Mon Sep 17 00:00:00 2001
From: Chris Eager <chris@signal.org>
Date: Tue, 30 May 2023 09:43:46 -0500
Subject: [PATCH] Change `MessagePersisterServiceCommand` to extend
 `ServerCommand`

---
 .../workers/MessagePersisterServiceCommand.java             | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/service/src/main/java/org/whispersystems/textsecuregcm/workers/MessagePersisterServiceCommand.java b/service/src/main/java/org/whispersystems/textsecuregcm/workers/MessagePersisterServiceCommand.java
index 081ed0772..467adfef1 100644
--- a/service/src/main/java/org/whispersystems/textsecuregcm/workers/MessagePersisterServiceCommand.java
+++ b/service/src/main/java/org/whispersystems/textsecuregcm/workers/MessagePersisterServiceCommand.java
@@ -6,7 +6,7 @@
 package org.whispersystems.textsecuregcm.workers;
 
 import io.dropwizard.Application;
-import io.dropwizard.cli.EnvironmentCommand;
+import io.dropwizard.cli.ServerCommand;
 import io.dropwizard.setup.Environment;
 import java.time.Duration;
 import java.util.Optional;
@@ -19,7 +19,7 @@ import org.whispersystems.textsecuregcm.storage.DynamicConfigurationManager;
 import org.whispersystems.textsecuregcm.storage.MessagePersister;
 import org.whispersystems.textsecuregcm.util.logging.UncaughtExceptionHandler;
 
-public class MessagePersisterServiceCommand extends EnvironmentCommand<WhisperServerConfiguration> {
+public class MessagePersisterServiceCommand extends ServerCommand<WhisperServerConfiguration> {
 
   private static final String WORKER_COUNT = "workers";
 
@@ -69,6 +69,8 @@ public class MessagePersisterServiceCommand extends EnvironmentCommand<WhisperSe
 
     environment.lifecycle().manage(deps.messagesCache());
     environment.lifecycle().manage(messagePersister);
+
+    super.run(environment, namespace, configuration);
   }
 
 }