Add a task to enable/disable accelerated crawling.

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

@@ -150,6 +150,7 @@ import org.whispersystems.textsecuregcm.workers.DisableRequestLoggingTask;
 import org.whispersystems.textsecuregcm.workers.EnableRequestLoggingTask;
 import org.whispersystems.textsecuregcm.workers.GetRedisCommandStatsCommand;
 import org.whispersystems.textsecuregcm.workers.GetRedisSlowlogCommand;
+import org.whispersystems.textsecuregcm.workers.SetCrawlerAccelerationTask;
 import org.whispersystems.textsecuregcm.workers.VacuumCommand;
 import org.whispersystems.textsecuregcm.workers.ZkParamsCommand;
 import org.whispersystems.websocket.WebSocketResourceProviderFactory;
@@ -448,6 +449,7 @@ public class WhisperServerService extends Application<WhisperServerConfiguration
 
     environment.admin().addTask(new EnableRequestLoggingTask());
     environment.admin().addTask(new DisableRequestLoggingTask());
+    environment.admin().addTask(new SetCrawlerAccelerationTask(accountDatabaseCrawlerCache));
 
 ///
 

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

@@ -122,7 +122,7 @@ public class AccountDatabaseCrawler implements Managed, Runnable {
       logger.info("Finished crawl");
       listeners.forEach(listener -> listener.onCrawlEnd(fromUuid));
       cache.setLastUuid(Optional.empty());
-      cache.clearAccelerate();
+      cache.setAccelerated(false);
     } else {
       try {
         for (AccountDatabaseCrawlerListener listener : listeners) {
@@ -131,7 +131,7 @@ public class AccountDatabaseCrawler implements Managed, Runnable {
         cache.setLastUuid(Optional.of(chunkAccounts.get(chunkAccounts.size() - 1).getUuid()));
       } catch (AccountDatabaseCrawlerRestartException e) {
         cache.setLastUuid(Optional.empty());
-        cache.clearAccelerate();
+        cache.setAccelerated(false);
       }
 
     }

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

@@ -31,8 +31,12 @@ public class AccountDatabaseCrawlerCache {
     this.unlockClusterScript = ClusterLuaScript.fromResource(cacheCluster, "lua/account_database_crawler/unlock.lua", ScriptOutputType.INTEGER);
   }
 
-  public void clearAccelerate() {
+  public void setAccelerated(final boolean accelerated) {
-    cacheCluster.useCluster(connection -> connection.sync().del(ACCELERATE_KEY));
+    if (accelerated) {
+      cacheCluster.useCluster(connection -> connection.sync().set(ACCELERATE_KEY, "1"));
+    } else {
+      cacheCluster.useCluster(connection -> connection.sync().del(ACCELERATE_KEY));
+    }
   }
 
   public boolean isAccelerated() {

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

@@ -0,0 +1,36 @@
+/*
+ * Copyright 2021 Signal Messenger, LLC
+ * SPDX-License-Identifier: AGPL-3.0-only
+ */
+
+package org.whispersystems.textsecuregcm.workers;
+
+import io.dropwizard.servlets.tasks.Task;
+import org.whispersystems.textsecuregcm.storage.AccountDatabaseCrawlerCache;
+
+import java.io.PrintWriter;
+import java.util.List;
+import java.util.Map;
+
+public class SetCrawlerAccelerationTask extends Task {
+
+    private final AccountDatabaseCrawlerCache crawlerCache;
+
+    public SetCrawlerAccelerationTask(final AccountDatabaseCrawlerCache crawlerCache) {
+        super("set-crawler-accelerated");
+
+        this.crawlerCache = crawlerCache;
+    }
+
+    @Override
+    public void execute(final Map<String, List<String>> parameters, final PrintWriter out) {
+        if (parameters.containsKey("accelerated") && parameters.get("accelerated").size() == 1) {
+            final boolean accelerated = "true".equalsIgnoreCase(parameters.get("accelerated").get(0));
+
+            crawlerCache.setAccelerated(accelerated);
+            out.println("Set accelerated: " + accelerated);
+        } else {
+            out.println("Usage: set-crawler-accelerated?accelerated=[true|false]");
+        }
+    }
+}

service/src/test/java/org/whispersystems/textsecuregcm/tests/storage/AccountDatabaseCrawlerTest.java

@@ -148,7 +148,7 @@ public class AccountDatabaseCrawlerTest {
     verify(account2, times(0)).getNumber();
     verify(listener, times(1)).timeAndProcessCrawlChunk(eq(Optional.of(ACCOUNT1)), eq(Arrays.asList(account2)));
     verify(cache, times(1)).setLastUuid(eq(Optional.empty()));
-    verify(cache, times(1)).clearAccelerate();
+    verify(cache, times(1)).setAccelerated(false);
     verify(cache, times(1)).isAccelerated();
     verify(cache, times(1)).releaseActiveWork(any(String.class));
 
@@ -175,7 +175,7 @@ public class AccountDatabaseCrawlerTest {
     verify(account2, times(0)).getNumber();
     verify(listener, times(1)).onCrawlEnd(eq(Optional.of(ACCOUNT2)));
     verify(cache, times(1)).setLastUuid(eq(Optional.empty()));
-    verify(cache, times(1)).clearAccelerate();
+    verify(cache, times(1)).setAccelerated(false);
     verify(cache, times(1)).isAccelerated();
     verify(cache, times(1)).releaseActiveWork(any(String.class));