@Service
public class ElasticsearchServiceImpl implements ElasticsearchService {
/**
* 锁标记
*/
private static final Object obj = new Object();
/**
* TransportClient连接
*/
@Autowired
private TransportClient getTransportClient;
/**
* 通过索引id查询
* @param index
* @param type
* @param id
* @return
*/
@Override
public ElasticSearchList getIndex(String index, String type, String id) {
ElasticSearchList result = new ElasticSearchList();
try {
JSONArray jsonArray = new JSONArray();
SearchRequestBuilder searchRequestBuilder = getTransportClient.prepareSearch(index).setTypes(type);
searchRequestBuilder.setQuery(QueryBuilders.idsQuery().addIds(id));
SearchResponse searchResponse = searchRequestBuilder.setExplain(true).execute().actionGet();
SearchHits searchHits = searchResponse.getHits();
for (SearchHit searchHit : searchHits) {
Map<String, Object> source = searchHit.getSourceAsMap();
jsonArray.add(JSONObject.parseObject(JSON.toJSONString(source)));
}
JSONObject jsonObject = ElasticSearchUtil.handleAggregations(searchResponse.getAggregations());
result.setList(jsonArray);
result.setAggregation(jsonObject);
} catch (Exception e) {
throw new ApiElasticSearchException("【查询索引异常】查询索引异常,失败原因:" + e.getMessage());
}
return result;
}
/**
* 通过queryBuilder查询
*
* @param index
* @param type
* @param queryBuilder
* @return
*/
@Override
public ElasticSearchList getIndex(String index, String type, QueryBuilder queryBuilder) {
ElasticSearchList result = new ElasticSearchList();
try {
JSONArray jsonArray = new JSONArray();
SearchRequestBuilder searchRequestBuilder = getTransportClient.prepareSearch(index).setTypes(type);
if (null != queryBuilder) {
searchRequestBuilder.setQuery(queryBuilder);
}
SearchResponse searchResponse = searchRequestBuilder.setExplain(true).execute().actionGet();
SearchHits searchHits = searchResponse.getHits();
for (SearchHit searchHit : searchHits) {
Map<String, Object> source = searchHit.getSourceAsMap();
jsonArray.add(JSONObject.parseObject(JSON.toJSONString(source)));
}
result.setList(jsonArray);
} catch (Exception e) {
throw new ApiElasticSearchException("【查询索引】" + e.getMessage());
}
return result;
}
/**
* 通过queryBuilder+SortBuilder查询
*
* @param index
* @param type
* @param queryBuilder
* @param sortBuilderList
* @return
*/
@Override
public ElasticSearchList getIndex(String index, String type, QueryBuilder queryBuilder, List<SortBuilder<?>> sortBuilderList) {
ElasticSearchList result = new ElasticSearchList();
try {
JSONArray jsonArray = new JSONArray();
SearchRequestBuilder searchRequestBuilder = getTransportClient.prepareSearch(index).setTypes(type);
if (null != queryBuilder) {
searchRequestBuilder.setQuery(queryBuilder);
}
if (null != sortBuilderList) {
for (SortBuilder<?> sortBuilder : sortBuilderList) {
searchRequestBuilder.addSort(sortBuilder);
}
}
SearchResponse searchResponse = searchRequestBuilder.setExplain(true).execute().actionGet();
SearchHits searchHits = searchResponse.getHits();
for (SearchHit searchHit : searchHits) {
Map<String, Object> source = searchHit.getSourceAsMap();
jsonArray.add(JSONObject.parseObject(JSON.toJSONString(source)));
}
result.setList(jsonArray);
} catch (Exception e) {
throw new ApiElasticSearchException("【查询索引】" + e.getMessage());
}
return result;
}
/**
* 通过queryBuilder+SortBuilder+highlightBuilder查询
*
* @param index
* @param type
* @param queryBuilder
* @param sortBuilderList
* @param highlightBuilder
* @return
*/
@Override
public ElasticSearchList getIndex(String index, String type, QueryBuilder queryBuilder, List<SortBuilder<?>> sortBuilderList, HighlightBuilder highlightBuilder) {
ElasticSearchList result = new ElasticSearchList();
try {
SearchRequestBuilder searchRequestBuilder = getTransportClient.prepareSearch(index).setTypes(type);
if (null != queryBuilder) {
searchRequestBuilder.setQuery(queryBuilder);
}
if (null != sortBuilderList) {
for (SortBuilder<?> sortBuilder : sortBuilderList) {
searchRequestBuilder.addSort(sortBu