Fix migration

This commit is contained in:
nielsandriesse 2020-05-09 09:28:25 +10:00
parent 0a41aafe45
commit de5116dd38
1 changed files with 6 additions and 7 deletions

View File

@ -20,13 +20,13 @@ public class LK001UpdateFriendRequestStatusStorage : OWSDatabaseMigration {
private func doMigrationAsync(completion: @escaping OWSDatabaseMigrationCompletion) { private func doMigrationAsync(completion: @escaping OWSDatabaseMigrationCompletion) {
DispatchQueue.global().async { DispatchQueue.global().async {
self.dbReadWriteConnection().readWrite { transaction in self.dbReadWriteConnection().readWrite { transaction in
guard let threads = TSThread.allObjectsInCollection() as? [TSThread] else { var threads: [TSContactThread] = []
owsFailDebug("Failed to convert objects to TSThread.") TSContactThread.enumerateCollectionObjects(with: transaction) { object, _ in
return guard let thread = object as? TSContactThread else { return }
threads.append(thread)
} }
for thread in threads { threads.forEach { thread in
guard let thread = thread as? TSContactThread, guard let friendRequestStatus = LKFriendRequestStatus(rawValue: thread.friendRequestStatus) else { return }
let friendRequestStatus = LKFriendRequestStatus(rawValue: thread.friendRequestStatus) else { continue }
OWSPrimaryStorage.shared().setFriendRequestStatus(friendRequestStatus, for: thread.contactIdentifier(), transaction: transaction) OWSPrimaryStorage.shared().setFriendRequestStatus(friendRequestStatus, for: thread.contactIdentifier(), transaction: transaction)
} }
self.save(with: transaction) self.save(with: transaction)
@ -34,5 +34,4 @@ public class LK001UpdateFriendRequestStatusStorage : OWSDatabaseMigration {
completion() completion()
} }
} }
} }