Parameterize worker thread count in CrawlAccountsCommand

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

@@ -11,6 +11,7 @@
 import io.dropwizard.cli.EnvironmentCommand;
 import io.dropwizard.setup.Environment;
 import java.util.List;
+import java.util.Objects;
 import java.util.concurrent.ExecutorService;
 import net.sourceforge.argparse4j.inf.Argument;
 import net.sourceforge.argparse4j.inf.ArgumentParser;
@@ -36,6 +37,7 @@
 public class CrawlAccountsCommand extends EnvironmentCommand<WhisperServerConfiguration> {
 
   private static final String CRAWL_TYPE = "crawlType";
+  private static final String WORKER_COUNT = "workers";
 
   public enum CrawlType implements ArgumentType<CrawlType> {
     GENERAL_PURPOSE,
@@ -67,6 +69,12 @@ public void configure(final Subparser subparser) {
         .dest(CRAWL_TYPE)
         .required(true)
         .help("The type of crawl to perform");
+
+    subparser.addArgument("--workers")
+        .type(Integer.class)
+        .dest(WORKER_COUNT)
+        .required(true)
+        .help("The number of worker threads");
   }
 
   @Override
@@ -99,6 +107,8 @@ protected void run(final Environment environment, final Namespace namespace,
     dynamicConfigurationManager.start();
     MetricsUtil.registerSystemResourceMetrics(environment);
 
+    final int workers = Objects.requireNonNull(namespace.getInt(WORKER_COUNT));
+
     final AccountDatabaseCrawler crawler = switch ((CrawlType) namespace.get(CRAWL_TYPE)) {
       case GENERAL_PURPOSE -> {
         final AccountDatabaseCrawlerCache accountDatabaseCrawlerCache = new AccountDatabaseCrawlerCache(
@@ -114,7 +124,7 @@ yield new AccountDatabaseCrawler("General-purpose account crawler",
       }
       case ACCOUNT_CLEANER -> {
         final ExecutorService accountDeletionExecutor = environment.lifecycle()
-            .executorService(name(getClass(), "accountCleaner-%d")).maxThreads(16).minThreads(16).build();
+            .executorService(name(getClass(), "accountCleaner-%d")).maxThreads(workers).minThreads(workers).build();
 
         final AccountDatabaseCrawlerCache accountDatabaseCrawlerCache = new AccountDatabaseCrawlerCache(
             cacheCluster, AccountDatabaseCrawlerCache.ACCOUNT_CLEANER_PREFIX);