Elasticsearch 应用

package com.es.elasticsearch; import com.alibaba.fastjson.JSON; import com.es.elasticsearch.pojo.User; import java.io.IOException; import java.util.ArrayList; import java.util.concurrent.TimeUnit; import javax.swing.text.Highlighter; import org.elasticsearch.action.admin.indices.create.CreateIndexRequest; import org.elasticsearch.action.admin.indices.create.CreateIndexResponse; import org.elasticsearch.action.admin.indices.delete.DeleteIndexRequest; import org.elasticsearch.action.bulk.BulkRequest; import org.elasticsearch.action.bulk.BulkResponse; import org.elasticsearch.action.delete.DeleteRequest; import org.elasticsearch.action.delete.DeleteResponse; import org.elasticsearch.action.get.GetRequest; import org.elasticsearch.action.get.GetResponse; import org.elasticsearch.action.index.IndexRequest; import org.elasticsearch.action.index.IndexResponse; import org.elasticsearch.action.search.SearchRequest; import org.elasticsearch.action.search.SearchResponse; import org.elasticsearch.action.support.master.AcknowledgedResponse; import org.elasticsearch.action.update.UpdateRequest; import org.elasticsearch.action.update.UpdateResponse; import org.elasticsearch.client.RequestOptions; import org.elasticsearch.client.RestHighLevelClient; import org.elasticsearch.client.XPackClient; import org.elasticsearch.client.indices.CreateDataStreamRequest; import org.elasticsearch.client.indices.GetIndexRequest; import org.elasticsearch.common.xcontent.XContent; import org.elasticsearch.common.xcontent.XContentType; import org.elasticsearch.core.TimeValue; import org.elasticsearch.index.query.QueryBuilders; import org.elasticsearch.index.query.TermQueryBuilder; import org.elasticsearch.search.SearchHit; import org.elasticsearch.search.builder.SearchSourceBuilder; import org.elasticsearch.search.fetch.subphase.FetchSourceContext; import org.junit.jupiter.api.Test; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Qualifier; import org.springframework.boot.test.context.SpringBootTest; @SpringBootTest class ElasticSearchApplicationTests { @Autowired @Qualifier("restHighLevelClient") private RestHighLevelClient restHighLevelClient; // 索引的创建 @Test void testCreateIndex() throws IOException { String index = "text_index"; // 1. 创建索引请求 等价于 PUT /gorit_index CreateIndexRequest request = new CreateIndexRequest(index); // 2. 执行创建请求 IndicesClient, 请求后获得响应 CreateIndexResponse response = restHighLevelClient.indices().create(request, RequestOptions.DEFAULT); System.out.println(response); } // 判断索引是否存在 如果存在则删除 @Test void testExistIndex() throws IOException { String index = "text_index"; GetIndexRequest getIndexRequest = new GetIndexRequest(index); boolean exists = restHighLevelClient.indices() .exists(getIndexRequest, RequestOptions.DEFAULT); if(exists){ // 如果存在则删除 DeleteIndexRequest deleteIndexRequest = new DeleteIndexRequest(index); AcknowledgedResponse delete = restHighLevelClient.indices() .delete(deleteIndexRequest, RequestOptions.DEFAULT); System.out.println(delete.isAcknowledged()); } } // 添加文档 @Test void testAddDoctoIndex() throws IOException { String index = "text_index"; IndexRequest indexRequest = new IndexRequest(index); // 规则 PUT /gorit_index/_doc/1 // 可以不指定ID indexRequest.id("1"); indexRequest.timeout(TimeValue.timeValueSeconds(1)); indexRequest.timeout("1s"); // 将数据放入请求json indexRequest.source(JSON.toJSON(new User("李四",19)), XContentType.JSON); //客户端发送请求 IndexResponse indexResponse = restHighLevelClient.index(indexRequest, RequestOptions.DEFAULT); System.out.println(indexResponse.toString()); System.out.println(indexResponse.status()); } // 获取文档 ，判断是否存在 get /index/doc/1 @Test void testIsExists() throws IOException { String index = "text_index"; GetRequest getRequest = new GetRequest(index,"1"); // 不获取返回的 _source 的上下文了 getRequest.fetchSourceContext(new FetchSourceContext(false)); getRequest.storedFields("_none_"); boolean exists = restHighLevelClient.exists(getRequest, RequestOptions.DEFAULT); System.out.println(exists); } // 获取文档信息 @Test void testGetDocument() throws IOException { String index = "text_index"; GetRequest getRequest = new GetRequest(index,"1"); GetResponse getResponse = restHighLevelClient.get(getRequest, RequestOptions.DEFAULT); System.out.println(getResponse); System.out.println(getResponse.getSourceAsString()); } // 更新文档信息 @Test void testUpdateDocument() throws IOException { String index = "text_index"; UpdateRequest updateRequest = new UpdateRequest(index,"1"); User user = new User("李四", 19); updateRequest.doc(JSON.toJSONString(user),XContentType.JSON); UpdateResponse update = restHighLevelClient.update(updateRequest, RequestOptions.DEFAULT); System.out.println(update.status()); } // 删除文档信息 @Test void testDeleteDocument() throws IOException { String index = "text_index"; DeleteRequest deleteRequest = new DeleteRequest(index,"1"); deleteRequest.timeout("1s"); DeleteResponse deleteResponse = restHighLevelClient.delete(deleteRequest, RequestOptions.DEFAULT); System.out.println(deleteResponse.status()); } // 批量插入数据 @Test void testBulkDocument() throws IOException { String index = "text_index"; BulkRequest BulkRequest = new BulkRequest(); BulkRequest.timeout("30s"); ArrayList<User> users = new ArrayList<>(); users.add(new User("李四", 19)); users.add(new User("李四", 19)); users.add(new User("李四", 19)); users.add(new User("李四", 19)); users.add(new User("李四", 19)); users.add(new User("李四", 19)); users.add(new User("李四", 19)); for(User user:users){ BulkRequest.add(new IndexRequest(index).source(JSON.toJSONString(user),XContentType.JSON)); } BulkResponse bulkResponse = restHighLevelClient.bulk(BulkRequest, RequestOptions.DEFAULT); // 是否失败 System.out.println(bulkResponse.hasFailures()); } // 查询 // SearchRequest 搜索请求 // SearchSourceBuilder条件构造 // HighlightBuilder 构建高亮 // TermQueryBuilder 精确查询 // MatchAllQueryBuilder // xxx QueryBuilder @Test void testSearch() throws IOException { String index = "text_index"; SearchRequest searchRequest = new SearchRequest(index); // 构建搜索的条件 SearchSourceBuilder sourceBuilder = new SearchSourceBuilder(); /** * 查询条件 使用 QueryBuilders 工具类来实现 * QueryBuilders.termQuery 精确 * QueryBuilders.matchAllQueryBuilder() 匹配所有 */ TermQueryBuilder termQueryBuilder = QueryBuilders.termQuery("name", "四");//精确查询 // MatchAllQueryBuilder matchAllQueryBuilder = QueryBuilders.matchAllQuery(); sourceBuilder.query(termQueryBuilder); // 分页 sourceBuilder.from(0); sourceBuilder.size(5); sourceBuilder.highlighter(); // 设置高亮 sourceBuilder.timeout(new TimeValue(60, TimeUnit.SECONDS)); // 构建搜索 searchRequest.source(sourceBuilder); SearchResponse searchResponse = restHighLevelClient.search(searchRequest, RequestOptions.DEFAULT); System.out.println(JSON.toJSONString(searchResponse.getHits())); System.out.println("=========================================="); for (SearchHit documentFields: searchResponse.getHits().getHits()) { System.out.println(documentFields.getSourceAsMap()); } } }