Flutter中如何获取Firebase Firestore集合名称以按用户类型跳转页面

核心思路是：当用户认证状态变更后，先查询Firestore的两个用户集合，确认当前用户属于哪个集合，再根据结果跳转对应页面。

1. 编写用户集合判断函数 #

先实现一个异步函数，通过用户UID检查其存在于哪个集合：

Future<String?> getUserCollection(String uid) async {
  // 检查第一个用户集合（示例：regular_users）
  final regularUserDoc = await FirebaseFirestore.instance
      .collection('regular_users')
      .doc(uid)
      .get();
  if (regularUserDoc.exists) {
    return 'regular_users';
  }

  // 检查第二个用户集合（示例：premium_users）
  final premiumUserDoc = await FirebaseFirestore.instance
      .collection('premium_users')
      .doc(uid)
      .get();
  if (premiumUserDoc.exists) {
    return 'premium_users';
  }

  return null; // 用户不存在于任何预设集合
}

2. 在StreamBuilder中嵌套FutureBuilder实现跳转 #

修改main.dart中的StreamBuilder逻辑，当检测到已登录用户时，调用上述函数判断集合，再跳转对应页面：

StreamBuilder<User?>(
  stream: FirebaseAuth.instance.authStateChanges(),
  builder: (context, authSnapshot) {
    // 加载状态显示加载动画
    if (authSnapshot.connectionState == ConnectionState.waiting) {
      return const Center(child: CircularProgressIndicator());
    }

    // 用户已登录
    if (authSnapshot.hasData) {
      return FutureBuilder<String?>(
        future: getUserCollection(authSnapshot.data!.uid),
        builder: (context, collectionSnapshot) {
          if (collectionSnapshot.connectionState == ConnectionState.waiting) {
            return const Center(child: CircularProgressIndicator());
          }

          // 根据集合名称跳转对应页面
          switch (collectionSnapshot.data) {
            case 'regular_users':
              return const RegularUserTabsScreen();
            case 'premium_users':
              return const PremiumUserTabsScreen();
            default:
              return const UserNotFoundScreen(); // 处理用户未在任何集合的异常情况
          }
        },
      );
    }

    // 用户未登录，返回登录页面
    return const LoginScreen();
  },
)

优化建议 #

• 缓存用户类型：首次查询后将用户类型存入SharedPreferences，下次启动直接读取缓存，减少Firestore查询次数，提升启动速度。
• 添加字段标识：如果允许，可以在每个用户文档中添加user_type字段（如'regular'/'premium'），这样只需查询一次即可判断类型，比遍历两个集合更高效。
• 异常处理：在getUserCollection函数中添加try-catch块，处理网络错误或权限问题，给用户友好提示。

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