基于Flamelink+Firebase：查询两个Firestore集合并获取媒体URL

你现在用Flamelink作为Firebase的无头CMS，字符串字段都能正常读取，但在获取上传到Firebase Storage的媒体URL时遇到了问题。fl_content集合里的imageUpload字段是指向fl_files集合的引用数组，每个fl_files文档中的file字段存储着图片的文件名，你需要把对应的图片URL拿到，填充到Task的imageUpload属性里，最终在UITableViewCell中显示图片。

先回顾下你提供的现有代码片段：

现有Task模型代码 #

struct Task{
 var string1:String
 var string2:String
 //var imageUpload:String
 var counter:Int
 var id: String
 var dictionary: [String: Any] {
 return [
 "string1": string1,
 "string2": string2,
 //"imageUpload": imageUpload,
 "counter": counter
 ]
 }
}
extension Task{
 init?(dictionary: [String : Any], id: String) {
 guard let string1 = dictionary["string1"] as? String,
 let string2 = dictionary["string2"] as? String,
 //let imageUpload = dictionary["imageUpload"] as? String,
 let counter = dictionary["counter"] as? Int else {
 return nil
 }
 self.init(string1:string1, string2: string2, /*imageUpload: imageUpload,*/ counter: counter, id: id)
 }
}

现有VC核心代码 #

private var documents: [DocumentSnapshot] = []
public var tasks: [Task] = []
private var listener : ListenerRegistration!

fileprivate func baseQuery() -> Query {
 return Firestore.firestore().collection("fl_content").limit(to: 50)
}

fileprivate var query: Query? {
 didSet {
 if let listener = listener {
 listener.remove()
 }
 }
}

override func viewDidLoad() {
 super.viewDidLoad()
 self.query = baseQuery()
}

override func viewWillDisappear(_ animated: Bool) {
 super.viewWillDisappear(animated)
 self.listener.remove()
}

override func viewWillAppear(_ animated: Bool) {
 super.viewWillAppear(animated)
 listenerStuff()
}

func listenerStuff(){
 self.listener = query?.addSnapshotListener { (documents, error) in
 guard let snapshot = documents else {
 print("Error fetching documents results: \(error!)")
 return
 }
 let results = snapshot.documents.map { (document) -> Task in
 if let task = Task(dictionary: document.data(), id: document.documentID) {
 return task
 } else {
 fatalError("Unable to initialize type \(Task.self) with dictionary \(document.data())")
 }
 }
 self.tasks = results
 self.documents = snapshot.documents
 self.databaseTableView.reloadData()
 }
}

解决方案思路 #

Firestore本身不支持直接的联表查询，所以针对imageUpload指向fl_files引用的情况，有两种常用处理方式：

方式一：预加载所有关联的fl_files数据（适合数据量不大的场景） #

一次性获取所有fl_content文档后，批量查询对应的fl_files文档，拿到图片文件名再生成Storage URL，最后填充到Task模型中。

第一步：修改Task模型

添加imageUpload属性存储图片URL，并更新初始化方法：

struct Task{
    var string1:String
    var string2:String
    var imageUpload: String? // 存储图片的下载URL
    var counter:Int
    var id: String
    
    var dictionary: [String: Any] {
        return [
            "string1": string1,
            "string2": string2,
            "counter": counter
        ]
    }
}

extension Task{
    init?(dictionary: [String : Any], id: String, imageURL: String?) {
        guard let string1 = dictionary["string1"] as? String,
              let string2 = dictionary["string2"] as? String,
              let counter = dictionary["counter"] as? Int else {
            return nil
        }
        self.init(string1:string1, string2: string2, imageUpload: imageURL, counter: counter, id: id)
    }
}

第二步：修改listenerStuff方法，批量关联查询

