在MongoDB中使用Criteria如何比较同一文档内的两个字段
Spring Data MongoDB Criteria 同文档两字段比较实现方案
你原来的写法本质是想调用MongoDB的$where运算符实现字段比较,这种方式性能较差且存在注入风险,优先推荐使用MongoDB 3.6及以上版本支持的$expr运算符实现,性能更高。
方案1:推荐使用$expr实现
将你标注的待修改行替换为以下代码即可:
// 等值比较campoC和campoD两个字段 criteria.add(Criteria.where("$expr").is(new Document("$eq", Arrays.asList("$campoC", "$campoD"))));
如果使用较高版本的Spring Data MongoDB,也可以用更贴合面向对象的写法:
import org.springframework.data.mongodb.core.query.Criteria; import org.springframework.data.mongodb.core.aggregation.ComparisonOperators; // 等值比较campoC和campoD两个字段 criteria.add(Criteria.expr(ComparisonOperators.Eq.valueOf("campoC").equalToField("campoD")));
方案2:兼容低版本MongoDB的$where实现
如果你的MongoDB版本低于3.6,可以用以下写法实现,但是不推荐生产环境使用:
criteria.add(Criteria.where("$where").is("this.campoC == this.campoD"));
完整修改后代码示例
import org.bson.Document; import org.springframework.data.mongodb.core.query.Criteria; import org.springframework.data.mongodb.core.query.Query; import java.util.ArrayList; import java.util.Arrays; import java.util.List; public Query getQuery() { Query query = new Query(); List<Criteria> criteria = new ArrayList<>(); criteria.add(Criteria.where("campoA").is(123)); criteria.add(Criteria.where("campoB").is("N")); // 同文档两字段比较逻辑 criteria.add(Criteria.where("$expr").is(new Document("$eq", Arrays.asList("$campoC", "$campoD")))); query.addCriteria(new Criteria().andOperator(criteria.toArray(new Criteria[0]))).limit(1); return query; }
注意事项
- 如果需要做大于、小于等其他比较逻辑,只需要把
$eq替换为对应的运算符即可,比如大于用$gt、小于用$lt - 使用
$expr时字段名前必须加$符号,否则MongoDB会将其识别为普通字符串而非字段引用
内容的提问来源于stack exchange,提问作者Sandro
相关产品推荐
相关产品推荐

