基于Springboot2.0如何構建ES多客戶端

這篇文章主要介紹了基于Springboot2.0如何構建ES多客戶端，具有一定借鑒價值，感興趣的朋友可以參考下，希望大家閱讀完這篇文章之后大有收獲，下面讓小編帶著大家一起了解一下。

有時候我們操作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
@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;

感謝你能夠認真閱讀完這篇文章，希望小編分享的“基于Springboot2.0如何構建ES多客戶端”這篇文章對大家有幫助，同時也希望大家多多支持億速云，關注億速云行業資訊頻道，更多相關知識等著你來學習!