Flutter Firebase Firestore：实现OR条件查询并确保获取3条随机文档

问题描述 #

我使用Firebase Firestore存储带有自动生成ID的文档，当前通过生成随机ID，执行where('id', isGreaterThanOrEqualTo: _randomIndex)查询来获取3条随机文档。但由于数据库数据量尚小，有时循环3次后仅能拿到0、1或2条数据，无法满足需求。想请教如何优化代码以确保总能获取3条文档，是否可以将查询修改为OR条件（id >= _randomIndex 或 id <= _randomIndex）？

当前代码如下：

String getRandomGeneratedId() {
  const int AUTO_ID_LENGTH = 20;
  const String AUTO_ID_ALPHABET = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789';
  
  const int maxRandom = AUTO_ID_ALPHABET.length;
  final Random randomGen = Random();
  
  String id = '';
  for (int i = 0; i < AUTO_ID_LENGTH; i++) {
    id = id + AUTO_ID_ALPHABET[randomGen.nextInt(maxRandom)];
     print('RandomID is $id');
  }
  return id;
}

Future<List> getData() async {
  List<Recipe> dataList = [];
  
  CollectionReference myRef = FirebaseFirestore.instance.collection('recipes');
  
  // Retrieves 3 random data in a loop
  for (int i = 0; i < 3; i++) {
    String _randomIndex = getRandomGeneratedId();
    print('RandomIndex is $_randomIndex');
    QuerySnapshot querySnapshot = await myRef
      .where('id', isGreaterThanOrEqualTo: _randomIndex)
      .orderBy('id', descending: false)
      .limit(1)
      .get();
      
    print('QUERYSNAP is $querySnapshot');
    dataList.addAll(querySnapshot.docs.map((d)=> Recipe.fromJson(d.data() as Map<String, dynamic>)));
  }
}

解决方案 #

关于OR条件的可行性 #

Firestore不支持这种跨范围的OR查询（id >= X 或 id <= X），这类查询无法利用索引高效检索，本质等同于扫描全集合，直接用这种逻辑走不通，得换更合理的方案。

方案1：全ID随机筛选（适合小数据量场景） #

先获取集合所有文档的ID，从中随机挑3个，再批量获取对应文档。逻辑简单，能确保拿到3条不重复的文档（只要集合文档数≥3）。

修改后的getData函数示例：

Future<List<Recipe>> getData() async {
  List<Recipe> dataList = [];
  CollectionReference myRef = FirebaseFirestore.instance.collection('recipes');
  
  // 获取所有文档
  QuerySnapshot allDocs = await myRef.get();
  if (allDocs.docs.length < 3) {
    // 文档不足3条时返回所有现有数据
    return allDocs.docs.map((d)=> Recipe.fromJson(d.data() as Map<String, dynamic>)).toList();
  }
  
  // 提取ID并打乱顺序
  List<String> allIds = allDocs.docs.map((d) => d.id).toList();
  allIds.shuffle();
  
  // 批量获取前3个随机ID对应的文档
  List<Future<DocumentSnapshot>> futures = allIds.take(3).map((id) => myRef.doc(id).get()).toList();
  List<DocumentSnapshot> selectedDocs = await Future.wait(futures);
  
  dataList = selectedDocs.map((d) => Recipe.fromJson(d.data() as Map<String, dynamic>)).toList();
  return dataList;
}

方案2：改进随机ID查询，处理边界情况 #

保留随机ID查询思路，但在查询无结果时切换逻辑，同时加入去重判断，确保凑够3条不重复文档。

修改后的getData函数示例：

Future<List<Recipe>> getData() async {
  List<Recipe> dataList = [];
  Set<String> collectedIds = {}; // 用于去重
  CollectionReference myRef = FirebaseFirestore.instance.collection('recipes');
  
  while (dataList.length < 3) {
    String randomId = getRandomGeneratedId();
    QuerySnapshot querySnapshot = await myRef
        .where('id', isGreaterThanOrEqualTo: randomId)
        .orderBy('id')
        .limit(1)
        .get();
    
    // 无结果时，切换查询小于等于randomId的最后一条文档
    if (querySnapshot.docs.isEmpty) {
      querySnapshot = await myRef
          .where('id', isLessThanOrEqualTo: randomId)
          .orderBy('id', descending: true)
          .limit(1)
          .get();
    }
    
    // 去重后添加数据
    if (querySnapshot.docs.isNotEmpty) {
      DocumentSnapshot doc = querySnapshot.docs.first;
      String docId = doc.id;
      if (!collectedIds.contains(docId)) {
        collectedIds.add(docId);
        dataList.add(Recipe.fromJson(doc.data() as Map<String, dynamic>));
      }
    }
  }
  
  return dataList;
}

方案3：添加随机数字字段（适合大数据量场景） #

给每个文档新增randomValue字段（值为0-1之间的随机浮点数），利用该字段做范围查询，效率更高。

1. 新增文档时添加字段：

// 创建新文档时
await myRef.add({
  // 原有字段
  'randomValue': Random().nextDouble(),
});

2. 查询逻辑：

Future<List<Recipe>> getData() async {
  List<Recipe> dataList = [];
  CollectionReference myRef = FirebaseFirestore.instance.collection('recipes');
  
  double randomVal = Random().nextDouble();
  QuerySnapshot querySnapshot = await myRef
      .where('randomValue', isGreaterThanOrEqualTo: randomVal)
      .orderBy('randomValue')
      .limit(3)
      .get();
  
  // 结果不足3条时，补充查询小于randomVal的部分
  if (querySnapshot.docs.length < 3) {
    int need = 3 - querySnapshot.docs.length;
    QuerySnapshot supplementQuery = await myRef
        .where('randomValue', isLessThan: randomVal)
        .orderBy('randomValue')
        .limit(need)
        .get();
    querySnapshot.docs.addAll(supplementQuery.docs);
  }
  
  dataList = querySnapshot.docs.map((d) => Recipe.fromJson(d.data() as Map<String, dynamic>)).toList();
  return dataList;
}

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