94 lines
3.0 KiB
Java
94 lines
3.0 KiB
Java
package com.zcloud.config;
|
||
|
||
import co.elastic.clients.elasticsearch.ElasticsearchAsyncClient;
|
||
import co.elastic.clients.elasticsearch.ElasticsearchClient;
|
||
import co.elastic.clients.json.jackson.JacksonJsonpMapper;
|
||
import co.elastic.clients.transport.ElasticsearchTransport;
|
||
import co.elastic.clients.transport.rest_client.RestClientTransport;
|
||
import lombok.Data;
|
||
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.impl.client.BasicCredentialsProvider;
|
||
import org.elasticsearch.client.RestClient;
|
||
import org.elasticsearch.client.RestClientBuilder;
|
||
import org.springframework.boot.context.properties.ConfigurationProperties;
|
||
import org.springframework.context.annotation.Bean;
|
||
import org.springframework.context.annotation.Configuration;
|
||
|
||
|
||
@Configuration
|
||
@ConfigurationProperties(prefix = "elasticsearch")
|
||
@Data
|
||
public class ElasticsearchConfiguration {
|
||
private String host;
|
||
private int port;
|
||
private String username;
|
||
private String password;
|
||
|
||
/**
|
||
* 解析配置的字符串，转为HttpHost对象数组
|
||
*
|
||
* @return
|
||
*/
|
||
private HttpHost toHttpHost() {
|
||
HttpHost httpHost = new HttpHost(host, port, "http");
|
||
return httpHost;
|
||
}
|
||
|
||
/**
|
||
* 同步客户端
|
||
*
|
||
* @return
|
||
* @throws Exception
|
||
*/
|
||
@Bean
|
||
public ElasticsearchClient clientBySync() throws Exception {
|
||
ElasticsearchTransport transport = getElasticsearchTransport(username, password, toHttpHost());
|
||
return new ElasticsearchClient(transport);
|
||
}
|
||
|
||
/**
|
||
* 异步客户端
|
||
*
|
||
* @return
|
||
* @throws Exception
|
||
*/
|
||
@Bean
|
||
public ElasticsearchAsyncClient clientByAsync() throws Exception {
|
||
ElasticsearchTransport transport = getElasticsearchTransport(username, password, toHttpHost());
|
||
return new ElasticsearchAsyncClient(transport);
|
||
}
|
||
|
||
/**
|
||
* 传输对象
|
||
*
|
||
* @return
|
||
* @throws Exception
|
||
*/
|
||
@Bean
|
||
public ElasticsearchTransport getTransport() throws Exception {
|
||
return getElasticsearchTransport(username, password, toHttpHost());
|
||
}
|
||
|
||
|
||
private static ElasticsearchTransport getElasticsearchTransport(String username, String passwd, HttpHost... hosts) {
|
||
// 账号密码的配置
|
||
final CredentialsProvider credentialsProvider = new BasicCredentialsProvider();
|
||
credentialsProvider.setCredentials(AuthScope.ANY, new UsernamePasswordCredentials(username, passwd));
|
||
|
||
RestClientBuilder.HttpClientConfigCallback callback = httpAsyncClientBuilder -> httpAsyncClientBuilder
|
||
.setDefaultCredentialsProvider(credentialsProvider);
|
||
|
||
// 用builder创建RestClient对象
|
||
RestClient client = RestClient
|
||
.builder(hosts)
|
||
.setHttpClientConfigCallback(callback)
|
||
.build();
|
||
return new RestClientTransport(client, new JacksonJsonpMapper());
|
||
}
|
||
|
||
|
||
}
|