Spring Boot 整合 ElasticSearch 这么简单?

ElasticSearch 作为一款常见的搜索引擎,在项目中随时会用到。为方便在项目中访问 ElasticSearch,Spring 官方封装了对 ElasticSearch 的支持。通过 Spring Data 的 ElasticSearch 模块,我们可以方便地使用 ElasticSearch。

本文主要涉及的知识点有:

  • ElasticSearch 的使用场景和技术。
  • spring-data-elasticsearch 中 Operations 和 Repository 相关技术的使用。
  • 如何在 Spring Boot 项目中引入 spring-data-elasticsearch。

ElasticSearch的使用场景和相关技术

搜索功能不仅在互联网项目中需要,在企业级项目中也需要。在通用型搜索引擎出现之前,通常实现搜索功能的方式是关系型数据库的模糊查询,但是使用模糊查询具有效率低、响应速度慢、不支持匹配度排序等缺陷。因此,在项目中引入搜索引擎就成了实现搜索功能的不二之选。

常用的搜索引擎除 ElasticSearch 之外还有 Solr,它和 ElasticSearch 都是基于 Lucene 开发出来的。Lucene 是 Apache 的开源搜索项目,这个项目的产出是一个搜索库,被称为 Lucene Core。Lucene Core 是 Java 实现的,提供了强大的索引和搜索功能,以及拼写检查,单击突出显示和高级分析/标记功能。

Apache Solr 是 Apache 的一个独立的顶级项目,其内置了完整的 Lucene 包。自从 Lucene 和 Solr 整合之后,Solr 和 Lucene 发布的版本都是一致的。

由于 Lucene Core 只是 Java 库,不能独立使用,因此平时在企业中使用最多的还是 ElasticSearch 或 Solr。ElasticSearch 和 Solr都能实现搜索,但是也不完全相同。Solr 有庞大的用户群,而且比较成熟,但是建立索引时会影响搜索效率,不适合用作实时搜索。而 ElasticSearch 支持分布式、实时分发,支持建立索引和搜索同时进行。也正是由于 ElasticSearch 的这些特点,使得 ElasticSearch 的使用变得越来越流行。

spring-data-elasticsearch 支持的ElasticSearch Client

要在项目中使用 ElasticSearch,首先要连接到 ElasticSearch。实现连接到 ElasticSearch 的模块被称作 ElasticSearch Client。

ElasticSearch 的 Client

ElasticSearch 官方提供了 3 个 Client,具体如下:

  • org.elasticsearch.client.transport.TransportClient
  • org.elasticsearch.client.RestClient
  • org.elasticsearch.client.RestHighLevelClient

TransportClient 位于 Elasticsearch 包下,是 Elasticsearch 官方早期支持的 ElasticSearch Client,但是在 ElasticSearch 7.x 版本中已经标注为 Deprecated,并且将在 8.0 版本中移除,所以建议不使用 TransportClient 作为 ElasticSearch Client。

RestHighLevelClient 是 TransportClient 的直接替代者,也是 ElasticSearch 官方推荐和默认的 Client(松哥注:该客户端在最新版本中已经废弃)。

除了 ElasticSearch 官方提供的 Client,spring-data-elasticsearch 还支持响应式的客户端 ReactiveElasticsearchClient。ReactiveElasticsearchClient 是基于 WebClient 技术实现的 ElasticSearch Client,依赖于 Spring 的响应式栈。响应式栈在本书中不会涉及。

创建 RestHighLevelClient

spring-data-elasticsearch 提供了接口AbstractElasticsearchConfiguration,使用该接口可以非常方便地在容器中引入 RestHighLevelClient。

AbstractElasticsearchConfiguration 接口位于org.springframework.data.elasticsearch.config 包下,可以按照如下方式来使用:

@Configuration
public class ESConfigutration extends AbstractElasticsearchConfiguration {
 
    @Override
    public RestHighLevelClient elasticsearchClient() {
        final ClientConfiguration clientConfiguration = ClientConfiguration.builder()
                .connectedTo("192.168.110.128:9200")
                .build();
        return RestClients.create(clientConfiguration).rest();
    }
}

如上面的代码,自定义一个配置类,继承 AbstractElasticsearchConfiguration 接口,并实现接口中定义的方法 RestHighLevelClient elasticsearchClient()。

上面的 ClientConfiguration 用来配置 ElasticSearch 客户端的属性,比如可以配置代理、连接超时时长以及 socket 超时时长等,上面的代码示例中只配置了 ElasticSearch 服务的地址和端口号。

