package com.wen.wenesapi;
import com.alibaba.fastjson.JSON;
import com.wen.wenesapi.pojo.Question;
import com.wen.wenesapi.utils.ESConst;
import org.apache.catalina.User;
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.indices.CreateIndexRequest;
import org.elasticsearch.client.indices.CreateIndexResponse;
import org.elasticsearch.client.indices.GetIndexRequest;
import org.elasticsearch.common.unit.TimeValue;
import org.elasticsearch.common.xcontent.XContentType;
import org.elasticsearch.index.query.MatchQueryBuilder;
import org.elasticsearch.index.query.QueryBuilders;
import org.elasticsearch.index.query.TermQueryBuilder;
import org.elasticsearch.search.SearchHit;
import org.elasticsearch.search.SearchHits;
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;
import java.io.IOException;
import java.util.ArrayList;
import java.util.concurrent.TimeUnit;
/**
* ES 7.4.2 高级客户端测试API
*/
@SpringBootTest
class WenEsApiApplicationTests {
//面向对象来操作
@Autowired
@Qualifier("restHighLevelClient")
private RestHighLevelClient client;
// 测试索引的创建 Request
@Test
void testCreateIndex() throws IOException {
//1、创建索引请求
CreateIndexRequest request = new CreateIndexRequest("wen_index");
//2、客户端执行请求IndicesClient，请求后获得响应
CreateIndexResponse createIndexResponse = client.indices().create(request, RequestOptions.DEFAULT);
System.out.println(createIndexResponse);
}
//测试获取索引，判断是否存在
@Test
void testExistIndex() throws IOException {
GetIndexRequest request = new GetIndexRequest("wen_index");
boolean exists = client.indices().exists(request, RequestOptions.DEFAULT);
System.out.println(exists);
}
//测试删除索引
@Test
void testDeleteIndex() throws IOException {
DeleteIndexRequest request = new DeleteIndexRequest("wen_index");
//删除
AcknowledgedResponse delete = client.indices().delete(request, RequestOptions.DEFAULT);
System.out.println(delete.isAcknowledged());
}
//测试创建文档
@Test
void testAddDocument() throws IOException {
//创建对象
Question question = new Question("测试问题", "测试答案");
//创建请求
IndexRequest request = new IndexRequest("wen_index");
//规则 post /wen_index/_doc
request.timeout("1s");
//将我们的数据放入请求 json
request.source(JSON.toJSONString(question), XContentType.JSON);
//客户端发送请求，获取响应的结果
IndexResponse indexResponse = client.index(request, RequestOptions.DEFAULT);
System.out.println(indexResponse.toString());
System.out.println(indexResponse.status());
}
//测试获取文档，判断是否存在
@Test
void testIsExist() throws IOException {
GetRequest getRequest = new GetRequest("wen_index", "eop9WnEBrCC7a6lBmEv7");
//不获取返回的_source的上下文
getRequest.fetchSourceContext(new FetchSourceContext(false));
//排序
getRequest.storedFields("_none_");
boolean exists = client.exists(getRequest, RequestOptions.DEFAULT);
System.out.println(exists);
}
//获得文档的信息
@Test
void testGetDocument() throws IOException {
GetRequest getRequest = new GetRequest("wen_index", "eop9WnEBrCC7a6lBmEv7");
GetResponse getResponse = client.get(getRequest, RequestOptions.DEFAULT);
System.out.println(getResponse.getSourceAsString());//打印文档的内容
System.out.println(getResponse);//返回的内容和命令式一致。
}
//更新文档的信息
@Test
void testUpdateDocument() throws IOException {
UpdateRequest updateRequest = new UpdateRequest("wen_index", "eop9WnEBrCC7a6lBmEv7");
updateRequest.timeout("1s");
Question question = new Question("测试答案二", "测试问题二");
updateRequest.doc(JSON.toJSONString(question), XContentType.JSON);
UpdateResponse updateResponse = client.update(updateRequest, RequestOptions.DEFAULT);
System.out.println(updateResponse.status());
}
//删除文档记录
@Test
void testDeleteDocument() throws IOException {
DeleteRequest deleteRequest = new DeleteRequest("wen_index", "eop9WnEBrCC7a6lBmEv7");
deleteRequest.timeout("1s");
DeleteResponse deleteResponse = client.delete(deleteRequest, RequestOptions.DEFAULT);
System.out.println(deleteResponse.status());
}
//批量插入数据
@Test
void testBulkRequest() throws IOException {
BulkRequest bulkRequest = new BulkRequest();
bulkRequest.timeout("10s");
ArrayList<Question> questions = new ArrayList<>();
questions.add(new Question("问题2", "答案2"));
questions.add(new Question("问题3", "答案3"));
questions.add(new Question("问题4", "答案4"));
questions.add(new Question("问题5", "答案5"));
questions.add(new Question("问题6", "答案6"));
//批处理请求
for (Question question : questions) {
//批量更新或删除，都可以在此完成
bulkRequest.add(new IndexRequest("wen_index")
.source(JSON.toJSONString(question), XContentType.JSON));
}
BulkResponse bulkResponse = client.bulk(bulkRequest, RequestOptions.DEFAULT);
//打印是否执行失败，返回false代表成功
System.out.println(bulkResponse.hasFailures());
}
//查询
//SearchRequest 搜索请求
//SearchSourceBuilder 搜索条件构造
//
@Test
void testSearch() throws IOException {
SearchRequest searchRequest = new SearchRequest(ESConst.ES_QUESTION_INDEX);
//搜索条件构造
SearchSourceBuilder sourceBuilder = new SearchSourceBuilder();
//查询条件，建议使用 QueryBuilders
MatchQueryBuilder matchQueryBuilder = QueryBuilders.matchQuery("question", "问");
sourceBuilder.query(matchQueryBuilder);
sourceBuilder.timeout(new TimeValue(60, TimeUnit.SECONDS));
searchRequest.source(sourceBuilder);
SearchResponse searchResponse = client.search(searchRequest, RequestOptions.DEFAULT);
SearchHits searchHits = searchResponse.getHits();
System.out.println(JSON.toJSONString(searchHits));
System.out.println("=====================");
for (SearchHit searchHit:searchHits.getHits()){
System.out.println(searchHit.getSourceAsMap());
}
}
}
本文由 guoxiaorui 创作，采用 知识共享署名4.0 国际许可协议进行许可
本站文章除注明转载/出处外，均为本站原创或翻译，转载前请务必署名
最后编辑时间为: Aug 28, 2020 at 11:50 pm