From 8c5811581ecf3b9c9d032da32854bf0f5a864202 Mon Sep 17 00:00:00 2001 From: Greyson Parrelli Date: Mon, 21 Apr 2025 13:18:37 -0400 Subject: [PATCH] Add additional logging around storage batch sizes. --- .../signalservice/api/storage/StorageServiceRepository.kt | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/libsignal-service/src/main/java/org/whispersystems/signalservice/api/storage/StorageServiceRepository.kt b/libsignal-service/src/main/java/org/whispersystems/signalservice/api/storage/StorageServiceRepository.kt index 0b6bd1a8de..f84d3604a7 100644 --- a/libsignal-service/src/main/java/org/whispersystems/signalservice/api/storage/StorageServiceRepository.kt +++ b/libsignal-service/src/main/java/org/whispersystems/signalservice/api/storage/StorageServiceRepository.kt @@ -11,6 +11,7 @@ import okio.ByteString.Companion.toByteString import okio.IOException import org.signal.core.util.bytes import org.signal.core.util.isNotEmpty +import org.signal.core.util.logging.Log import org.signal.libsignal.protocol.InvalidKeyException import org.whispersystems.signalservice.api.NetworkResult import org.whispersystems.signalservice.api.push.exceptions.NonSuccessfulResponseCodeException @@ -30,6 +31,8 @@ import java.lang.Exception class StorageServiceRepository(private val storageServiceApi: StorageServiceApi) { companion object { + private val TAG = Log.tag(StorageServiceRepository::class) + private const val STORAGE_READ_MAX_ITEMS: Int = 1000 } @@ -116,6 +119,9 @@ class StorageServiceRepository(private val storageServiceApi: StorageServiceApi) val knownIds = storageIds.filterNot { it.isUnknown } val batches = knownIds.chunked(STORAGE_READ_MAX_ITEMS) + + Log.d(TAG, "Reading ${knownIds.size} storage records in ${batches.size} batches of size $STORAGE_READ_MAX_ITEMS") + val results = batches.map { batch -> readStorageRecordsBatch(auth, storageKey, recordIkm, batch) }