如何用Spring Data MongoDB实现部分词匹配的预测搜索？

针对你需要的前缀/部分词匹配预测搜索需求，结合Spring Data MongoDB，以下是三种可行的实现方案：

方案1：正则表达式查询（小数据量场景） #

直接用MongoDB的正则匹配实现前缀或包含匹配，适合数据量不大的场景，无需额外索引。

实现步骤： #

1. 定义存储查询语句的实体类：

@Document(collection = "saved_queries")
public class SavedQuery {
    @Id
    private String id;
    private String searchQuery;

    // getter、setter、构造方法省略
}

2. 两种实现方式：

• Spring Data自动生成查询（前缀匹配）
  在Repository接口中定义方法即可自动生成查询逻辑：

public interface SavedQueryRepository extends MongoRepository<SavedQuery, String> {
    List<SavedQuery> findBySearchQueryStartingWith(String prefix);
}

调用时传入用户输入的"tesla"，就能返回所有以该词开头的查询语句。

• MongoTemplate自定义正则（支持任意位置匹配）
  如果需要匹配包含关键词的所有结果，用自定义正则查询：

@Service
public class SavedQueryService {
    @Autowired
    private MongoTemplate mongoTemplate;

    public List<SavedQuery> searchByPartialWord(String keyword) {
        // "i"参数表示忽略大小写，如需仅前缀匹配，将正则改为"^" + keyword
        Criteria criteria = Criteria.where("searchQuery").regex(".*" + keyword + ".*", "i");
        Query query = Query.query(criteria);
        return mongoTemplate.find(query, SavedQuery.class);
    }
}

优缺点： #

• 优点：实现简单，无需额外配置
• 缺点：大数据量下性能差，正则查询无法高效利用索引

方案2：文本索引+通配符搜索（中等数据量场景） #

利用MongoDB文本索引的后缀通配符特性，性能优于正则查询，适合中等数据规模。

实现步骤： #

1. 给字段添加文本索引：
   通过实体类注解配置：

@Document(collection = "saved_queries")
@TextIndexed(fields = @TextIndexedField(value = "searchQuery"))
public class SavedQuery {
    // 同方案1的字段定义
}

或手动在MongoDB Shell创建索引：

db.saved_queries.createIndex({ searchQuery: "text" })

2. 构造文本搜索查询：

@Service
public class SavedQueryService {
    @Autowired
    private MongoTemplate mongoTemplate;

    public List<SavedQuery> searchWithTextIndex(String keyword) {
        // matchingPrefix会自动添加后缀通配符，匹配所有以keyword开头的词
        TextCriteria textCriteria = TextCriteria.forDefaultLanguage().matchingPrefix(keyword);
        Query query = TextQuery.queryText(textCriteria);
        return mongoTemplate.find(query, SavedQuery.class);
    }
}

优缺点： #

• 优点：利用文本索引提升性能，支持多语言分词
• 缺点：仅支持后缀通配符（无法匹配中间包含关键词的结果）

方案3：自动补全索引（大数据量/高性能场景） #

MongoDB 3.6+提供的自动补全索引，专为预测搜索/自动补全场景优化，性能最优。

实现步骤： #

1. 创建自动补全索引：
   通过实体类注解配置：

@Document(collection = "saved_queries")
@CompoundIndex(name = "search_query_autocomplete", def = "{'searchQuery': 'autocomplete'}")
public class SavedQuery {
    // 同方案1的字段定义
}

或手动在MongoDB Shell创建：

db.saved_queries.createIndex({ searchQuery: "autocomplete" }, { name: "search_query_autocomplete" })

2. 使用自动补全查询：

@Service
public class SavedQueryService {
    @Autowired
    private MongoTemplate mongoTemplate;

    public List<SavedQuery> autocompleteSearch(String keyword) {
        Criteria criteria = Criteria.where("searchQuery").autocomplete(keyword);
        Query query = Query.query(criteria);
        return mongoTemplate.find(query, SavedQuery.class);
    }
}

优缺点： #

• 优点：专为自动补全设计，性能最优，大数据量下表现优异
• 缺点：仅支持前缀匹配，不支持中间或后缀的部分匹配

方案选择建议 #

• 小数据集（几千条以内）：用正则表达式查询，快速实现
• 中等数据集（几万到几十万条）：文本索引+通配符搜索，兼顾性能和灵活性
• 大数据集（百万条以上）或对性能要求高：自动补全索引，最优选择

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