如何将API获取的图片存入Core Data？urlString为nil问题排查

我想要实现将从API获取的图片保存到Core Data的功能，尝试了以下Swift代码但运行失败，调试时发现urlString为nil。请问问题出在哪里，该如何修复？

我尝试的代码 #

func downloadImage(from urlString: String, completed: @escaping (UIImage?, Bool) -> Void) {
    let cacheKey = NSString(string: urlString)
    let updatedURLString = urlString.replacingOccurrences(of: "312x231", with: "636x393")
    
    if let image = loadImageFromCoreData(with: cacheKey as String) {
        completed(image, true)
        return
    }
    
    guard let url = URL(string: updatedURLString) else {
        completed(nil, false)
        return
    }
    
    let task = URLSession.shared.dataTask(with: url) { [weak self] data, response, error in
        guard let self = self,
              error == nil,
              let response = response as? HTTPURLResponse, response.statusCode == 200,
              let data = data,
              let image = UIImage(data: data) else {
            completed(nil, false)
            return
        }
        
        self.saveImageToCoreData(image, with: cacheKey as NSString)
        completed(image, false)
    }
    task.resume()
}


func loadImageFromCoreData(with cacheKey: String) -> UIImage? {
    // Assuming RecipeImage is your Core Data entity
    let fetchRequest: NSFetchRequest<RecipeImage> = RecipeImage.fetchRequest()
    fetchRequest.predicate = NSPredicate(format: "cacheKey == %@", cacheKey)
    fetchRequest.fetchLimit = 1
    
    do {
        let result = try HomeVC().context.fetch(fetchRequest)
        if let recipeImage = result.first, let imageData = recipeImage.imageData {
            return UIImage(data: imageData)
        }
    } catch {
        print("Error fetching image from Core Data: \(error)")
    }
    
    return nil
}


func saveImageToCoreData(_ image: UIImage, with cacheKey: NSString) {
    let context = HomeVC().context
    let newImage = RecipeImage(context: context)
    newImage.cacheKey = String(cacheKey)
    newImage.imageData = image.pngData()
    
    do {
        try context.save()
    } catch {
        print("Error saving image to Core Data: \(error)")
    }
}

问题分析与修复方案 #

1. 核心问题：urlString为nil的根源 #

函数downloadImage的参数urlString定义为非可选String，但调试时发现它为nil，说明调用该函数时传入的参数本身就是nil。Swift中直接给非可选参数传nil会触发崩溃，你能调试到这一步，大概率是调用时通过隐式解包或其他方式绕过了编译检查。

2. 代码中的其他致命问题 #

• 每次新建HomeVC实例获取Context：HomeVC().context会创建全新的视图控制器实例，它的Context和你实际使用的Core Data上下文不是同一个，导致Core Data的读写操作完全无效。
• 图片转Data未处理nil：image.pngData()可能返回nil（比如图片是JPEG格式或损坏时），直接赋值会导致Core Data保存空数据。
• 缺少错误日志：出现问题时没有打印具体错误信息，难以快速定位问题。

3. 修复后的完整代码 #

修正downloadImage函数（处理可选参数+添加日志）

func downloadImage(from urlString: String?, completed: @escaping (UIImage?, Bool) -> Void) {
    // 先校验urlString非空
    guard let urlString = urlString, !urlString.isEmpty else {
        completed(nil, false)
        print("错误：urlString为空或nil")
        return
    }
    
    let cacheKey = urlString
    let updatedURLString = urlString.replacingOccurrences(of: "312x231", with: "636x393")
    
    if let image = loadImageFromCoreData(with: cacheKey) {
        completed(image, true)
        return
    }
    
    guard let url = URL(string: updatedURLString) else {
        completed(nil, false)
        print("错误：无法从字符串生成URL：\(updatedURLString)")
        return
    }
    
    let task = URLSession.shared.dataTask(with: url) { [weak self] data, response, error in
        guard let self = self,
              error == nil,
              let response = response as? HTTPURLResponse, response.statusCode == 200,
              let data = data,
              let image = UIImage(data: data) else {
            completed(nil, false)
            print("获取图片失败：\(error?.localizedDescription ?? "未知错误")")
            return
        }
        
        self.saveImageToCoreData(image, with: cacheKey)
        completed(image, false)
    }
    task.resume()
}

修正loadImageFromCoreData函数（使用共享Context）

// 注意：确保当前类拥有一个有效的NSManagedObjectContext属性（比如从父类或依赖注入获取）
func loadImageFromCoreData(with cacheKey: String) -> UIImage? {
    let fetchRequest: NSFetchRequest<RecipeImage> = RecipeImage.fetchRequest()
    fetchRequest.predicate = NSPredicate(format: "cacheKey == %@", cacheKey)
    fetchRequest.fetchLimit = 1
    
    do {
        // 使用当前类的context，而非新建HomeVC实例
        let result = try context.fetch(fetchRequest)
        if let recipeImage = result.first, let imageData = recipeImage.imageData {
            return UIImage(data: imageData)
        }
    } catch {
        print("从Core Data读取图片失败：\(error)")
    }
    
    return nil
}

修正saveImageToCoreData函数（处理Data nil+共享Context）

func saveImageToCoreData(_ image: UIImage, with cacheKey: String) {
    // 使用当前类的context
    let newImage = RecipeImage(context: context)
    newImage.cacheKey = cacheKey
    
    // 优先保存PNG，失败则保存JPEG
    if let pngData = image.pngData() {
        newImage.imageData = pngData
    } else if let jpegData = image.jpegData(compressionQuality: 1.0) {
        newImage.imageData = jpegData
    } else {
        print("错误：图片无法转换为Data")
        return
    }
    
    do {
        try context.save()
    } catch {
        print("保存图片到Core Data失败：\(error)")
    }
}

4. 额外注意事项 #

• 调用downloadImage时，确保传入的urlString是有效的非空字符串，比如从API返回的字段中先做非空校验。
• 确保context属性是正确的Core Data上下文（比如从AppDelegate的PersistentContainer或视图控制器的viewContext获取）。

内容的提问来源于stack exchange，提问作者Ali Düşkün