spring 操作elasticsearch查询使用方法

spring 操作elasticsearch查询使用方法

最近学习了一下elasticsearch使用，网上的资料又很少，真是一个头两个大。好歹最后终于了解了。留个笔记做日后查询。

package com.gooddeep.dev.elasticsearch.commons.dao;

import java.util.ArrayList;

import java.util.List;

import javaXRuFW.util.Map;

import org.elasticsearch.action.ActionFuture;

import org.elasticsearch.action.admin.cluster.health.ClusterHealthRequest;

import org.elasticsearch.action.admin.cluster.health.ClusterHealthResponse;

import org.elasticsearch.action.search.SearchResponse;

import org.elasticsearch.client.Client;

import org.elasticsearch.cluster.health.ClusterHealthStatus;

import org.elasticsearch.common.text.Text;

import org.elasticsearch.index.query.BoolQueryBuilder;

import org.elasticsearch.index.query.QueryBuilders;

import org.elasticsearch.search.SearchHit;

import org.elasticsearch.search.highlight.HighlightBuilder;

import org.elasticsearch.search.highlight.HighlightBuilder.Field;

import org.elasticsearch.search.sort.FieldSortBuilder;

import org.elasticsearch.search.sort.SortOrder;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.data.domain.Page;

import org.springframework.data.domain.PageImpl;

import org.springframework.data.domain.PageRequest;

import org.springframework.data.domain.Pageable;

import org.springframework.data.elasticsearch.core.ElasticsearchTemplate;

import org.springframework.data.elasticsearch.core.SearchResultMapper;

import org.springframework.data.elasticsearch.core.query.Criteria;

import org.springframework.data.elasticsearch.core.query.CriteriaQuery;

import org.springframework.data.elasticsearch.core.query.DeleteQuery;

import org.springframework.data.elasticsearch.core.query.IndexQuery;

import org.springframework.data.elasticsearch.core.query.IndexQueryBuilder;

import org.springframework.data.elasticsearch.core.query.NativeSearchQueryBuilder;

import org.springframework.data.elasticsearch.core.query.SearchQuery;

import org.springframework.data.elasticsearch.core.query.StringQuery;

import org.springframework.stereotype.Component;

import com.gooddeep.dev.core.helper.PropertyHelper;

import com.gooddeep.dev.core.helper.UuidHelper;

import com.gooddeep.dev.core.model.BasePage;

import com.gooddeep.dev.elasticsearch.commons.model.EsBaseBean;

import com.gooddeep.dev.elasticsearch.commons.service.EsBaseService;

@Component("esBaseDao")

public abstract class EsBaseDaoImpl implements EsBaseDao {

private Logger logger = LoggerFactory.getLogger(EsBaseService.class);

@Autowired

private ElasticsearchTemplate elasticsearchTemplate;

@Autowired

private Client esClient;

/**

* 插入或等新，需要有id，id需要自己生成

*

* @param tList

* @return

*/

public boolean insertOrUpdate(List tList) {

List queries = new ArrayList();

for (T t : tList) {

String id = ((EsBaseBean) t).getId();

if (id == null) {

id = UuidHelper.getRandomUUID();

((EsBaseBean) t).setId(id);

}

IndexQuery indexQuery = new IndexQueryBuilder().withId(id).withObject(t).build();

queries.add(indexQuery);

}

elasticsearchTemplate.bulkIndex(queries);

return true;

}

/**

* 插入或更新

*

* @param t

* @return

*/

public boolean insertOrUpdate(T t) {

String id = ((EsBaseBean) t).getId();

if (id == null) {

id = UuidHelper.getRandomUUID();

((EsBaseBean) t).setId(id);

}

try {

IndexQuery indexQuery = new IndexQueryBuilder().withId(id).withObject(t).build();

elasticsearchTemplate.index(indexQuery);

return true;

} catch (Exception e) {

logger.error("insert or update user info error.", e);

return false;

}

}

/**

* 删除

*

* @param id

* @return

*/

public boolean deleteById(String id) {

try {

elasticsearchTemplate.delete(getEntityClass(), id);

return true;

} catch (Exception e) {

logger.error("delete " + getEntityClass() + " by id " + id

+ " error.", e);

return false;

}

}

/**

* 删除ids

* @param idList

* @return

*/

@Override

public boolean deleteByIds(List idList) {

try {

CriteriaQuery criteriaQuery = new CriteriaQuery(new Criteria());

criteriaQuery.setIds(idList);

elasticsearchTemplate.delete(criteriaQuery, getEntityClass());

return true;

} catch (Exception e) {

e.printStackTrace();

return false;

}

}

/**

* 根据条件查询

* @param filedContentMap 不能为null

* @return

*/

public boolean deleteByQuery(Map filedContentMap) {

try {

DeleteQuery dq = new DeleteQuery();

BoolQueryBuilder qb=QueryBuilders. boolQuery();

if(filedContentMap!=null)

for (String key : filedContentMap.keySet()) {//字段查询

qb.must(QueryBuilders.matchQuery(key,filedContentMap.get(key)));

}

dq.setQuery(qb);;

elasticsearchTemplate.delete(dq, getEntityClass());;

return true;

} catch (Exception e) {

e.printStackTrace();

return false;

}

}

/**

* 检查健康状态

*

* @return

*/

public boolean ping() {

try {

ActionFuture health = esClient.admin()

.cluster().health(new ClusterHealthRequest());

ClusterHealthStatus status = health.actionGet().getStatus();

if (status.value() == ClusterHealthStatus.RED.value()) {

throw new RuntimeException(

"elasticsearch cluster health status is red.");

}

return true;

} catch (Exception e) {

logger.error("ping elasticsearch error.", e);

return false;

}

}

/**

* 条件查询

*

* @param searchfields

* 查询字段

* @param filedContentMap

* 字段和查询内容

* @param sortField

* 排序 字段

* @param order

* 排序

* @param from

* @param size

* @return

*/

@Override

public BasePage queryPage(Map filedContentMap, final List heightFields, String sortField, SortOrder order, BasePagebasePage) {

Field[] hfields=new Field[0];

if(heightFields!=null)

{

hfields = new Field[heightFields.size()];

for (int i = 0; i < heightFields.size(); i++) {

hfields[i] = new HighlightBuilder.Field(heightFields.get(i)).preTags("").postTags("").fragmentSize(250);

}

}

NativeSearchQueryBuilder nsb = new NativeSearchQueryBuilder().withHighlightFields(hfields);//高亮字段

if (sortField != null && order != null)//排序

nsb.withSort(new FieldSortBuilder(sortField).ignoreUnmapped(true).order(order));

if (basePage != null)//分页

nsb.withPageable(new PageRequest(basePage.getPageNo(), basePage.getPageSize()));

BoolQueryBuilder qb=QueryBuilders. boolQuery();

for (String key : filedContentMap.keySet()) {//字段查询

qb.must(QueryBuilders.matchQuery(key,filedContentMap.get(key)));

}

//userKey=78e48b85e94911e0d285f4eec990d556

//fa6e9c5bb24a21807c59e5fd3b609e12

nsb.withQuery(qb);

SearchQuery searchQuery = nsb.build();//查询建立

Page page = null;

if (heightFields!=null&&heightFields.size() > 0) {//如果设置高亮

page = elasticsearchTemplate.queryForPahttp://ge(searchQuery,

getEntityClass(), new SearchResultMapper() {

@SuppressWarnings("unchecked")

@Override

public Page mapResults(SearchResponse response,Class clazz, Pageable pageable) {

List chunk = new ArrayList();

for (SearchHit searchHit : response.getHits()) {

if (response.getHits().getHits().length <= 0) {

return null;

}

Map entityMap = searchHit.getSource();

for (String highName : heightFields) {

Text text[]=searchHit.getHighlightFields().get(highName).fragments();

if(text.length>0)

{

String highValue = searchHit.getHighlightFields().get(highName).fragments()[0].toString();

entityMap.put(highName, highValue);

}

}

chunk.add((T) PropertyHelper.getFansheObj(

getEntityClass(), entityMap));

}

if (chunk.size() > 0) {

return new PageImpl((List) chunk);

}

return new PageImpl(new ArrayList());

}

});

} else//如果不设置高亮

{

logger.info("#################"+qb.toString());

page = elasticsearchTemplate.queryForPage(searchQuery,getEntityClass());

}

// List ts = page.getContent();

basePage.setTotalRecord(page.getTotalElements());

basePage.setResults(page.getContent());

return basePage;

}

