Make `MessagesManager#clear` asynchronous

service/src/main/java/org/whispersystems/textsecuregcm/controllers/DeviceController.java

@@ -142,10 +142,10 @@ public class DeviceController {
 
     final CompletableFuture<Void> deleteKeysFuture = keys.delete(account.getUuid(), deviceId);
 
-    messages.clear(account.getUuid(), deviceId);
+    messages.clear(account.getUuid(), deviceId).join();
     account = accounts.update(account, a -> a.removeDevice(deviceId));
     // ensure any messages that came in after the first clear() are also removed
-    messages.clear(account.getUuid(), deviceId);
+    messages.clear(account.getUuid(), deviceId).join();
 
     deleteKeysFuture.join();
   }
@@ -419,7 +419,7 @@ public class DeviceController {
           keys.delete(a.getUuid(), device.getId()),
           keys.delete(a.getPhoneNumberIdentifier(), device.getId()));
 
-      messages.clear(a.getUuid(), device.getId());
+      messages.clear(a.getUuid(), device.getId()).join();
 
       deleteKeysFuture.join();
 

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

@@ -235,7 +235,7 @@ public class AccountsManager {
               keysManager.delete(actualUuid),
               keysManager.delete(account.getPhoneNumberIdentifier()));
 
-          messagesManager.clear(actualUuid);
+          messagesManager.clear(actualUuid).join();
           profilesManager.deleteAll(actualUuid);
 
           deleteKeysFuture.join();
@@ -877,12 +877,15 @@ public class AccountsManager {
         keysManager.delete(account.getUuid()),
         keysManager.delete(account.getPhoneNumberIdentifier()));
 
+    final CompletableFuture<Void> deleteMessagesFuture = CompletableFuture.allOf(
+        messagesManager.clear(account.getUuid()),
+        messagesManager.clear(account.getPhoneNumberIdentifier()));
+
     profilesManager.deleteAll(account.getUuid());
-    messagesManager.clear(account.getUuid());
-    messagesManager.clear(account.getPhoneNumberIdentifier());
     registrationRecoveryPasswordsManager.removeForNumber(account.getNumber());
 
     deleteKeysFuture.join();
+    deleteMessagesFuture.join();
     deleteStorageServiceDataFuture.join();
     deleteBackupServiceDataFuture.join();
     deleteSecureValueRecoveryServiceDataFuture.join();

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

@@ -102,14 +102,16 @@ public class MessagesManager {
         .tap(Micrometer.metrics(Metrics.globalRegistry));
   }
 
-  public void clear(UUID destinationUuid) {
+  public CompletableFuture<Void> clear(UUID destinationUuid) {
-    messagesCache.clear(destinationUuid).join();
+    return CompletableFuture.allOf(
-    messagesDynamoDb.deleteAllMessagesForAccount(destinationUuid).join();
+        messagesCache.clear(destinationUuid),
+        messagesDynamoDb.deleteAllMessagesForAccount(destinationUuid));
   }
 
-  public void clear(UUID destinationUuid, long deviceId) {
+  public CompletableFuture<Void> clear(UUID destinationUuid, long deviceId) {
-    messagesCache.clear(destinationUuid, deviceId).join();
+    return CompletableFuture.allOf(
-    messagesDynamoDb.deleteAllMessagesForDevice(destinationUuid, deviceId).join();
+        messagesCache.clear(destinationUuid, deviceId),
+        messagesDynamoDb.deleteAllMessagesForDevice(destinationUuid, deviceId));
   }
 
   public CompletableFuture<Optional<Envelope>> delete(UUID destinationUuid, long destinationDeviceId, UUID guid,

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

@@ -68,7 +68,7 @@ public class UnlinkDeviceCommand extends EnvironmentCommand<WhisperServerConfigu
     deps.keysManager().delete(account.getUuid(), deviceId).join();
 
     System.out.format("Clearing additional messages for %s::%d\n", aci, deviceId);
-    deps.messagesManager().clear(account.getUuid(), deviceId);
+    deps.messagesManager().clear(account.getUuid(), deviceId).join();
 
     System.out.format("Clearing presence state for %s::%d\n", aci, deviceId);
     deps.clientPresenceManager().disconnectPresence(aci, deviceId);

service/src/test/java/org/whispersystems/textsecuregcm/controllers/DeviceControllerTest.java

@@ -128,6 +128,8 @@ class DeviceControllerTest {
 
     when(keysManager.storePqLastResort(any(), any())).thenReturn(CompletableFuture.completedFuture(null));
     when(keysManager.delete(any(), anyLong())).thenReturn(CompletableFuture.completedFuture(null));
+
+    when(messagesManager.clear(any(), anyLong())).thenReturn(CompletableFuture.completedFuture(null));
   }
 
   @AfterEach

service/src/test/java/org/whispersystems/textsecuregcm/storage/AccountsManagerChangeNumberIntegrationTest.java

@@ -103,6 +103,9 @@ class AccountsManagerChangeNumberIntegrationTest {
       final KeysManager keysManager = mock(KeysManager.class);
       when(keysManager.delete(any())).thenReturn(CompletableFuture.completedFuture(null));
 
+      final MessagesManager messagesManager = mock(MessagesManager.class);
+      when(messagesManager.clear(any())).thenReturn(CompletableFuture.completedFuture(null));
+
       accountsManager = new AccountsManager(
           accounts,
           phoneNumberIdentifiers,
@@ -110,7 +113,7 @@ class AccountsManagerChangeNumberIntegrationTest {
           accountLockManager,
           deletedAccounts,
           keysManager,
-          mock(MessagesManager.class),
+          messagesManager,
           mock(ProfilesManager.class),
           secureStorageClient,
           secureBackupClient,

service/src/test/java/org/whispersystems/textsecuregcm/storage/AccountsManagerTest.java

@@ -188,6 +188,7 @@ class AccountsManagerTest {
     }).when(accountLockManager).withLock(any(), any());
 
     when(keysManager.delete(any())).thenReturn(CompletableFuture.completedFuture(null));
+    when(messagesManager.clear(any())).thenReturn(CompletableFuture.completedFuture(null));
 
     accountsManager = new AccountsManager(
         accounts,