首页 > 编程 > Java > 正文

基于Springboot2.0构建ES的多客户端

2019-11-26 09:03:47
字体:
来源:转载
供稿:网友

有时候我们操作es的时候会有一些特殊的需求,例如需要操作的index使用了不同的es服务器、用户名、密码、参数等,这个时候我们需要使用不同的es的客户端进行操作,但是我们又不希望拆分成多个项目进行使用,这个时候我们就需要在我们的配置中自己构建一套ES的多客户端了。

文章目录

pom.xml
ElasticsearchConfig.java
ElasticsearchRestClient.java
最终

pom.xml

首先是我们的pom.xml:

<dependencies>  <dependency>   <groupId>org.springframework.boot</groupId>   <artifactId>spring-boot-starter</artifactId>   <version>2.0.5.RELEASE</version>  </dependency>    <!--es for transport-->  <dependency>   <groupId>org.elasticsearch</groupId>   <artifactId>elasticsearch</artifactId>   <version>5.6.11</version>  </dependency>  <dependency>   <groupId>org.elasticsearch.client</groupId>   <artifactId>elasticsearch-rest-client</artifactId>   <version>5.6.11</version>  </dependency>  <!--es sniffer-->  <dependency>   <groupId>org.elasticsearch.client</groupId>   <artifactId>elasticsearch-rest-client-sniffer</artifactId>   <version>5.6.3</version>   <scope>compile</scope>  </dependency>  <!--es for rest-high-level-client-->  <dependency>   <groupId>org.elasticsearch.client</groupId>   <artifactId>elasticsearch-rest-high-level-client</artifactId>   <version>5.6.11</version>  </dependency>  <dependency>   <groupId>org.apache.commons</groupId>   <artifactId>commons-lang3</artifactId>   <version>3.4</version>  </dependency>  <dependency>   <groupId>com.alibaba</groupId>   <artifactId>fastjson</artifactId>   <version>1.2.39</version>  </dependency>  <dependency>   <groupId>org.apache.logging.log4j</groupId>   <artifactId>log4j-core</artifactId>   <version>2.9.1</version>  </dependency>  <!-- SpringBoot 热启动 -->  <dependency>   <groupId>org.springframework.boot</groupId>   <artifactId>spring-boot-devtools</artifactId>  </dependency>  <dependency>   <groupId>org.apache.commons</groupId>   <artifactId>commons-pool2</artifactId>   <version>2.4.2</version>  </dependency>  <dependency>   <groupId>org.elasticsearch.client</groupId>   <artifactId>elasticsearch-rest-client-sniffer</artifactId>   <version>5.6.0</version>  </dependency>  <dependency>   <groupId>commons-io</groupId>   <artifactId>commons-io</artifactId>   <version>2.6</version>  </dependency></dependencies>

ElasticsearchConfig.java

然后是我们的配置文件,我这里使用的是application.properties的配置文件,因为我们使用不同的信息,所以这里我就不写了,可以根据需求自行获取。

ElasticsearchRestClient.java

import cnkj.site.config.ElasticsearchConfig;import lombok.extern.slf4j.Slf4j;import org.apache.http.HttpHost;import org.apache.http.auth.AuthScope;import org.apache.http.auth.UsernamePasswordCredentials;import org.apache.http.client.CredentialsProvider;import org.apache.http.client.config.RequestConfig;import org.apache.http.impl.client.BasicCredentialsProvider;import org.elasticsearch.client.RestClient;import org.elasticsearch.client.RestClientBuilder;import org.elasticsearch.client.RestHighLevelClient;import org.elasticsearch.client.sniff.SniffOnFailureListener;import org.elasticsearch.client.sniff.Sniffer;import org.elasticsearch.client.sniff.SnifferBuilder;import org.springframework.beans.factory.annotation.Qualifier;import org.springframework.context.annotation.Bean;import org.springframework.context.annotation.Configuration;/* * @version 1.0 created by LXW on 2018/11/22 9:43 */@Slf4j@Configurationpublic class ElasticsearchClient { @Bean(name = "HighESClient") public RestClient restTomcatClient(ElasticsearchConfig elasticsearchConfig) {  final CredentialsProvider credentialsProvider = new BasicCredentialsProvider();  credentialsProvider.setCredentials(AuthScope.ANY,    //es账号密码    new UsernamePasswordCredentials(elasticsearchConfig.getUsername(), elasticsearchConfig.getPassword()));  //自动扫描网段  //监听同网段服务  //Low Level Client init  RestClientBuilder builder = RestClient.builder(    new HttpHost(      elasticsearchConfig.getHost(),      Integer.valueOf(elasticsearchConfig.getPort()),      elasticsearchConfig.getSchema()    )  ).setHttpClientConfigCallback(    httpClientBuilder -> httpClientBuilder.setDefaultCredentialsProvider(credentialsProvider)  ).setRequestConfigCallback(new RestClientBuilder.RequestConfigCallback() {     @Override     public RequestConfig.Builder customizeRequestConfig(RequestConfig.Builder builder) {      builder.setConnectTimeout(elasticsearchConfig.getConnectTimeout());      builder.setSocketTimeout(elasticsearchConfig.getSocketTimeout());      return builder;     }    })    .setMaxRetryTimeoutMillis(elasticsearchConfig.getMaxRetryTimeoutMillis());  builder.setMaxRetryTimeoutMillis(elasticsearchConfig.getMaxRetryTimeoutMillis());  SniffOnFailureListener sniffOnFailureListener = new SniffOnFailureListener();  builder.setFailureListener(sniffOnFailureListener);  RestClient lowLevelRestClient = builder.build();  SnifferBuilder snifferBuilder = Sniffer.builder(lowLevelRestClient).setSniffIntervalMillis(elasticsearchConfig.getSnifferinterval());  if (elasticsearchConfig.getFailuredelay() > 0) {   snifferBuilder.setSniffAfterFailureDelayMillis(elasticsearchConfig.getFailuredelay());  }  sniffOnFailureListener.setSniffer(snifferBuilder.build());  return lowLevelRestClient; } @Bean(name = "HighLevelESClient") public RestHighLevelClient restHighLevelClient(@Qualifier("HighESClient") RestClient restClient) {  return new RestHighLevelClient(restClient); }}

最终

在需要使用的地方直接通过注入的方式使用不同的客户端

@Resource(name = "HighLevelESClient")private RestHighLevelClient client;

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持武林网。

发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表