Add additional logging around storage batch sizes.

This commit is contained in:
Greyson Parrelli
2025-04-21 13:18:37 -04:00
committed by Cody Henthorne
parent 4b4d3d33b1
commit 8c5811581e

View File

@@ -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)
}