使用 operations 相关 API 操作 ElasticSearch

spring-data-elasticsearch 中定义了 4 个命名以 Operations 结尾的接口,用来操作 ElasticSearch 的不同对象,接口分别是:

  • IndexOperations
  • DocumentOperations
  • SearchOperations
  • ElasticsearchOperations。

4 个 Operations 接口

IndexOperations 和 DocumentOperations 接口从命名上可以看出,分别定义的是 Index 级别的接口和 Document 的接口,比如创建和删除索引(Index)以及创建和根据 id 更新、查询 entity。spring-data-elasticsearch 提供了 IndexOperations 的默认实现,如图 6.1 所示,有 DetaultIndexOperations 和 DefaultTransportIndexOperations 两个实现类。这两个实现类使用的客户端不同,前者使用的是 RestHighLevelClient,后者使用的是 TransportClient,当然更推荐使用前者。

在 spring-data-elasticsearch 的 4.2.3 版本中,IndexOperations 中定义了操作 Index 的方法,对应 ElasticSearch 官方文档的 Index APIs 部分,用于管理单个索引、索引设置、别名、映射和索引模板等。注意,表 6.1 中不包含已经被标记为过时的方法,以及不包含使用 ElasticSearch Legacy API 实现的方法。

SearchOperation、DocumentOperations 和 ElasticsearchOperations 的继承关系及其实现类的继承关系如图 6.2 所示。

从图 6.2 中可以看出,ElasticsearchOperations 继承了 SearchOperation 和 DocumentOperations,并且 spring-data-elasticsearch 为 ElasticsearchOperations 提供了两个实现类,即 ElasticsearchRestTemplate 和 ElasticsearchTemplate,同上面的实现类一致,也是基于两种客户端的实现。

DocumentOperations 接口中定义了 Document 级别的方法,对应 ElasticSearch 官方文档的 Document APIs 部分,包括对 Document 的增删改查等操作,具体方法列举在表 6.2 中。

接口 SearchOperations 中定义了搜索和聚合索引的相关操作,对应 ElasticSearch 官方文档的 Search APIs 部分,具体方法列举在表 6.3 中。

最后,接口 ElasticsearchOperations 继承了 DocumentOperations 和 SearchOperations,所以在接口 DocumentOperations 和 SearchOperations 中定义的方法都可以通过接口 ElasticsearchOperations 来使用。除了继承的方法外,接口 ElasticsearchOperations 中还定义了一些通用的辅助性的方法,具体列举在表 6.4 中。

搜索结果类型

ElasticSearch 搜索 API 在返回搜索数据的同时也会返回搜索产生的额外信息,比如匹配到的总数量、排序字段值、高亮显示等,这些伴随着搜索的额外信息就被放置在 spring-data-elasticsearch 提供的搜索结果包装类中。本节将介绍 spring-data-elasticsearch 搜索结果的包装类。

由于部分类使用了和 ElasticSearch官方提供的相同的类名,因此先对这些类所属的包说明一下,以下所提及的类如果没有特殊说明,默认都是包 org.springframework.data. elasticsearch.core 下的,而不是 ElasticSearch 官方的包 org.elasticsearch.search 下的同名类。

SearchHit

搜索接口返回的数据实体都会使用 SearchHit类作为包装,用来放置数据实体相关的搜索信息,具体字段(get方法)信息见表6.5。

SearchHits

上面的 SearchHit是对单条数据的封装,而接口 SearchHits是对整体搜索结果的封装,其内部定义了获取 SearchHit列表的方法,以及获取一次搜索的总体数据的方法等,具体方法信息见表 6.6。

查询条件的封装

在接口 SearchOperations 中定义的方法,除了最后两个查询建议的方法外,其他方法中都使用了类型为 org.springframework.data.elasticsearch.core.query.Query 的参数。该类型为接口类型,spring-data-elasticsearch 提供了 3 个实现类,分别是:

  • CriteriaQuery
  • StringQuery
  • NativeSearchQuery。

下面将依次介绍这3个实现类。

  1. CriteriaQuery

CriteriaQuery 允许我们通过 API 调用的方式来定义查询条件,好处就是不需要用户理解 ElasticSearch 原生的查询语法。

