Fix migration

pull/190/head
nielsandriesse 6 years ago
parent 0a41aafe45
commit de5116dd38

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

Loading…
Cancel
Save