func listenerStuff(){
    self.listener = query?.addSnapshotListener { [weak self] (snapshot, error) in
        guard let self = self, let snapshot = snapshot else {
            print("Error fetching documents results: \(error!)")
            return
        }
        
        // 1. 收集所有需要查询的fl_files引用
        var fileReferences: [DocumentReference] = []
        let contentDocs = snapshot.documents
        contentDocs.forEach { doc in
            // 注意：如果你的imageUpload是单个引用而不是数组，就改成 as? DocumentReference
            if let imageRefs = doc.get("imageUpload") as? [DocumentReference] {
                fileReferences.append(contentsOf: imageRefs)
            }
        }
        
        // 2. 批量查询fl_files文档，存储文件名对应的URL
        let dispatchGroup = DispatchGroup()
        var fileURLMap: [String: String] = [:] // 用fl_files文档ID作为key
        
        fileReferences.forEach { ref in
            dispatchGroup.enter()
            ref.getDocument { fileDoc, error in
                defer { dispatchGroup.leave() }
                guard let fileDoc = fileDoc, fileDoc.exists,
                      let fileData = fileDoc.data(),
                      let fileName = fileData["file"] as? String else {
                    return
                }
                // 生成Firebase Storage的下载URL（路径要和Flamelink配置一致）
                let storageRef = Storage.storage().reference().child("flamelink/media/\(fileName)")
                storageRef.downloadURL { url, error in
                    if let url = url {
                        fileURLMap[ref.documentID] = url.absoluteString
                    }
                }
            }
        }
        
        // 3. 所有查询完成后，生成Task数组并刷新表格
        dispatchGroup.notify(queue: .main) {
            let results = contentDocs.map { doc -> Task in
                var targetURL: String?
                if let imageRefs = doc.get("imageUpload") as? [DocumentReference] {
                    // 取第一个图片的URL（多图场景可以自行调整逻辑）
                    targetURL = imageRefs.first.flatMap { fileURLMap[$0.documentID] }
                } else if let imageRef = doc.get("imageUpload") as? DocumentReference {
                    targetURL = fileURLMap[imageRef.documentID]
                }
                
                guard let task = Task(dictionary: doc.data(), id: doc.documentID, imageURL: targetURL) else {
                    fatalError("Unable to initialize type \(Task.self) with dictionary \(doc.data())")
                }
                return task
            }
            
            self.tasks = results
            self.documents = contentDocs
            self.databaseTableView.reloadData()
        }
    }
}

方式二：在cellForRow中懒加载图片（适合数据量大的场景） #

不在初始化Task时就加载图片，而是等到表格cell要显示的时候，再根据引用去查询fl_files并获取图片URL，避免一次性加载过多数据。

第一步：修改Task模型，存储引用

struct Task{
    var string1:String
    var string2:String
    var imageReferences: [DocumentReference]? // 存储fl_files的引用
    var counter:Int
    var id: String
    
    var dictionary: [String: Any] {
        return [
            "string1": string1,
            "string2": string2,
            "counter": counter
        ]
    }
}

extension Task{
    init?(dictionary: [String : Any], id: String) {
        guard let string1 = dictionary["string1"] as? String,
              let string2 = dictionary["string2"] as? String,
              let counter = dictionary["counter"] as? Int else {
            return nil
        }
        self.imageReferences = dictionary["imageUpload"] as? [DocumentReference]
        self.init(string1:string1, string2: string2, counter: counter, id: id)
    }
}

第二步：修改cellForRowAt方法，懒加载图片

func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {
    let cell = tableView.dequeueReusableCell(withIdentifier: "databaseCellID", for: indexPath) as! DatabaseTableViewCell
    let item = tasks[indexPath.row]
    
    cell.label1.text = item.string1
    cell.label2.text = item.string2
    // 设置占位图
    cell.imageView?.image = UIImage(named: "placeholder")
    
    // 获取第一个图片的引用
    guard let firstImageRef = item.imageReferences?.first else {
        return cell
    }
    
    // 查询fl_files文档获取文件名
    firstImageRef.getDocument { fileDoc, error in
        guard let fileDoc = fileDoc, fileDoc.exists,
              let fileData = fileDoc.data(),
              let fileName = fileData["file"] as? String else {
            return
        }
        
        // 获取Storage下载URL并加载图片
        let storageRef = Storage.storage().reference().child("flamelink/media/\(fileName)")
        storageRef.downloadURL { url, error in
            guard let url = url else { return }
            // 建议用SDWebImage/Kingfisher等库加载图片，自带缓存和异步处理
            DispatchQueue.global().async {
                if let data = try? Data(contentsOf: url), let image = UIImage(data: data) {
                    DispatchQueue.main.async {
                        cell.imageView?.image = image
                    }
                }
            }
        }
    }
    
    return cell
}

注意事项 #

• 确保你的Firebase Storage规则允许读取这些媒体文件，否则会获取不到下载URL
• 推荐使用第三方图片加载库（比如SDWebImage、Kingfisher）来处理图片加载，它们支持异步加载、缓存和错误处理，性能更优
• 如果你的imageUpload是单个引用而不是数组，记得调整代码中的类型转换逻辑

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