Swift+Firebase Firestore高效获取非好友用户的实现方案

问题本质 #

Firestore的notIn查询有硬限制：最多只能传入10个元素，超过就会触发崩溃；另外你用notIn查询数组字段的逻辑完全错误——notIn是判断字段值不在指定数组内，而非数组字段不包含某个值，所以那段代码根本达不到你要的效果。

推荐解决方案 #

方案1：数据结构优化+分段范围查询（平衡读写效率） #

这种方案不用改太多现有逻辑，能避开notIn的10元素限制：

• 给users集合的文档加一个可排序字段，比如userId本身（如果是字符串ID）或者createdAt时间戳。
• 将你的MyFriends数组按这个排序字段拆分成多个≤10元素的小批次。
• 结合范围查询（比如whereField("userId", isGreaterThan: lastFetchedId)）和notIn分批查询，每次查询只处理一个批次的好友ID，同时分页获取结果。
• 客户端合并结果时自动去重，避免重复加载同一用户。

示例代码片段：

let currentUserId = UserDefaults.standard.string(forKey: "userId")!
var friendBatches = splitArrayIntoBatches(MyFriends, batchSize: 10) // 自定义数组拆分函数
var lastFetchedUserId = ""
var allNonFriends: [User] = []

func fetchNextBatch() {
    var query = db.collection("users")
        .whereField("userId", isGreaterThan: lastFetchedUserId)
    
    if let currentBatch = friendBatches.first {
        query = query.whereField("userId", notIn: currentBatch)
        friendBatches.removeFirst()
    }
    
    query.limit(to: 20).getDocuments { [weak self] snapshot, error in
        guard let docs = snapshot?.documents, error == nil else { return }
        let nonFriends = docs.compactMap { doc -> User? in
            let userId = doc["userId"] as! String
            guard userId != currentUserId else { return nil }
            // 转换为你的User模型
            return User(id: userId, name: doc["name"] as! String)
        }
        self?.allNonFriends.append(contentsOf: nonFriends)
        if let lastDoc = docs.last {
            self?.lastFetchedUserId = lastDoc["userId"] as! String
            self?.fetchNextBatch()
        } else {
            // 所有批次查询完成，处理最终结果
        }
    }
}

// 辅助函数：拆分数组为指定大小的批次
func splitArrayIntoBatches<T>(_ array: [T], batchSize: Int) -> [[T]] {
    stride(from: 0, to: array.count, by: batchSize).map {
        Array(array[$0..<min($0+batchSize, array.count)])
    }
}

方案2：云函数预计算（最高效，适合中大型用户量） #

通过云函数提前计算并维护用户的非好友列表，客户端直接读现成数据：

• 给每个用户在Firestore里建一个userNonFriends/{userId}集合，专门存储该用户的非好友ID/引用。
• 写云函数监听好友关系变化（比如friends集合的增删，或用户文档friendIds字段的更新），每次变化时自动计算并更新对应的userNonFriends集合。
• 客户端只需查询userNonFriends/{currentUserId}就能直接拿到非好友列表，不用做任何复杂过滤。

这种方案把计算逻辑放在服务端，客户端零负担，读写成本远低于多次查询。

方案3：客户端本地过滤（适合小用户量） #

如果你的平台总用户数不多（比如几千以内），可以直接分页拉取所有用户数据，在客户端本地过滤掉好友和当前用户：

let currentUserId = UserDefaults.standard.string(forKey: "userId")!
let friendsSet = Set(MyFriends) // 用Set提高contains效率
var allNonFriends: [User] = []
var lastSnapshot: DocumentSnapshot?

func fetchUsers() {
    var query = db.collection("users").limit(to: 50)
    if let snapshot = lastSnapshot {
        query = query.start(afterDocument: snapshot)
    }
    
    query.getDocuments { [weak self] snapshot, error in
        guard let docs = snapshot?.documents, error == nil else { return }
        let filtered = docs.compactMap { doc -> User? in
            let userId = doc["userId"] as! String
            guard userId != currentUserId && !friendsSet.contains(userId) else { return nil }
            return User(id: userId, name: doc["name"] as! String)
        }
        self?.allNonFriends.append(contentsOf: filtered)
        if docs.count == 50 {
            self?.lastSnapshot = docs.last
            self?.fetchUsers()
        } else {
            // 所有用户加载完成，处理结果
        }
    }
}

关键注意点 #

• 永远不要用notIn处理超过10个元素的列表，这是Firestore的硬限制，没有例外。
• Firestore没有直接的“数组不包含”查询操作符，所有这类需求都要通过数据结构优化或间接方式实现。

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