From b75456acf3a60f64cfcd376f6d8c47f0a9e2e90a Mon Sep 17 00:00:00 2001
From: Chris Eager <chris@signal.org>
Date: Wed, 21 Apr 2021 15:10:38 -0500
Subject: [PATCH] Allow migration thread pool to be scaled up

---
 .../textsecuregcm/storage/AccountsDynamoDb.java          | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/service/src/main/java/org/whispersystems/textsecuregcm/storage/AccountsDynamoDb.java b/service/src/main/java/org/whispersystems/textsecuregcm/storage/AccountsDynamoDb.java
index 739636084..1465aa7cb 100644
--- a/service/src/main/java/org/whispersystems/textsecuregcm/storage/AccountsDynamoDb.java
+++ b/service/src/main/java/org/whispersystems/textsecuregcm/storage/AccountsDynamoDb.java
@@ -260,8 +260,13 @@ public class AccountsDynamoDb extends AbstractDynamoDbStore implements AccountSt
 
   public CompletableFuture<Void> migrate(List<Account> accounts, int threads) {
 
-    migrationThreadPool.setCorePoolSize(threads);
-    migrationThreadPool.setMaximumPoolSize(threads);
+    if (threads > migrationThreadPool.getMaximumPoolSize()) {
+      migrationThreadPool.setMaximumPoolSize(threads);
+      migrationThreadPool.setCorePoolSize(threads);
+    } else {
+      migrationThreadPool.setCorePoolSize(threads);
+      migrationThreadPool.setMaximumPoolSize(threads);
+    }
 
     final List<CompletableFuture<?>> futures = accounts.stream()
         .map(this::migrate)