Elastic Search #

package com.laoshiren.hello.elasticsearch.provider.configure;

import org.apache.http.HttpHost;
import org.elasticsearch.client.RestClient;
import org.elasticsearch.client.RestHighLevelClient;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

/**
 * ProjectName:     hello-elasticsearch
 * Package:         com.laoshiren.hello.elasticsearch.provider.configure
 * ClassName:       ElasticSearchClientConfiguration
 * Author:          laoshiren
 * Date:            2020/7/9 16:21
 * Version:         1.0.0
 */
@Configuration
public class ElasticSearchClientConfiguration {

    @Value("${laoshiren.elastic.hostname}")
    private String hostname;
    @Value("${laoshiren.elastic.port}")
    private int port;
    @Value("${laoshiren.elastic.scheme}")
    private String scheme;

    @Bean
    public RestHighLevelClient restHighLevelClient(){
        return new RestHighLevelClient(
                RestClient.builder(
                        new HttpHost(hostname, port, scheme)));
    }

}

@SpringBootTest
@RunWith(SpringRunner.class)
public class ESClient {

    @Test
    public void runEmpty(){

    }

    @Resource
    private RestHighLevelClient client;
    @Resource
    private ObjectMapper objectMapper;

    @Test
    public void initClient(){
        System.out.println(client);
    }
}

@Test
public void createIndex() throws IOException {
    // 索引请求
    CreateIndexRequest request = new CreateIndexRequest("organization");
    // 执行
    CreateIndexResponse response = client.indices()
            .create(request, RequestOptions.DEFAULT);
    System.out.println(response.index());
}

@Test
public void existsIndex() throws IOException {
    GetIndexRequest request = new GetIndexRequest("organization");
    boolean exists = client.indices()
            .exists(request, RequestOptions.DEFAULT);
    System.out.println(exists);
}

@Test
public void deleteIndex() throws IOException{
    DeleteIndexRequest request = new DeleteIndexRequest("tb_user");
    AcknowledgedResponse delete = client.indices()
            .delete(request, RequestOptions.DEFAULT);
    System.out.println(delete.isAcknowledged());
}

@Test
public void createDoc() throws Exception {
    TbUser tbUser = new TbUser();
    tbUser.setCustomerNo("0001")
            .setGrpContNo("2020")
            .setFirstName("laoshiren1207")
            .setTransAmt(900)
            .setCreateDate(new Date());
    // 指定索引
    IndexRequest request = new IndexRequest("tb_user");
    // 设置规则
    request.id("1")
            .timeout(TimeValue.timeValueSeconds(5));
    // 对象转换json
    request.source(objectMapper.writeValueAsString(tbUser), XContentType.JSON);
    // 发送请求
    IndexResponse index = client.index(request, RequestOptions.DEFAULT);
    System.out.println(index.toString());
    // 命令返回的状态
    System.out.println(index.status());
}

@Test
public void existsDoc()throws Exception{
    GetRequest request = new GetRequest("tb_user","1");
    boolean exists = client.exists(request, RequestOptions.DEFAULT);
    System.out.println(exists);
}

@Test
public void getDoc() throws Exception{
    GetRequest request = new GetRequest("tb_user","1");
    GetResponse response = client.get(request, RequestOptions.DEFAULT);
    // 所有信息
    System.out.println(response.toString());
    // 获取doc
    System.out.println(response.getSourceAsString());
}

@Test
public void postDocForUpdate() throws Exception {
    UpdateRequest request = new UpdateRequest("tb_user","1");
    request.timeout(TimeValue.timeValueSeconds(5));
    // 新对象
    TbUser tbUser = new TbUser();
    // 202007111030
    tbUser.setCreateDate(new Date());
    // 文档类型 XContentType
    request.doc(objectMapper.writeValueAsString(tbUser),XContentType.JSON);
    UpdateResponse update = client.update(request, RequestOptions.DEFAULT);
    System.out.println(update.status());
}

@Test
public void deleteDoc() throws Exception{
    DeleteRequest request = new DeleteRequest("tb_user","1");
    request.timeout(TimeValue.timeValueSeconds(5));
    DeleteResponse delete = client.delete(request,RequestOptions.DEFAULT);
    System.out.println(delete.status());
}

@Test
public void bulkInsert()throws Exception{
    BulkRequest bulkRequest = new BulkRequest();
    // 批量操作
    bulkRequest.timeout(TimeValue.timeValueSeconds(30));
    List<TbUser> list = new ArrayList<>();
    TbUser tbUser =new TbUser();
    tbUser.setCreateDate(new Date())
            .setTransAmt(23)
            .setFirstName("laoshiren")
            .setGrpContNo("00001")
            .setCustomerNo("00003");
    list.add(tbUser);
    TbUser tbUser2 =new TbUser();
    tbUser2.setCreateDate(new Date())
            .setTransAmt(23)
            .setFirstName("周杰伦")
            .setGrpContNo("00001")
            .setCustomerNo("00003");
    list.add(tbUser2);
    // 获取索引
    for (int i = 0; i< list.size(); i++) {
        //批处理请求
        IndexRequest index = new IndexRequest("tb_user")
                .id("" + (i + 2))
                // 转换json string
                .source(objectMapper.writeValueAsString(list.get(i)), XContentType.JSON);
        bulkRequest.add(index);
    }
    BulkResponse bulk = client.bulk(bulkRequest, RequestOptions.DEFAULT);
    System.out.println(bulk.status());
    System.out.println(bulk.hasFailures());
}

@Test
public void search() throws Exception{
    SearchRequest request = new SearchRequest("tb_user");
    // 构造
    SearchSourceBuilder builder = new SearchSourceBuilder();
    // 中文或者自定义字符串就要加上 field.keyword
    MatchQueryBuilder query= QueryBuilders.MatchQuery("firstName", "a");
    // 构建所需查询
    builder.query(query);
    builder.from(0);
    builder.size(5);
    builder.timeout(new TimeValue(60,TimeUnit.SECONDS));
    request.source(builder);
    SearchResponse search = client.search(request, RequestOptions.DEFAULT);
    System.out.println(search.toString());
    // 所需数据
    System.out.println(search.getHits());
    for (SearchHit hit : search.getHits().getHits()) {
        String s = objectMapper.writeValueAsString(hit);
        System.out.println(s);
    }
}

BoolQueryBuilder boolQuery = QueryBuilders.boolQuery();
map.keySet().forEach(it -> {
    if (!StringUtils.isBlank(map.get(it).toString())) {
        boolQuery.must(
                QueryBuilders.matchQuery(it, map.get(it).toString())
                        // 指定分词
                        .analyzer(analyzer));
    }
});

 @Bean(name = "restSearchClient")
public RestHighLevelClient restHighLevelClient(){
    return new RestHighLevelClient(
            RestClient.builder(new HttpHost(host,port,scheme))
                    .setRequestConfigCallback(requestConfigBuilder -> {
                        requestConfigBuilder.setConnectTimeout(-1);
                        requestConfigBuilder.setSocketTimeout(30000);
                        requestConfigBuilder.setConnectionRequestTimeout(30000);
                        return requestConfigBuilder;
                    })
    );
}

@Scheduled(cron = "0 0 * * * ? ")
public void restClientKeepAlive() {
    try {
        log.info("schedule 保持ES客户端存活 start");
        MainResponse response = restClient.info(RequestOptions.DEFAULT);
        log.info("schedule 保持ES客户端存活 end");
    } catch (IOException ignore) {}
}