Android Studio(Java)查询Firestore匹配推荐码获取对应文档ID

前置准备 #

• 确保项目已完成Firebase Firestore SDK接入，google-services.json配置文件已放置到对应模块目录
• 调整Firestore安全规则，开放客户端对users集合下myRefer字段的查询权限，避免权限拦截
• 提前在业务逻辑中保证myRefer字段存储的推荐码全局唯一，避免匹配到多个用户

具体实现步骤 #

1. 初始化实例与控件 #

首先在注册页面对应的Activity/Fragment中初始化Firestore实例和页面控件：

// 初始化Firestore数据库实例
private FirebaseFirestore db;
private EditText etReferInput;
private TextView tvReferOwner;
private Button btnVerifyRefer;

@Override
protected void onCreate(Bundle savedInstanceState) {
    super.onCreate(savedInstanceState);
    setContentView(R.layout.activity_register);
    db = FirebaseFirestore.getInstance();
    // 绑定布局控件，替换为你实际布局中的ID
    etReferInput = findViewById(R.id.et_input_refer);
    tvReferOwner = findViewById(R.id.tv_refer_owner_id);
    btnVerifyRefer = findViewById(R.id.btn_verify_refer);
}

2. 实现推荐码校验逻辑 #

给校验按钮绑定点击事件，查询时直接通过字段匹配检索目标文档，不需要遍历全表：

btnVerifyRefer.setOnClickListener(v -> {
    String inputCode = etReferInput.getText().toString().trim();
    // 先做输入合法性校验
    if (inputCode.isEmpty()) {
        etReferInput.setError("推荐码不能为空");
        return;
    }

    // 按myRefer字段匹配查询
    db.collection("users")
            .whereEqualTo("myRefer", inputCode)
            .limit(1) // 推荐码唯一，仅取1条结果降低查询开销
            .get()
            .addOnCompleteListener(task -> {
                if (task.isSuccessful()) {
                    QuerySnapshot result = task.getResult();
                    if (result != null && !result.isEmpty()) {
                        // 匹配到有效推荐码，取第一个文档
                        DocumentSnapshot ownerDoc = result.getDocuments().get(0);
                        // 获取文档ID（即推荐人UID）
                        String referOwnerUid = ownerDoc.getId();
                        // 将ID展示到TextView
                        tvReferOwner.setText(String.format("匹配到推荐人ID：%s", referOwnerUid));
                        // 此处可追加后续逻辑：比如标记校验通过、允许提交注册申请、暂存推荐人UID用于注册时写入
                    } else {
                        // 无匹配结果，推荐码无效
                        tvReferOwner.setText("");
                        etReferInput.setError("输入的推荐码不存在，请核对后重试");
                    }
                } else {
                    // 查询异常处理：网络错误、权限错误等
                    Log.e("ReferCheck", "推荐码查询失败", task.getException());
                    Toast.makeText(RegisterActivity.this, "校验失败，请检查网络连接", Toast.LENGTH_SHORT).show();
                }
            });
});

优化建议 #

• 单字段等值查询Firestore默认自带索引，无需手动创建索引即可正常使用
• 生产环境建议将用户数据写入逻辑放到云函数/服务端实现，不要开放客户端对users集合的写权限，避免数据篡改
• 若需要在注册成功后给推荐人发奖励，务必在服务端再次校验推荐码有效性，不要完全信任客户端返回的结果
• 可以在查询时加加载弹窗，避免用户重复点击按钮

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