session-ios/SessionMessagingKit/Database/Storage+Contacts.swift

19 lines
633 B
Swift
Raw Normal View History

2020-12-17 06:32:39 +01:00
import SessionProtocolKit
extension Storage {
private static let contactCollection = "LokiContactCollection"
public func getContact(with sessionID: String) -> Contact? {
var result: Contact?
Storage.read { transaction in
result = transaction.object(forKey: sessionID, inCollection: Storage.contactCollection) as? Contact
}
return result
}
public func setContact(_ contact: Contact, using transaction: Any) {
(transaction as! YapDatabaseReadWriteTransaction).setObject(contact, forKey: contact.sessionID, inCollection: Storage.contactCollection)
}
}