Remove MessagePersister from WhisperServerService environment

Persistence is now exclusively done by a separate command.

service/src/main/java/org/whispersystems/textsecuregcm/WhisperServerService.java

@@ -163,7 +163,6 @@ import org.whispersystems.textsecuregcm.storage.DeletedAccounts;
 import org.whispersystems.textsecuregcm.storage.DynamicConfigurationManager;
 import org.whispersystems.textsecuregcm.storage.IssuedReceiptsManager;
 import org.whispersystems.textsecuregcm.storage.KeysManager;
-import org.whispersystems.textsecuregcm.storage.MessagePersister;
 import org.whispersystems.textsecuregcm.storage.MessagesCache;
 import org.whispersystems.textsecuregcm.storage.MessagesDynamoDb;
 import org.whispersystems.textsecuregcm.storage.MessagesManager;
@@ -547,9 +546,6 @@ public class WhisperServerService extends Application<WhisperServerConfiguration
     RateLimitChallengeManager rateLimitChallengeManager = new RateLimitChallengeManager(pushChallengeManager,
         captchaChecker, rateLimiters);
 
-    MessagePersister messagePersister = new MessagePersister(messagesCache, messagesManager, accountsManager,
-        dynamicConfigurationManager, Duration.ofMinutes(config.getMessageCacheConfiguration().getPersistDelayMinutes()),
-        Optional.empty());
     ChangeNumberManager changeNumberManager = new ChangeNumberManager(messageSender, accountsManager);
 
     HttpClient currencyClient = HttpClient.newBuilder().version(HttpClient.Version.HTTP_2).connectTimeout(Duration.ofSeconds(10)).build();
@@ -562,7 +558,6 @@ public class WhisperServerService extends Application<WhisperServerConfiguration
     environment.lifecycle().manage(apnPushNotificationScheduler);
     environment.lifecycle().manage(provisioningManager);
     environment.lifecycle().manage(messagesCache);
-    environment.lifecycle().manage(messagePersister);
     environment.lifecycle().manage(clientPresenceManager);
     environment.lifecycle().manage(currencyManager);
     environment.lifecycle().manage(registrationServiceClient);

service/src/main/java/org/whispersystems/textsecuregcm/configuration/dynamic/DynamicMessagePersisterConfiguration.java

@@ -10,16 +10,9 @@ import com.fasterxml.jackson.annotation.JsonProperty;
 public class DynamicMessagePersisterConfiguration {
 
   @JsonProperty
-  private boolean serverPersistenceEnabled = true;
+  private boolean persistenceEnabled = true;
 
-  @JsonProperty
-  private boolean dedicatedProcessEnabled = false;
-
-  public boolean isServerPersistenceEnabled() {
-    return serverPersistenceEnabled;
-  }
-
-  public boolean isDedicatedProcessEnabled() {
-    return dedicatedProcessEnabled;
+  public boolean isPersistenceEnabled() {
+    return persistenceEnabled;
   }
 }

service/src/main/java/org/whispersystems/textsecuregcm/storage/MessagePersister.java

@@ -22,7 +22,6 @@ import java.util.UUID;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.whispersystems.textsecuregcm.configuration.dynamic.DynamicConfiguration;
-import org.whispersystems.textsecuregcm.configuration.dynamic.DynamicMessagePersisterConfiguration;
 import org.whispersystems.textsecuregcm.entities.MessageProtos;
 import org.whispersystems.textsecuregcm.util.Constants;
 import org.whispersystems.textsecuregcm.util.Util;
@@ -52,8 +51,6 @@ public class MessagePersister implements Managed {
 
   private static final long EXCEPTION_PAUSE_MILLIS = Duration.ofSeconds(3).toMillis();
 
-  private static final int DEFAULT_WORKER_THREAD_COUNT = 4;
-
   private static final int CONSECUTIVE_EMPTY_CACHE_REMOVAL_LIMIT = 3;
 
   private static final Logger logger = LoggerFactory.getLogger(MessagePersister.class);
@@ -62,19 +59,19 @@ public class MessagePersister implements Managed {
       final AccountsManager accountsManager,
       final DynamicConfigurationManager<DynamicConfiguration> dynamicConfigurationManager,
       final Duration persistDelay,
-      final Optional<Integer> dedicatedProcessWorkerThreadCount) {
+      final int dedicatedProcessWorkerThreadCount) {
     this.messagesCache = messagesCache;
     this.messagesManager = messagesManager;
     this.accountsManager = accountsManager;
     this.persistDelay = persistDelay;
-    this.workerThreads = dedicatedProcessWorkerThreadCount.map(Thread[]::new)
-        .orElseGet(() -> new Thread[DEFAULT_WORKER_THREAD_COUNT]);
-    this.dedicatedProcess = dedicatedProcessWorkerThreadCount.isPresent();
+    this.workerThreads = new Thread[dedicatedProcessWorkerThreadCount];
+    this.dedicatedProcess = true;
 
     for (int i = 0; i < workerThreads.length; i++) {
       workerThreads[i] = new Thread(() -> {
         while (running) {
-          if (enabled(dynamicConfigurationManager)) {
+          if (dynamicConfigurationManager.getConfiguration().getMessagePersisterConfiguration()
+              .isPersistenceEnabled()) {
             try {
               final int queuesPersisted = persistNextQueues(Instant.now());
               queueCountHistogram.update(queuesPersisted);
@@ -94,17 +91,6 @@ public class MessagePersister implements Managed {
     }
   }
 
-  @VisibleForTesting
-  boolean enabled(final DynamicConfigurationManager<DynamicConfiguration> dynamicConfigurationManager) {
-    final DynamicMessagePersisterConfiguration messagePersisterConfiguration = dynamicConfigurationManager.getConfiguration()
-        .getMessagePersisterConfiguration();
-    if (dedicatedProcess) {
-      return messagePersisterConfiguration.isDedicatedProcessEnabled();
-    }
-
-    return messagePersisterConfiguration.isServerPersistenceEnabled();
-  }
-
   @VisibleForTesting
   Duration getPersistDelay() {
     return persistDelay;

service/src/main/java/org/whispersystems/textsecuregcm/workers/MessagePersisterServiceCommand.java

@@ -9,7 +9,6 @@ import io.dropwizard.Application;
 import io.dropwizard.cli.ServerCommand;
 import io.dropwizard.setup.Environment;
 import java.time.Duration;
-import java.util.Optional;
 import net.sourceforge.argparse4j.inf.Namespace;
 import net.sourceforge.argparse4j.inf.Subparser;
 import org.whispersystems.textsecuregcm.WhisperServerConfiguration;
@@ -65,7 +64,7 @@ public class MessagePersisterServiceCommand extends ServerCommand<WhisperServerC
         deps.accountsManager(),
         dynamicConfigurationManager,
         Duration.ofMinutes(configuration.getMessageCacheConfiguration().getPersistDelayMinutes()),
-        Optional.of(namespace.getInt(WORKER_COUNT)));
+        namespace.getInt(WORKER_COUNT));
 
     environment.lifecycle().manage(deps.messagesCache());
     environment.lifecycle().manage(messagePersister);