如何使用NSUserData本地保存图片？iOS图片持久化求助

你的代码已经实现了图片本地存储，但缺少持久化记录已保存图片ID的逻辑，导致重启后无法恢复数据。下面是具体修改方案，同时修复LocalFileManager中的路径错误（之前用appendingPathExtension是错误的，应该用appendingPathComponent来创建文件夹/文件路径）：

一、核心思路 #

1. 用UserDefaults保存两个关键数据：countImage（图片计数）和imageIDs（已保存的图片ID列表），这两个数据仅存于内存，重启后会丢失，需要持久化。
2. 启动时从UserDefaults恢复这两个数据，再根据imageIDs从本地加载图片到imageArray。
3. 每次添加新图片后，更新UserDefaults中的数据。
4. 修复LocalFileManager的路径生成逻辑，确保图片正确存入本地目录。

二、修改ViewController.swift #

1. 移除全局变量，改为ViewController实例属性 #

把原来的全局变量移到ViewController内部，避免全局状态混乱：

class ViewController: UIViewController, PHPickerViewControllerDelegate {
    // 替换全局变量为实例属性
    private var imageArray = [UIImage]()
    private var imageIDs = [String]()
    private var countImage = 0
    
    let userDefaults = UserDefaults.standard
    @IBOutlet var collectionView: UICollectionView!
    
    // ... 其他原有代码不变
}

2. 在viewDidLoad中恢复持久化数据 #

添加从UserDefaults加载数据的逻辑，同时加载本地图片到内存：

override func viewDidLoad() {
    super.viewDidLoad()
    
    // 恢复持久化数据
    countImage = userDefaults.integer(forKey: "countImage")
    imageIDs = userDefaults.stringArray(forKey: "imageIDs") ?? []
    
    // 从本地加载图片到imageArray
    imageIDs.forEach { imageID in
        if let image = LocalFileManager.instance.getImage(imageName: imageID, folderName: "closet_folder") {
            imageArray.append(image)
        }
    }
    
    // ... 原有的collectionView配置代码不变
}

3. 添加图片后保存数据到UserDefaults #

在picker(_:didFinishPicking:)的group.notify中，添加持久化逻辑：

group.notify(queue: .main) {
    self.collectionView.reloadData()
    
    // 保存数据到UserDefaults
    self.userDefaults.set(self.countImage, forKey: "countImage")
    self.userDefaults.set(self.imageIDs, forKey: "imageIDs")
    self.userDefaults.synchronize() // 可选，确保立即写入磁盘
}

4. 修正cellForItemAt中的索引错误 #

原来的imageIDs[indexPath.row + 1]是错误的，现在imageIDs直接对应imageArray的索引，改为：

func collectionView(_ collectionView: UICollectionView, cellForItemAt indexPath: IndexPath) -> UICollectionViewCell {
    let cell = collectionView.dequeueReusableCell(withReuseIdentifier: "ClosetCollectionViewCell", for: indexPath) as! ClosetCollectionViewCell
    // 直接用内存中的图片，或从本地加载（适合大量图片场景）
    cell.imageView.image = imageArray[indexPath.row]
    // 大量图片建议用下面的方式，减少内存占用：
    // cell.imageView.image = LocalFileManager.instance.getImage(imageName: imageIDs[indexPath.row], folderName: "closet_folder")
    return cell
}

三、修改LocalFileManager.swift #

修复路径生成错误，把appendingPathExtension改为appendingPathComponent，同时给图片添加格式后缀：

// get URL for that folder
private func getURLforFolder(folderName: String) -> URL? {
    guard let url = FileManager.default.urls(for: .cachesDirectory, in: .userDomainMask).first
    else {
        return nil
    }
    // 替换appendingPathExtension为appendingPathComponent，正确创建文件夹路径
    return url.appendingPathComponent(folderName)
}

// generate URL for images in folder
private func getURLforImage(imageName: String, folderName: String) -> URL? {
    guard let folderURL = getURLforFolder(folderName: folderName) else {
        return nil
    }
    // 给图片添加.png后缀，确保文件格式正确
    return folderURL.appendingPathComponent("\(imageName).png")
}

同时优化saveImage的错误提示：

func saveImage(image: UIImage, imageName: String, folderName: String) {
    createFolderIfNeeded(folderName: folderName)
    
    guard
        let data = image.pngData(),
        let url = getURLforImage(imageName: imageName, folderName: folderName)
    else { return }
    
    do {
        try data.write(to: url)
    } catch let error {
        print("保存图片失败 \(imageName): \(error)")
    }
}

四、补充说明 #

• UserDefaults适合存储轻量数据（比如ID列表、计数），不适合直接存图片数据，你的方案用本地文件存图片、UserDefaults存ID是合理的。
• 修复路径错误后，图片会正确保存在caches/closet_folder/目录下，重启后能正常读取。
• 如果图片数量较多，建议不要把所有图片加载到imageArray，而是在cellForItemAt中直接从LocalFileManager加载，减少内存占用。

内容的提问来源于stack exchange，提问作者Jenni Joo