如何在Java+MongoDB中按搜索词位置排序查询结果

嘿，这个需求我之前做类似功能的时候碰到过，直接用普通的find查询加sort搞不定，得靠MongoDB的聚合框架来实现，结合Java驱动的API就能完美达到你要的效果。下面给你拆解具体步骤和代码：

核心思路 #

我们需要通过聚合操作给每个匹配的文档计算两个关键排序依据：

• 标记文档的name是否以搜索词开头（优先排前面）
• 计算搜索词在name字符串中的起始索引位置（索引越小越靠前）
  然后按照这两个字段组合排序，最后返回原文档内容。

Java代码实现（以Spring Data MongoDB的MongoTemplate为例） #

假设你的搜索词是kuma，我们可以这样写：

import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.aggregation.Aggregation;
import org.springframework.data.mongodb.core.aggregation.AggregationOperation;
import org.springframework.data.mongodb.core.aggregation.ProjectionOperation;
import org.springframework.data.mongodb.core.query.Criteria;

import java.util.List;

// 注入MongoTemplate
@Autowired
private MongoTemplate mongoTemplate;

public List<User> searchUsersByKeyword(String keyword) {
    // 1. 匹配包含搜索词的文档
    AggregationOperation match = Aggregation.match(Criteria.where("name").regex(keyword));

    // 2. 添加辅助排序字段
    AggregationOperation addFields = Aggregation.addFields()
            // 判断name是否以搜索词开头，是则为true，否则false
            .addFieldWithValue("isStartWith", Aggregation.expression(
                    "{ $eq: [ { $indexOfBytes: ['$name', ?0] }, 0 ] }", keyword))
            // 计算搜索词在name中的起始位置
            .addFieldWithValue("matchPosition", Aggregation.expression(
                    "{ $indexOfBytes: ['$name', ?0] }", keyword))
            .build();

    // 3. 按辅助字段排序：先按isStartWith降序（开头的排前面），再按matchPosition升序（位置越靠前越前）
    AggregationOperation sort = Aggregation.sort(
            org.springframework.data.domain.Sort.Direction.DESC, "isStartWith",
            org.springframework.data.domain.Sort.Direction.ASC, "matchPosition");

    // 4. 移除辅助字段，只返回原文档字段
    ProjectionOperation project = Aggregation.project()
            .andExclude("isStartWith", "matchPosition");

    // 组装聚合管道
    Aggregation aggregation = Aggregation.newAggregation(match, addFields, sort, project);

    // 执行聚合查询，返回User对象列表
    return mongoTemplate.aggregate(aggregation, "user", User.class).getMappedResults();
}

代码解释 #

• $indexOfBytes：MongoDB的聚合运算符，用来返回搜索词在字符串中的起始索引位置（我们已经通过match过滤了不包含搜索词的文档，所以这里结果肯定>=0）
• isStartWith字段：通过判断索引是否为0，标记是否以搜索词开头，排序时降序让true的结果排在最前面
• 最后用project移除辅助字段，避免返回多余数据

原生MongoDB聚合语句参考 #

对应的原生聚合查询语句是这样的，你可以先在MongoDB Compass里测试验证效果：

db.user.aggregate([
    { $match: { name: { $regex: "kuma" } } },
    {
        $addFields: {
            isStartWith: { $eq: [ { $indexOfBytes: ["$name", "kuma"] }, 0 ] },
            matchPosition: { $indexOfBytes: ["$name", "kuma"] }
        }
    },
    { $sort: { isStartWith: -1, matchPosition: 1 } },
    { $project: { isStartWith: 0, matchPosition: 0 } }
])

这样执行后，返回的结果就会和你期望的顺序一致：kumar → kumarasan → rajkumar → arunkumar → rajaskumar...

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