From 671b0d3b6781a3dcba88d8a815be652103049d2a Mon Sep 17 00:00:00 2001 From: Pete Walters Date: Wed, 31 Jan 2024 13:44:10 -0600 Subject: [PATCH] Use the taskQueue for checkPreKey operations --- .../src/Account/PreKeys/PrekeyManagerImpl.swift | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/SignalServiceKit/src/Account/PreKeys/PrekeyManagerImpl.swift b/SignalServiceKit/src/Account/PreKeys/PrekeyManagerImpl.swift index 4283685e68..323ff070b3 100644 --- a/SignalServiceKit/src/Account/PreKeys/PrekeyManagerImpl.swift +++ b/SignalServiceKit/src/Account/PreKeys/PrekeyManagerImpl.swift @@ -133,12 +133,15 @@ public class PreKeyManagerImpl: PreKeyManager { let shouldPerformPniOp = hasPniIdentityKey(tx: tx) Task { [weak self, taskManager, targets] in - try Task.checkCancellation() - try await taskManager.refresh(identity: .aci, targets: targets, auth: .implicit()) - if shouldPerformPniOp { + let task = await Self.taskQueue.enqueue { try Task.checkCancellation() - try await taskManager.refresh(identity: .pni, targets: targets, auth: .implicit()) + try await taskManager.refresh(identity: .aci, targets: targets, auth: .implicit()) + if shouldPerformPniOp { + try Task.checkCancellation() + try await taskManager.refresh(identity: .pni, targets: targets, auth: .implicit()) + } } + try await task.value self?.refreshPreKeysDidSucceed() } }