亚博vip86.com 前端设计 程序设计 Java相关 移动开发 数据库/运维 软件/图像 大数据/云计算 其他经验
当前位置:亚博vip86.com ???Java相关???Spring???查看文章
基于Springboot2.0构建ES的多客户端
来源:jb51  时间:2019/4/24 12:29:49  对本文有异议

有时候我们操作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;

/*
 * @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;

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