如何嵌套TaskGroup实现并行任务？Firebase并发数据获取问题

问题根源拆解 #

• 方案一错误原因：Firebase的observeSingleEvent回调是同步闭包（无async修饰），无法在其中调用await或创建TaskGroup（这类操作属于异步上下文专属），因此编译器报错。
• 方案二缺陷：外层获取图片路径后，通过串行循环调用fetchPictureDataL，完全浪费了并发能力，无法实现图片并行加载。

核心解决思路 #

采用嵌套TaskGroup架构：

1. 外层TaskGroup并行获取所有帖子
2. 每个帖子的获取任务中，先拉取帖子基础数据，再通过内层TaskGroup并行加载该帖子的所有图片
3. 所有帖子+图片任务完成后，统一更新UI展示数据
4. 全程保证线程安全，避免多线程数据竞争

代码实现 #

1. 封装Firebase异步操作 #

先将Firebase的回调式API封装为标准async函数，确保异步逻辑符合Swift Concurrency规范：

// 获取单条帖子数据
func fetchPostData(reference: String) async throws -> [String: Any] {
    let dbRef = Database.database(url: FIREBASEURL.url)
        .reference()
        .child("posts")
        .child(reference)
    
    return try await withCheckedThrowingContinuation { continuation in
        dbRef.observeSingleEvent(of: .value) { snapshot in
            guard snapshot.exists(), let postData = snapshot.value as? [String: Any] else {
                continuation.resume(throwing: NSError(domain: "PostError", code: 0, userInfo: [NSLocalizedDescriptionKey: "帖子数据无效或不存在"]))
                return
            }
            continuation.resume(returning: postData)
        }
    }
}

// 获取单张图片数据
func fetchPictureData(picRef: String) async throws -> [String: Any] {
    let dbRef = Database.database(url: FIREBASEURL.url)
        .reference()
        .child("pictures")
        .child(picRef)
    
    return try await withCheckedThrowingContinuation { continuation in
        dbRef.observeSingleEvent(of: .value) { snapshot in
            guard let pictureData = snapshot.value as? [String: Any] else {
                continuation.resume(throwing: NSError(domain: "PictureError", code: 0, userInfo: [NSLocalizedDescriptionKey: "图片数据无效"]))
                return
            }
            continuation.resume(returning: pictureData)
        }
    }
}

2. 嵌套TaskGroup并行加载所有数据 #

// 线程安全的数据管理器，避免多线程写入竞争
actor PostDataStore {
    var posts: [String: [String: Any]] = [:]
    var pictures: [String: [String: Any]] = [:]
    
    func savePost(reference: String, data: [String: Any]) {
        posts[reference] = data
    }
    
    func savePictures(_ newPictures: [String: [String: Any]]) {
        pictures.merge(newPictures) { _, new in new }
    }
}

// 加载所有帖子及对应图片的主方法
func loadAllContent() async throws -> ([String: [String: Any]], [String: [String: Any]]) {
    let dataStore = PostDataStore()
    
    try await withThrowingTaskGroup(of: Void.self) { outerGroup in
        // 外层并行处理每个帖子
        for reference in referencePostsDict.keys {
            outerGroup.addTask {
                // 1. 获取当前帖子基础数据
                let postData = try await self.fetchPostData(reference: reference)
                try await dataStore.savePost(reference: reference, data: postData)
                
                // 2. 并行加载当前帖子的所有图片
                guard let picturePaths = postData["picturePaths"] as? [String: String] else { return }
                
                try await withThrowingTaskGroup(of: (key: String, data: [String: Any]).self) { innerGroup in
                    for (key, _) in picturePaths where key != "O" {
                        innerGroup.addTask {
                            let picData = try await self.fetchPictureData(picRef: key)
                            return (key, picData)
                        }
                    }
                    
                    // 收集图片结果并存储
                    var postPictures = [String: [String: Any]]()
                    for try await result in innerGroup {
                        postPictures[result.key] = result.data
                    }
                    try await dataStore.savePictures(postPictures)
                }
            }
        }
    }
    
    // 从线程安全的存储中获取最终数据
    return (await dataStore.posts, await dataStore.pictures)
}

3. 调用并更新UI #

在UI层触发加载，完成后回到主线程更新界面：

Task {
    do {
        let (allPosts, allPictures) = try await loadAllContent()
        // 回到主线程更新UI
        DispatchQueue.main.async {
            self.threePosts = allPosts
            self.threePicturesFetched = allPictures
            // 刷新列表或其他UI组件
        }
    } catch {
        print("加载失败: \(error.localizedDescription)")
    }
}

关键说明 #

• 嵌套TaskGroup：外层保证帖子并行加载，内层保证单帖子的图片并行加载，最大化利用并发能力
• 线程安全：使用actor封装数据存储，避免多线程写入导致的数据竞争问题
• 错误处理：所有异步操作都保留了错误抛出逻辑，可根据业务需求自定义错误处理流程

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