elasticsearch自动装配

image.png

  1. package com.np.pm.common.elasticsearch.configure;
  3. import com.np.pm.common.core.utils.ListUtil;
  4. import com.np.pm.common.elasticsearch.properties.RestClientPoolProperties;
  5. import org.apache.http.auth.AuthScope;
  6. import org.apache.http.auth.UsernamePasswordCredentials;
  7. import org.apache.http.client.CredentialsProvider;
  8. import org.apache.http.impl.client.BasicCredentialsProvider;
  9. import org.elasticsearch.client.RestClientBuilder;
  10. import org.elasticsearch.client.RestHighLevelClient;
  11. import org.springframework.beans.factory.annotation.Autowired;
  12. import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
  13. import org.springframework.boot.autoconfigure.elasticsearch.ElasticsearchRestClientProperties;
  14. import org.springframework.boot.autoconfigure.elasticsearch.RestClientBuilderCustomizer;
  15. import org.springframework.boot.context.properties.EnableConfigurationProperties;
  16. import org.springframework.boot.context.properties.PropertyMapper;
  17. import org.springframework.context.annotation.Bean;
  18. import org.springframework.context.annotation.Configuration;
  19. import org.springframework.data.elasticsearch.client.ClientConfiguration;
  20. import org.springframework.data.elasticsearch.client.RestClients;
  21. import org.springframework.data.elasticsearch.config.AbstractElasticsearchConfiguration;
  22. import org.springframework.data.elasticsearch.core.ElasticsearchRestTemplate;
  24. /**
  25.  * es自动配置
  26.  *
  27.  * @author tangxingpeng
  28.  * @date 2021/2/22 21:58
  29.  */
  30. @Configuration
  31. @EnableConfigurationProperties(RestClientPoolProperties.class)
  32. public class RestAutoConfigure extends AbstractElasticsearchConfiguration {
  34.     @Autowired
  35.     private ElasticsearchRestClientProperties properties;
  37.     @Bean
  38.     public RestClientBuilderCustomizer restClientBuilderCustomizer(RestClientPoolProperties poolProperties
  39.             , ElasticsearchRestClientProperties restProperties) {
  40.         return (builder) -> {
  41.             setRequestConfig(builder, poolProperties);
  43.             setHttpClientConfig(builder, poolProperties, restProperties);
  44.         };
  45.     }
  47.     /**
  48.      * 异步httpclient连接延时配置
  49.      */
  50.     private void setRequestConfig(RestClientBuilder builder, RestClientPoolProperties poolProperties) {
  51.         builder.setRequestConfigCallback(requestConfigBuilder -> {
  52.             requestConfigBuilder.setConnectTimeout(poolProperties.getConnectTimeOut())
  53.                     .setSocketTimeout(poolProperties.getSocketTimeOut())
  54.                     .setConnectionRequestTimeout(poolProperties.getConnectionRequestTimeOut());
  55.             return requestConfigBuilder;
  56.         });
  57.     }
  59.     /**
  60.      * 异步httpclient连接数配置
  61.      */
  62.     private void setHttpClientConfig(RestClientBuilder builder, RestClientPoolProperties poolProperties, ElasticsearchRestClientProperties restProperties) {
  63.         builder.setHttpClientConfigCallback(httpClientBuilder -> {
  64.             httpClientBuilder.setMaxConnTotal(poolProperties.getMaxConnectNum())
  65.                     .setMaxConnPerRoute(poolProperties.getMaxConnectPerRoute());
  67.             PropertyMapper map = PropertyMapper.get().alwaysApplyingWhenNonNull();
  68.             map.from(restProperties::getUsername).to(username -> {
  69.                 CredentialsProvider credentialsProvider = new BasicCredentialsProvider();
  70.                 credentialsProvider.setCredentials(AuthScope.ANY,
  71.                         new UsernamePasswordCredentials(username, restProperties.getPassword()));
  72.                 httpClientBuilder.setDefaultCredentialsProvider(credentialsProvider);
  73.             });
  74.             return httpClientBuilder;
  75.         });
  76.     }
  78.     @Bean
  79.     @ConditionalOnMissingBean
  80.     public ElasticsearchRestTemplate elasticsearchRestTemplate(RestHighLevelClient restHighLevelClient) {
  81.         return new ElasticsearchRestTemplate(restHighLevelClient);
  82.     }
  84.     @Override
  85.     public RestHighLevelClient elasticsearchClient() {
  86.         final ClientConfiguration clientConfiguration = ClientConfiguration.builder()
  87.                 .connectedTo(ListUtil.toArray(properties.getUris()))
  88.                 .withBasicAuth(properties.getUsername(), properties.getPassword())
  89.                 .build();
  91.         return RestClients.create(clientConfiguration).rest();
  92.     }
  93. }