iOS Swift中Firebase Firestore查询分页实现方案咨询

原代码问题分析 #

第一段代码问题 #

• 误用addSnapshotListener实时监听接口：分页场景下按需加载更适合用单次查询getDocuments()，实时监听会在数据更新时自动触发，不符合分页逻辑。
• 仅手动执行了一次后续查询，没有实现按需重复触发的逻辑，也未维护上一页的最后文档引用。

第二段代码问题 #

• 同步循环与异步查询冲突：repeat-while是同步执行逻辑，但Firestore查询是异步操作，循环会在查询返回结果前持续执行，直接导致无限循环。
• 变量作用域错误：循环内部的lastSnapshot是局部变量，无法更新外部的lastSnapshot，导致while(lastSnapshot.exists)条件永远为true，无法终止循环。
• 每次查询都清空postList，会丢失之前加载的数据，不符合分页追加数据的逻辑。

正确分页实现思路 #

分页核心是按需触发（比如用户下拉到底部、点击加载更多），需要维护一个全局变量记录上一页的最后文档，作为下一页查询的起点，每次加载下一页时追加数据而非清空列表。

实现代码 #

首先定义类级别的变量：

var lastDocument: DocumentSnapshot? = nil // 记录上一页最后一个文档
var postList: [Post] = [] // 存储所有加载的帖子
let pageSize = 5 // 每页加载数量

加载第一页

func loadFirstPage(completion: @escaping (Error?) -> Void) {
    let db = Firestore.firestore()
    db.collection("posts")
        .order(by: "date", descending: true)
        .limit(to: pageSize)
        .getDocuments { snapshot, error in
            if let error = error {
                completion(error)
                return
            }
            guard let snapshot = snapshot else {
                completion(NSError(domain: "Firebase", code: -1, userInfo: [NSLocalizedDescriptionKey: "无数据返回"]))
                return
            }
            
            // 清空列表，加载第一页数据
            self.postList.removeAll()
            self.processDocuments(snapshot.documents)
            
            // 更新上一页最后文档
            self.lastDocument = snapshot.documents.last
            
            completion(nil)
        }
}

加载下一页

func loadNextPage(completion: @escaping (Error?, Bool) -> Void) {
    guard let lastDoc = lastDocument else {
        // 没有上一页的最后文档，说明已加载完所有数据
        completion(nil, false)
        return
    }
    
    let db = Firestore.firestore()
    db.collection("posts")
        .order(by: "date", descending: true)
        .start(afterDocument: lastDoc)
        .limit(to: pageSize)
        .getDocuments { snapshot, error in
            if let error = error {
                completion(error, false)
                return
            }
            guard let snapshot = snapshot else {
                completion(NSError(domain: "Firebase", code: -1, userInfo: [NSLocalizedDescriptionKey: "无数据返回"]), false)
                return
            }
            
            if snapshot.documents.isEmpty {
                // 当前页无数据，说明已到最后一页
                self.lastDocument = nil
                completion(nil, false)
                return
            }
            
            // 追加新数据到列表
            self.processDocuments(snapshot.documents)
            
            // 更新上一页最后文档
            self.lastDocument = snapshot.documents.last
            
            completion(nil, true)
        }
}

文档解析辅助方法

private func processDocuments(_ documents: [QueryDocumentSnapshot]) {
    for document in documents {
        // 替换成你的Post解析逻辑，示例：
        if let postData = document.data() as? [String: Any],
           let post = Post(data: postData) {
            self.postList.append(post)
        }
    }
}

使用方式 #

1. 初始化加载：页面启动时调用loadFirstPage，加载第一页数据，完成后刷新UI。
2. 加载更多：当用户下拉到底部（或点击“加载更多”按钮）时，调用loadNextPage。回调中的Bool参数表示是否还有更多数据：
   • 返回true：还有数据，可继续触发加载
   • 返回false：已加载完所有数据，隐藏加载更多控件

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