@Override

public List queryList(Map filedContentMap,final List heightFields, String sortField, SortOrder order) {

Field[] hfields=new Field[0];

if(heightFields!=null)

{

hfields = new Field[heightFields.size()];

for (int i = 0; i < heightFields.size(); i++) {

//String o="{\"abc\" : \"[abc]\"}";

hfields[i] = new HighlightBuilder.Field(heightFields.get(i)).preTags("").postTags("").fragmentSize(250);

}

}

NativeSearchQueryBuilder nsb = new NativeSearchQueryBuilder().withHighlightFields(hfields);//高亮字段

if (sortField != null && order != null)//排序

nsb.withSort(new FieldSortBuilder(sortField).ignoreUnmapped(true).order(order));

BoolQueryBuilder qb=QueryBuilders. boolQuery();

for (String key : filedContentMap.keySet()) {//字段查询

qb.must(QueryBuilders.matchQuery(key,filedContentMap.get(key)));

}

nsb.withQuery(qb);

SearchQuery searchQuery = nsb.build();//查询建立

Page page = null;

if (heightFields!=null&&heightFields.size() > 0) {//如果设置高亮

page = elasticsearchTemplate.queryForPage(searchQuery,

getEntityClass(), new SearchResultMapper() {

@SuppressWarnings("unchecked")

@Override

public Page mapResults(SearchResponse response,Class clazz, Pageable pageable) {

List chunk = new ArrayList();

for (SearchHit searchHit : response.getHits()) {

if (response.getHits().getHits().length <= 0) {

return null;

}

Map entityMap = searchHit.getSource();

for (String highName : heightFields) {

String highValue = searchHit.getHighlightFields().get(highName).fragments()[0].toString();

entityMap.put(highName, highValue);

}

chunk.add((T) PropertyHelper.getFansheObj(getEntityClass(), entityMap));

}

if (chunk.size() > 0) {

return new PageImpl((List) chunk);

}

return null;

}

});

} else//如果不设置高亮

page = elasticsearchTemplate.queryForPage(searchQuery,getEntityClass());

return page.getContent();

}

/**

* 本类查询

*

* @param id

* @return

*/

public T queryById(String id) {

StringQuery stringQuery = new StringQuery("id=" + id);

T t = elasticsearchTemplate.queryForObject(stringQuery,

getEntityClass());

return t;

}

public ElasticsearchTemplate getElasticsearchTemplate() {

return elasticsearchTemplate;

}

public Client getEsClient() {

return esClient;

}

/**

* 得到类型

*

* @return

*/

public abstract Class getEntityClass();

/**

* 添加各自类的影射

*/

public abstract void putClassMapping();

}

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。