CriteriaQuery 有两个构造器,也是创建 CriteriaQuery 的两种方式。这两个构造器都需要类为 org.springframework.data.elasticsearch.core.query.Criteria 的封装查询条件,所以可以理解为  CriteriaQuery只是 Criteria 的包装类,我们创建和封装查询条件主要通过 Criteria 来实现。

Criteria 方法的命名仿照了 SQL 关键字,比如创建查询条件可以使用其静态方法 where()。另外,对于多个条件组合,可以使用 and() 和 or()。

下面通过示例说明 CriteriaQuery 类的使用方式。

【示例 6.1】查询出版时间为给定年份的图书

假定 publishYear 字段为图书出版年份,那么查询出版年份为 2021 年的图书的查询条件封装的代码如下:

Criteria criteria = Criteria.where("publishYear").is(2021);
Query query = new CriteriaQuery(criteria);

【示例6.2】 查询出版时间在 2015~2019 年,并且类别为科学技术的图书。使用静态方法 and() 创建第二个条件并与第一个条件关联起来,具体代码如下:

Criteria criteria = Criteria.where("publishYear").between(2015,2019). and("category").is("科学技术");
CriteriaQuery criteriaQuery = new CriteriaQuery(criteria);

【示例6.3】查询出版时间在 2015~2019 年,并且类别为科学技术或历史人文的图书。

这里用到了或和与的组合,使用 subCriteria 来实现,具体代码如下:

Criteria criteria = Criteria.where("publishYear").between(2015, 2019).subCriteria(Criteria.where("category").is("科学技术").or("category").is("历史人文"));
CriteriaQuery criteriaQuery = new CriteriaQuery(criteria);
  1. StringQuery

StringQuery 以 ElasticSearch 可以理解 JSON 格式封装查询条件,因此比较适合熟悉 ElasticSearch 查询语法的用户。

这里的 JSON 格式可以对应 ElasticSearch 的 Query DSL(Domain Specific Language)语法,具体可以参考 ElasticSearch 官方文档的 Query DSL 部分。

【示例6.4】  使用 StringQuery 查询出版时间在 2015~2019 年,并且类别为历史人文的图书。使用到了 Boolean 查询来组合两个查询条件,具体代码如下:

Query query = new StringQuery("{\n" +
        "    \"bool\": {\n" +
        "      \"must\": [\n" +
        "        {\n" +
        "          \"match\": {\n" +
        "            \"category\": \"历史人文\"\n" +
        "          }\n" +
        "        },\n" +
        "        {\n" +
        "          \"range\": {\n" +
        "            \"publishYear\": {\n" +
        "              \"gte\": 2015,\n" +
        "              \"lte\": 2019\n" +
        "            }\n" +
        "          }\n" +
        "        }\n" +
        "      ]\n" +
        "    }\n" +
        "  }");
SearchHits<Book> searchHits = operations.search(query, Book.class);

从上面的例子可以看出,使用 StringQuery 虽然创建对象简单,但是可读性差,并且需要熟悉 ElasticSearch 的 DSL 语法。

  1. NativeSearchQuery

NativeSearchQuery 使用实现比较复杂的查询,比如聚合操作等。由于其可以和 ElasticSearch 官方 API 结合使用,因此命名为 Native。

虽然从功能上讲 NativeSearchQuery 比 CriteriaQuery 强大,但是由于其使用既需要熟悉 ElasticSearch 官方 API,又需要学习 NativeSearchQuery 的 API,学习成本相比前两种要更高一些,并且调试起来也不比 DSL 容易,所以也没有比较明显的优势。

在工作中,简单的查询使用 CriteriaQuery,复杂的查询先在 Kibana 中使用 DSL 调试好查询语句,然后直接复制到代码中创建 StringQuery 来构建查询,这样或许效率更高一些。


以上内容节选自 《Spring Boot 从零开始学》,作者郭浩然。

送书啦送书啦~

小伙伴们留言说说你为什么需要这本书,松哥会从留言的小伙伴中选出来 10 位幸运小伙伴,《Spring Boot 从零开始学》包邮到家!

相关文章

这篇文章主要介绍了spring的事务传播属性REQUIRED_NESTED的原...
今天小编给大家分享的是一文解析spring中事务的传播机制,相...
这篇文章主要介绍了SpringCloudAlibaba和SpringCloud有什么区...
本篇文章和大家了解一下SpringCloud整合XXL-Job的几个步骤。...
本篇文章和大家了解一下Spring延迟初始化会遇到什么问题。有...
这篇文章主要介绍了怎么使用Spring提供的不同缓存注解实现缓...