如何在后台线程获取图片URL分辨率并主线程返回Cell高度？

1. 重构分辨率获取函数为异步版本 #

先把同步的resolutionForImage改成异步回调形式，将IO操作放到后台线程执行，结果回调回主线程，同时加入缓存避免重复请求：

// 全局缓存，用于存储已获取的图片分辨率
private let imageResolutionCache = NSCache<NSString, NSValue>()

public func resolutionForImage(url: String, completion: @escaping (CGSize?) -> Void) {
    // 优先读取缓存，避免重复请求
    if let cachedSize = imageResolutionCache.object(forKey: url as NSString)?.cgSizeValue {
        DispatchQueue.main.async { completion(cachedSize) }
        return
    }
    
    // 后台线程执行图片属性读取操作
    DispatchQueue.global(qos: .userInitiated).async {
        guard let url = URL(string: url) else {
            DispatchQueue.main.async { completion(nil) }
            return
        }
        guard let source = CGImageSourceCreateWithURL(url as CFURL, nil) else {
            DispatchQueue.main.async { completion(nil) }
            return
        }
        
        let propertiesOptions = [kCGImageSourceShouldCache: false] as CFDictionary
        guard let properties = CGImageSourceCopyPropertiesAtIndex(source, 0, propertiesOptions) as? [CFString: Any],
              let width = properties[kCGImagePropertyPixelWidth] as? CGFloat,
              let height = properties[kCGImagePropertyPixelHeight] as? CGFloat else {
            DispatchQueue.main.async { completion(nil) }
            return
        }
        
        let size = CGSize(width: width, height: height)
        // 将结果存入缓存
        imageResolutionCache.setObject(NSValue(cgSize: size), forKey: url as NSString)
        
        // 主线程回调结果
        DispatchQueue.main.async {
            completion(size)
        }
    }
}

2. 修改Cell的高度计算函数为异步形式 #

原同步返回的calculateImageHeight无法适配异步逻辑，改成闭包回调形式，确保最终在主线程返回计算后的高度：

public func calculateImageHeight(ratio: Double = 0.0, completion: @escaping (CGFloat) -> Void) {
    // 已有传入比例时，直接计算并回调
    if ratio != 0.0 {
        let height = imageViewWidth * ratio
        DispatchQueue.main.async {
            completion(height.isFinite ? height : 100)
        }
        return
    }
    
    // 异步获取分辨率后计算高度，weak self避免循环引用
    resolutionForImage(url: imageUrl) { [weak self] size in
        guard let self = self else { return }
        var calculatedRatio = ratio
        if let size = size {
            calculatedRatio = size.height / size.width
        }
        let height = self.imageViewWidth * calculatedRatio
        completion(height.isFinite ? height : 100)
    }
}

3. 在CollectionView中使用异步函数更新Cell #

在配置Cell时调用异步计算函数，更新Cell的高度约束并刷新CollectionView布局：

func collectionView(_ collectionView: UICollectionView, cellForItemAt indexPath: IndexPath) -> UICollectionViewCell {
    let cell = collectionView.dequeueReusableCell(withReuseIdentifier: "YourCellID", for: indexPath) as! YourCustomCell
    // 配置Cell的图片URL和显示宽度
    cell.imageUrl = yourDataArray[indexPath.item].imageUrl
    cell.imageViewWidth = collectionView.bounds.width // 根据实际布局调整宽度值
    
    // 异步计算并更新Cell高度
    cell.calculateImageHeight { [weak self] calculatedHeight in
        guard let self = self else { return }
        // 更新Cell的高度约束（假设Cell已配置imageHeightConstraint约束）
        cell.imageHeightConstraint.constant = calculatedHeight
        cell.layoutIfNeeded()
        
        // 刷新CollectionView布局，确保Cell高度正确生效
        self.collectionView.performBatchUpdates(nil, completion: nil)
    }
    
    return cell
}

关键注意点 #

• 用[weak self]避免循环引用，防止Cell被复用或销毁时出现内存泄漏
• 选择.userInitiated优先级的后台队列，平衡性能与用户体验
• 缓存机制可大幅减少重复网络IO，提升滚动流畅度
• 所有UI更新操作必须在主线程执行

内容的提问来源于stack exchange，提问作者martin wotterman