package com.corecmd.webapp.tiansjportal;
import com.alibaba.fastjson.JSON;
import com.corecmd.webapp.tiansjportal.common.enums.ResourceTypeEnum;
import com.corecmd.webapp.tiansjportal.common.utils.DateUtil;
import com.corecmd.webapp.tiansjportal.elasticsearch.domain.ResUserSearchVO;
import lombok.extern.slf4j.Slf4j;
import org.elasticsearch.action.DocWriteResponse;
import org.elasticsearch.action.admin.indices.delete.DeleteIndexRequest;
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.xcontent.XContentType;
import org.elasticsearch.index.query.BoolQueryBuilder;
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.builder.SearchSourceBuilder;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.junit4.SpringRunner;
import java.util.Date;
@Slf4j
@SpringBootTest
@RunWith(SpringRunner.class)
public class ElasticsearchTests {
@Autowired
private RestHighLevelClient restHighLevelClient;
/**
* 判断索引是否存在
*/
@Test
public void getIndex() {
try {
String indexName = "logstash-system-localhost-2021.10.31";
GetIndexRequest re = new GetIndexRequest(indexName);
boolean isExist = restHighLevelClient.indices().exists(re, RequestOptions.DEFAULT);
log.info("索引:{}是否存在:{}", indexName, isExist);
} catch (Exception e) {
e.printStackTrace();
}
}
@Test
public void createIndex() {
try {
String indexName = "tiansjportal-user-file-search";
log.info("创建索引:" + indexName);
CreateIndexRequest createIndexRequest = new CreateIndexRequest(indexName);
CreateIndexResponse response = restHighLevelClient.indices().create(createIndexRequest, RequestOptions.DEFAULT);
log.info("创建索引结果:" + response.isAcknowledged());
} catch (Exception e) {
e.printStackTrace();
}
}
@Test
public void deleteIndex() {
try {
String indexName = "tim_user_search_his";
log.info("删除索引:" + indexName);
DeleteIndexRequest deleteIndexRequest = new DeleteIndexRequest(indexName);
AcknowledgedResponse delete = restHighLevelClient.indices().delete(deleteIndexRequest, RequestOptions.DEFAULT);
log.info("删除索引结果:" + delete.isAcknowledged());
} catch (Exception e) {
e.printStackTrace();
}
}
/**
* 创建文档
* 批量插入文档时,需要建立BulkRequest,并向BulkRequest中add IndexRequest
* 最后:restHighLevelClient.bulk()发生请求
*/
@Test
public void createDoc() {
try {
String indexName = "tiansjportal-user-video-search";
log.info("向索引:{}中添加文档", indexName);
ResUserSearchVO userSearchVO = new ResUserSearchVO();
userSearchVO.setUserId(7);
userSearchVO.setKeyWord("杨");
userSearchVO.setResourceType(ResourceTypeEnum.ALBUM.getTypeEn());
userSearchVO.setSearchTime(DateUtil.getDateStr(new Date(), DateUtil.DATE_TIME_FORMAT));
String jsonType = JSON.toJSONString(userSearchVO);
log.info("文档数据:{}", jsonType);
IndexRequest indexRequest = new IndexRequest(indexName);
indexRequest.source(jsonType, XContentType.JSON);
IndexResponse response = restHighLevelClient.index(indexRequest, RequestOptions.DEFAULT);
log.info("文档创建结果:" + DocWriteResponse.Result.CREATED.equals(response.getResult()));
} catch (Exception e) {
e.printStackTrace();
}
}
//更新文档
@Test
public void updateDoc() {
try {
String indexName = "tiansjportal-user-album-search";
log.info("在索引:{}中更新文档", indexName);
ResUserSearchVO userSearchVO = new ResUserSearchVO();
userSearchVO.setUserId(7);
userSearchVO.setKeyWord("祖国风景");
userSearchVO.setResourceType(ResourceTypeEnum.ALBUM.getTypeEn());
userSearchVO.setSearchTime(DateUtil.getDateStr(new Date(), DateUtil.DATE_TIME_FORMAT));
String jsonType = JSON.toJSONString(userSearchVO);
log.info("文档数据:{}", jsonType);
UpdateRequest updateRequest = new UpdateRequest(indexName, "peRS1HwBv8L2V11VN_8F");
updateRequest.doc(jsonType, XContentType.JSON);
//获取修改后的结果
updateRequest.fetchSource(true);
UpdateResponse response = restHighLevelClient.update(updateRequest, RequestOptions.DEFAULT);
log.info("文档更新结果:" + DocWriteResponse.Result.UPDATED.equals(response.getResult()));
} catch (Exception e) {
e.printStackTrace();
}
}
//删除文档
@Test
public void deleteDoc() {
try {
String indexName = "tiansjportal-user-album-search";
log.info("在索引:{}中删除文档", indexName);
DeleteRequest deleteRequest = new DeleteRequest(indexName, "peRS1HwBv8L2V11VN_8F");
DeleteResponse response = restHighLevelClient.delete(deleteRequest, RequestOptions.DEFAULT);
log.info("文档更新结果:" + DocWriteResponse.Result.DELETED.equals(response.getResult()));
} catch (Exception e) {
e.printStackTrace();
}
}
//查询文档,根据文档id查询
@Test
public void searchDoc() {
try {
String indexName = "tiansjportal-user-album-search";
log.info("在索引:{}中查询文档", indexName);
GetRequest getRequest = new GetRequest(indexName, "oORN1HwBv8L2V11VQP9L");
GetResponse response = restHighLevelClient.get(getRequest, RequestOptions.DEFAULT);
if (response.isExists()) {
log.info("文档查询结果:" + response.toString());
} else {
log.info("文档不存在");
}
} catch (Exception e) {
e.printStackTrace();
}
}
/**
* 自定义查询,指定查询字段
*/
@Test
public void customSearch() {
try {
String indexName = "tiansjportal-user-album-search";
log.info("在索引:{}中查询文档", indexName);
SearchRequest searchRequest = new SearchRequest(indexName);
//构建查询参数
MatchQueryBuilder matchQueryBuilder = QueryBuilders.matchQuery("keyWord", "杨");
TermQueryBuilder resourceType = QueryBuilders.termQuery("resourceType", ResourceTypeEnum.ALBUM.getTypeEn());
TermQueryBuilder author = QueryBuilders.termQuery("userId", "8");
//查询参数构造器
SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder();
//向构造器添加参数
BoolQueryBuilder boolQueryBuilder = QueryBuilders.boolQuery();
boolQueryBuilder.filter(matchQueryBuilder);
boolQueryBuilder.filter(resourceType);
boolQueryBuilder.filter(author);
searchSourceBuilder.query(boolQueryBuilder);
//设置请求查询擦书
searchRequest.source(searchSourceBuilder);
SearchResponse response = restHighLevelClient.search(searchRequest, RequestOptions.DEFAULT);
//可能有多条结果
if (response.getHits().getTotalHits().value > 0) {
for (SearchHit searchHit : response.getHits().getHits()) {
log.info("查询结果:" + searchHit.getSourceAsString());
}
} else {
log.info("没有查询到数据");
}
} catch (Exception e) {
e.printStackTrace();
}
}
}
Elasticsearch-Java高级API的简单使用
最新推荐文章于 2024-07-07 16:23:04 发布