Firestore用户分库重构后，如何通过UID获取当前用户数据？

关于Firestore Authentication UID的自定义问题 #

Firestore Authentication的UID是系统自动生成的唯一标识符，开发者无法修改其生成规则或添加自定义前缀。UID的格式由Firebase Auth服务控制，不支持自定义扩展或修改，所以通过UID前缀关联用户名首字母的思路不可行。

更优的数据库结构方案 #

结合你的用户规模（月活5000）和搜索需求，以下几个方案更实用：

方案1：单集合+前缀搜索（推荐） #

5000用户的规模完全不需要拆分集合，Firestore单集合可轻松支撑十万级数据的查询。只需在users集合中用UID作为文档ID，存储用户完整信息（包含username字段），然后通过复合索引+前缀查询实现用户搜索：

查询当前用户信息的代码

Firestore.firestore().collection("users")
    .document(uid)
    .getDocument { snapshot, error in
        guard let snapshot = snapshot, snapshot.exists else { return }
        guard let user = try? snapshot.data(as: User.self) else { return }
        completion(user)
    }

按用户名首字母搜索的代码

假设要搜索首字母为"A"的用户：

let prefix = "A"
Firestore.firestore().collection("users")
    .whereField("username", isGreaterThanOrEqualTo: prefix)
    .whereField("username", isLessThanOrEqualTo: prefix + "\u{F7FF}") // 匹配所有以prefix开头的字符串
    .getDocuments { snapshot, error in
        // 处理搜索结果
    }

注意：需要在Firestore控制台为username字段创建升序索引，确保查询性能。

方案2：主集合+分组辅助集合 #

如果担心单集合查询性能（5000用户其实完全没必要），可以保留主users集合（UID为文档ID），同时创建一个user_groups辅助集合：

• user_groups的文档ID为用户名首字母（如"A"、"B"）
• 每个文档中存储该首字母下的用户UID列表或精简信息（如username、avatar）

这样：

• 登录后直接从users集合获取当前用户完整信息
• 搜索时先从user_groups获取对应首字母的用户UID列表，再批量从users集合拉取详细信息

方案3：UID到首字母的映射集合 #

如果坚持要按首字母拆分用户子集合，可以额外创建一个uid_to_letter集合：

• 文档ID为用户UID
• 文档中仅存储firstLetter字段（用户名首字母）

登录流程变为：

1. 用UID查询uid_to_letter集合，获取用户名首字母
2. 根据首字母到对应子集合查询用户详细信息

示例代码

// 第一步：获取首字母
Firestore.firestore().collection("uid_to_letter")
    .document(uid)
    .getDocument { snapshot, error in
        guard let snapshot = snapshot, let letter = snapshot.data()?["firstLetter"] as? String else { return }
        // 第二步：查询对应子集合的用户
        Firestore.firestore().collection("users").document(letter).collection("users")
            .document(uid)
            .getDocument { userSnapshot, _ in
                guard let userSnapshot = userSnapshot else { return }
                guard let user = try? userSnapshot.data(as: User.self) else { return }
                completion(user)
            }
    }

注意：创建用户时需要同时写入users/[letter]/users/[uid]和uid_to_letter/[uid]两个文档，保证数据一致性。

对现有代码的修正 #

你当前的代码通过uid.first获取首字母是完全错误的——Firebase Auth的UID是随机字符串，和用户名首字母没有任何关联，这个逻辑会导致无法正确找到用户数据，必须废弃。

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