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<T> implements EsBaseDao<T> { 

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

  @Autowired
  private ElasticsearchTemplate elasticsearchTemplate; 

  @Autowired
  private Client esClient; 

  /**
   * 插入或等新，需要有id，id需要自己生成
   *
   * @param tList
   * @return
   */
  public boolean insertOrUpdate(List<T> tList) {
    List<IndexQuery> queries = new ArrayList<IndexQuery>();
    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<String> 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<String,Object> 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<ClusterHealthResponse> 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<T> queryPage(Map<String,Object> filedContentMap, final List<String> heightFields, String sortField, SortOrder order, BasePage<T>basePage) { 

    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("<em style='color:red'>").postTags("</em>").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<T> page = null;
    if (heightFields!=null&&heightFields.size() > 0) {//如果设置高亮
      page = elasticsearchTemplate.queryForPage(searchQuery,
          getEntityClass(), new SearchResultMapper() {
            @SuppressWarnings("unchecked")
            @Override
            public <T> Page<T> mapResults(SearchResponse response,Class<T> clazz, Pageable pageable) {
              List<T> chunk = new ArrayList<T>();
              for (SearchHit searchHit : response.getHits()) {
                if (response.getHits().getHits().length <= 0) {
                  return null;
                } 

                Map<String, Object> 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<T>((List<T>) chunk);
              }
              return new PageImpl<T>(new ArrayList<T>());
            } 

          });
    } else//如果不设置高亮
    {
      logger.info("#################"+qb.toString());
      page = elasticsearchTemplate.queryForPage(searchQuery,getEntityClass());
    } 

  // List<T> ts = page.getContent(); 

    basePage.setTotalRecord(page.getTotalElements());
    basePage.setResults(page.getContent());
    return basePage;
  } 

  @Override
  public List<T> queryList(Map<String, Object> filedContentMap,final List<String> 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("<em>").postTags("</em>").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<T> page = null;
    if (heightFields!=null&&heightFields.size() > 0) {//如果设置高亮
      page = elasticsearchTemplate.queryForPage(searchQuery,
          getEntityClass(), new SearchResultMapper() {
            @SuppressWarnings("unchecked")
            @Override
            public <T> Page<T> mapResults(SearchResponse response,Class<T> clazz, Pageable pageable) {
              List<T> chunk = new ArrayList<T>();
              for (SearchHit searchHit : response.getHits()) {
                if (response.getHits().getHits().length <= 0) {
                  return null;
                } 

                Map<String, Object> 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<T>((List<T>) 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<T> getEntityClass();
  /**
   * 添加各自类的影射
   */
  public abstract void putClassMapping(); 

} 

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持我们。