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

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

文章目录

pom.xml
ElasticsearchConfig.java
ElasticsearchRestClient.java
最终

pom.xml

首先是我们的pom.xml：

  
   org.springframework.boot
   spring-boot-starter
   2.0.5.RELEASE
  
  
  
  
   org.elasticsearch
   elasticsearch
   5.6.11
  

  
   org.elasticsearch.client
   elasticsearch-rest-client
   5.6.11
  

  
  
   org.elasticsearch.client
   elasticsearch-rest-client-sniffer
   5.6.3
   compile
  

  
  
   org.elasticsearch.client
   elasticsearch-rest-high-level-client
   5.6.11
  

  
   org.apache.commons
   commons-lang3
   3.4
  

  
   com.alibaba
   fastjson
   1.2.39
  

  
   org.apache.logging.log4j
   log4j-core
   2.9.1
  

  
  
   org.springframework.boot
   spring-boot-devtools
  

  
   org.apache.commons
   commons-pool2
   2.4.2
  

  
   org.elasticsearch.client
   elasticsearch-rest-client-sniffer
   5.6.0
  

  
   commons-io
   commons-io
   2.6
  

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;


@Slf4j
@Configuration
public 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;

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