如何在上传文件至Firebase Storage时等待完成并返回文件URL

你遇到的是异步操作的典型坑：Firebase的putData和downloadURL都是后台异步执行的，但你的uploadImage方法是同步逻辑——在异步任务完成前，就直接返回了初始值为空的imageURL，所以拿到的自然是空字符串，等上传完成后再赋值imageURL已经晚了。

解决方案1：闭包回调（兼容iOS 13+） #

这是最通用的解决方案，把方法改成通过闭包异步返回结果，在上传和获取URL的所有异步步骤完成后，再通过闭包把结果传出来：

import Foundation
import FirebaseFirestore
import FirebaseAuth
import Firebase

class FirebaseStorageService {

    // 用闭包异步返回结果：成功返回URL，失败返回对应错误标识/信息
    static func uploadImage(image: UIImage, fileType: String, completion: @escaping (String) -> Void) {
        // 先检查用户是否登录
        guard let currentUser = Auth.auth().currentUser else {
            completion("UserNotLoggedIn")
            return
        }
        
        let uid = currentUser.uid
        let fileName = UUID().uuidString
        let storageReference = Storage.storage().reference().child("\(fileType)/\(uid)/\(fileName).jpg")
        
        // 转换图片为Data
        guard let photoData = image.jpegData(compressionQuality: 0.1) else {
            print("No photo provided")
            completion("NoPhoto")
            return
        }
        
        // 异步上传数据
        storageReference.putData(photoData, metadata: nil) { (metadata, err) in
            if let uploadErr = err {
                print("上传失败：\(uploadErr)")
                completion("UploadFailed: \(uploadErr.localizedDescription)")
                return
            }
            
            // 上传成功后，异步获取下载URL
            storageReference.downloadURL { (url, urlErr) in
                if let urlErr = urlErr {
                    print("获取URL失败：\(urlErr)")
                    completion("GetURLFailed: \(urlErr.localizedDescription)")
                    return
                }
                
                guard let downloadURL = url?.absoluteString else {
                    completion("URLNil")
                    return
                }
                
                print("Upload successful!")
                completion(downloadURL) // 在这里返回最终的URL
            }
        }
    }
}

调用示例： #

// 假设你有一个UIImage实例叫selectedImage
FirebaseStorageService.uploadImage(image: selectedImage, fileType: "profilePhotos") { result in
    // 处理返回结果
    if result.hasPrefix("http") {
        print("图片上传完成，URL：\(result)")
        // 这里可以把URL存入Firestore、更新UI等操作
        DispatchQueue.main.async {
            // 如果要更新UI，记得切回主线程
            // self.profileImageURL = result
        }
    } else {
        print("上传出错：\(result)")
    }
}

解决方案2：Async/Await（iOS 15+/macOS 12+） #

如果你的项目支持iOS 15及以上，推荐用Swift的异步等待语法，代码更简洁易读，不需要嵌套闭包：

import Foundation
import FirebaseFirestore
import FirebaseAuth
import Firebase

class FirebaseStorageService {

    static func uploadImage(image: UIImage, fileType: String) async throws -> String {
        guard let currentUser = Auth.auth().currentUser else {
            throw StorageError.userNotLoggedIn
        }
        
        let uid = currentUser.uid
        let fileName = UUID().uuidString
        let storageReference = Storage.storage().reference().child("\(fileType)/\(uid)/\(fileName).jpg")
        
        guard let photoData = image.jpegData(compressionQuality: 0.1) else {
            print("No photo provided")
            throw StorageError.noPhoto
        }
        
        // 异步等待上传完成
        _ = try await storageReference.putDataAsync(photoData, metadata: nil)
        
        // 异步等待获取下载URL
        let downloadURL = try await storageReference.downloadURL()
        
        print("Upload successful!")
        return downloadURL.absoluteString
    }
    
    // 自定义错误类型，方便区分不同失败场景
    enum StorageError: Error, LocalizedError {
        case userNotLoggedIn
        case noPhoto
        
        var errorDescription: String? {
            switch self {
            case .userNotLoggedIn:
                return "用户未登录"
            case .noPhoto:
                return "未提供有效图片"
            }
        }
    }
}

调用示例： #

在异步上下文（比如Task或者标记为async的函数）中调用：

Task {
    do {
        let imageURL = try await FirebaseStorageService.uploadImage(image: selectedImage, fileType: "profilePhotos")
        print("图片URL：\(imageURL)")
        // 更新UI需要切回主线程
        DispatchQueue.main.async {
            // self.profileImageURL = imageURL
        }
    } catch {
        print("上传失败：\(error.localizedDescription)")
    }
}

为什么原代码不行？ #

原代码的return imageURL是在putData闭包执行前就运行的——闭包是后台线程执行的，等上传完成再给imageURL赋值时，方法早就返回空字符串了。必须用异步的方式（闭包/Async/Await）等待所有异步步骤完成后，再传递结果。

内容的提问来源于stack exchange，提问作者Tyler Wasick