spring 操作elasticsearch查询使用方法

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

package com.gooddeep.dev.elasticsearch.commons.dao; 
 
import java.util.ArrayList; 
import java.util.List; 
import java.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; 
 
   
   
  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; 
  } 
 
   
  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; 
    } 
  } 
 
   
  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; 
    } 
  } 
   
   
  @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; 
    } 
  } 
 
 
   
  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; 
    } 
  } 
   
  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; 
    } 
  } 
 
   
  @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.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) { 
    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(); 
  } 
   
  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; 
  } 
 
 
 
   
  public abstract Class getEntityClass(); 
   
  public abstract void putClassMapping(); 
   
 
 
   
 
}