如何正确获取图像数据？AVFoundation异步拍照返回空数据问题

嗨，我来帮你搞定这个AVFoundation拍照的异步返回问题！你的核心困扰在于——AVFoundation的拍照API是异步执行的，但你想让函数同步返回图像数据，结果要么拿到空值，要么用等待的方式导致UI冻结。咱们先拆解一下问题，再给出两种实用的解决方案。

为什么你的原代码会失效？ #

你写的takePicture()函数里，captureStillImageAsynchronously(from:completionHandler:)是异步执行的：它会立刻返回，而实际的拍照和数据处理逻辑是在后台队列完成的。这就导致函数返回时，imageData还没被赋值，强制解包imageData!要么返回空，要么直接崩溃。

如果尝试用信号量等方式强制等待异步任务完成，又会阻塞主线程，造成UI完全冻结——这在iOS开发里是绝对要避免的，因为主线程负责所有UI更新，一旦阻塞超过几秒，系统甚至会触发App崩溃。

解决方案1：使用闭包回调（兼容所有Swift版本） #

把函数改成接受一个闭包参数，在异步拍照完成后，通过闭包把图像数据或错误传递出去。这种方式是iOS异步编程的传统方案，兼容性拉满。

改造后的代码： #

func takePicture(completion: @escaping (Data?, Error?) -> Void) {
    // 先检查必要的对象是否存在
    guard let stillImageOutput = self.stillImageOutput, let videoConnection = videoConnection else {
        let error = NSError(domain: "CaptureError", code: -1, userInfo: [NSLocalizedDescriptionKey: "拍照输出或连接不可用"])
        completion(nil, error)
        return
    }
    
    // 执行异步拍照
    stillImageOutput.captureStillImageAsynchronously(from: videoConnection) { [weak self] imageDataBuffer, error in
        guard let self = self else { return }
        
        // 优先处理错误
        if let error = error {
            completion(nil, error)
            return
        }
        
        // 检查图像缓冲区是否有效
        guard let buffer = imageDataBuffer else {
            let error = NSError(domain: "CaptureError", code: -2, userInfo: [NSLocalizedDescriptionKey: "未获取到图像缓冲区"])
            completion(nil, error)
            return
        }
        
        // 转换缓冲区为JPEG数据
        if let imageData = AVCapturePhotoOutput.jpegPhotoDataRepresentation(forJPEGSampleBuffer: buffer, previewPhotoSampleBuffer: nil) {
            completion(imageData, nil)
        } else {
            let error = NSError(domain: "CaptureError", code: -3, userInfo: [NSLocalizedDescriptionKey: "图像缓冲区转换JPEG失败"])
            completion(nil, error)
        }
        
        // 在这里处理Session停止等后续操作
        // self.stopCaptureSession()
    }
}

调用方式： #

// 在需要拍照的地方调用
takePicture { imageData, error in
    if let data = imageData {
        // 拿到图像数据后的处理逻辑，比如显示到UIImageView
        DispatchQueue.main.async {
            let image = UIImage(data: data)
            self.photoImageView.image = image
        }
    } else if let error = error {
        print("拍照出错：\(error.localizedDescription)")
    }
}

解决方案2：使用Async/Await（Swift 5.5+） #

如果你用的是Swift 5.5及以上版本，推荐用Async/Await语法，它能让异步代码看起来像同步代码，同时不会阻塞UI。我们可以用withCheckedThrowingContinuation把传统的闭包回调封装成异步函数。

改造后的代码： #

func takePicture() async throws -> Data {
    guard let stillImageOutput = self.stillImageOutput, let videoConnection = videoConnection else {
        throw NSError(domain: "CaptureError", code: -1, userInfo: [NSLocalizedDescriptionKey: "拍照输出或连接不可用"])
    }
    
    return try await withCheckedThrowingContinuation { continuation in
        stillImageOutput.captureStillImageAsynchronously(from: videoConnection) { [weak self] imageDataBuffer, error in
            guard let self = self else {
                continuation.resume(throwing: NSError(domain: "CaptureError", code: -4, userInfo: [NSLocalizedDescriptionKey: "对象已被释放"]))
                return
            }
            
            if let error = error {
                continuation.resume(throwing: error)
                return
            }
            
            guard let buffer = imageDataBuffer else {
                continuation.resume(throwing: NSError(domain: "CaptureError", code: -2, userInfo: [NSLocalizedDescriptionKey: "未获取到图像缓冲区"]))
                return
            }
            
            guard let imageData = AVCapturePhotoOutput.jpegPhotoDataRepresentation(forJPEGSampleBuffer: buffer, previewPhotoSampleBuffer: nil) else {
                continuation.resume(throwing: NSError(domain: "CaptureError", code: -3, userInfo: [NSLocalizedDescriptionKey: "图像缓冲区转换JPEG失败"]))
                return
            }
            
            continuation.resume(returning: imageData)
            
            // 处理Session停止等后续操作
            // self.stopCaptureSession()
        }
    }
}

调用方式： #

// 在Task里调用异步函数
Task {
    do {
        let imageData = try await takePicture()
        // 更新UI必须回到主线程
        DispatchQueue.main.async {
            let image = UIImage(data: imageData)
            self.photoImageView.image = image
        }
    } catch {
        print("拍照失败：\(error.localizedDescription)")
    }
}

一些额外注意点 #

• 确保videoConnection是有效的：可以通过stillImageOutput.connection(with: .video)来获取，并且检查isActive状态。
• 所有UI更新操作必须放在DispatchQueue.main.async里，因为异步回调默认在后台队列执行。
• Session的停止、资源释放等操作，要放在拍照完成后的回调里，避免提前停止导致拍照失败。

内容的提问来源于stack exchange，提问作者ɪ ᴀᴍ ᴀʀᴃɪᴏs