在CoreData上保存和检索UIImage

import CoreData

extension ImageEntity {
    @NSManaged var image: Data?
}

func saveImage(image: UIImage, context: NSManagedObjectContext) {
    let imageEntity = NSEntityDescription.insertNewObject(forEntityName: "ImageEntity", into: context) as! ImageEntity
    imageEntity.image = image.pngData()
    do {
        try context.save()
    } catch {
        print("Error saving image: \(error.localizedDescription)")
    }
}

func fetchImage(context: NSManagedObjectContext) -> UIImage? {
    let fetchRequest = NSFetchRequest<NSFetchRequestResult>(entityName: "ImageEntity")
    do {
        let results = try context.fetch(fetchRequest)
        if let imageEntity = results.first as? ImageEntity {
            return UIImage(data: imageEntity.image!)
        }
    } catch {
        print("Error fetching image: \(error.localizedDescription)")
    }
    return nil
}