diff --git a/service/src/main/java/org/whispersystems/textsecuregcm/workers/CrawlAccountsCommand.java b/service/src/main/java/org/whispersystems/textsecuregcm/workers/CrawlAccountsCommand.java index 4f97347cd..29618a5fb 100644 --- a/service/src/main/java/org/whispersystems/textsecuregcm/workers/CrawlAccountsCommand.java +++ b/service/src/main/java/org/whispersystems/textsecuregcm/workers/CrawlAccountsCommand.java @@ -11,6 +11,7 @@ import io.dropwizard.Application; 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 @@ import org.whispersystems.textsecuregcm.util.logging.UncaughtExceptionHandler; public class CrawlAccountsCommand extends EnvironmentCommand { private static final String CRAWL_TYPE = "crawlType"; + private static final String WORKER_COUNT = "workers"; public enum CrawlType implements ArgumentType { GENERAL_PURPOSE, @@ -67,6 +69,12 @@ public class CrawlAccountsCommand extends EnvironmentCommand { final AccountDatabaseCrawlerCache accountDatabaseCrawlerCache = new AccountDatabaseCrawlerCache( @@ -114,7 +124,7 @@ public class CrawlAccountsCommand extends EnvironmentCommand { 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);