使用Java客户端在Firestore中执行OR查询及多Query合并咨询

原代码故障原因 #

• Tasks.whenAllSuccess() 返回的列表中每个元素为单个查询对应的 QuerySnapshot 对象，原代码未对该对象做二次解析直接使用，无法拿到实际的文档数据
• 未处理结果去重：若某份文档同时满足多个查询条件，会在结果中重复出现

完整可运行实现（Android 端 Firebase Java SDK） #

import com.google.firebase.firestore.DocumentSnapshot;
import com.google.firebase.firestore.FirebaseFirestore;
import com.google.firebase.firestore.Query;
import com.google.firebase.firestore.QuerySnapshot;
import com.google.android.gms.tasks.Tasks;
import com.google.android.gms.tasks.Task;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

// 初始化Firestore实例
FirebaseFirestore rootRef = FirebaseFirestore.getInstance();
// 定义OR逻辑对应的多个查询条件
Query firstQuery = rootRef.collection("目标集合名").whereEqualTo("字段A", "值A");
Query secondQuery = rootRef.collection("目标集合名").whereEqualTo("字段B", "值B");

Task<QuerySnapshot> firstTask = firstQuery.get();
Task<QuerySnapshot> secondTask = secondQuery.get();

Tasks.whenAllSuccess(firstTask, secondTask).addOnSuccessListener(results -> {
    Set<String> docIdCache = new HashSet<>();
    List<DocumentSnapshot> orQueryResult = new ArrayList<>();

    // 遍历所有查询的返回结果
    for (Object res : results) {
        QuerySnapshot snapshot = (QuerySnapshot) res;
        for (DocumentSnapshot doc : snapshot.getDocuments()) {
            // 按文档唯一ID去重
            if (!docIdCache.contains(doc.getId())) {
                docIdCache.add(doc.getId());
                orQueryResult.add(doc);
            }
        }
    }

    // 此处直接使用合并后的OR查询结果orQueryResult
}).addOnFailureListener(e -> {
    // 处理查询异常逻辑
});

核心注意事项 #

• 若使用服务端 Java Admin SDK，将 Task 替换为 ApiFuture，使用 ApiFutures.allAsList() 合并任务，解析去重逻辑完全一致
• 存在N个OR查询条件时，只需将对应查询的Task/ApiFuture全部传入合并方法即可，无需修改后续解析逻辑
• 如需自定义结果排序，可在合并完成后对orQueryResult按指定字段做排序处理

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