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(); 
   
 
 
   
 
} 

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

# spring  # elasticsearch  # 操作elasticsearch  # java使用es查询的示例代码  # SpringBoot整合ES高级查询方式  # JAVA使用ElasticSearch查询in和not in的实现方式  # java 使用ElasticSearch完成百万级数据查询附近的人功能  # Elasticsearch实现复合查询高亮结果功能  # Elasticsearch聚合查询及排序操作示例  # Elasticsearch(ES)多种查询方式案例  # 本类  # 分页  # 能为  # 大家多多  # 留个  # 日后  # 网上  # PropertyHelper  # UuidHelper  # helper  # Component  # stereotype  # model  # service  # EsBaseService  # esBaseDao  # BasePage  # EsBaseBean  # PageImpl  # StringQuery 

相关文章： 如何解决VPS建站LNMP环境配置常见问题？  香港服务器选型指南：免备案配置与高效建站方案解析  建站之星客服服务时间及联系方式如何？  如何通过西部建站助手安装IIS服务器？  常州自助建站费用包含哪些项目？  如何用IIS7快速搭建并优化网站站点？  制作假网页,招聘网的薪资待遇，会有靠谱的吗？一面试又各种折扣？  我的世界制作壁纸网站下载,手机怎么换我的世界壁纸？  网站制作多少钱一个,建一个论坛网站大约需要多少钱？  贸易公司网站制作流程,出口贸易网站设计怎么做？  网站制作费用多少钱,一个网站的运营，需要哪些费用？  交易网站制作流程,我想开通一个网站，注册一个交易网址，需要那些手续？  如何在Mac上搭建Golang开发环境_使用Homebrew安装和管理Go版本  ,购物网站怎么盈利呢？  高防服务器如何保障网站安全无虞？  制作网站的基本流程,设计网站的软件是什么？  青岛网站建设如何选择本地服务器？  如何用5美元大硬盘VPS安全高效搭建个人网站？  制作网站外包平台,自动化接单网站有哪些？  重庆网站制作公司哪家好,重庆中考招生办官方网站？  Android自定义listview布局实现上拉加载下拉刷新功能  c# 在高并发场景下，委托和接口调用的性能对比  整蛊网站制作软件,手机不停的收到各种网站的验证码短信,是手机病毒还是人为恶搞?有这种手机病毒吗？  厦门模型网站设计制作公司,厦门航空飞机模型掉色怎么办？  个人网站制作流程图片大全,个人网站如何注销？  东莞市网站制作公司有哪些,东莞找工作用什么网站好？  如何高效完成自助建站业务培训？  上海制作企业网站有哪些,上海有哪些网站可以让企业免费发布招聘信息？  如何在云指建站中生成FTP站点？  郑州企业网站制作公司,郑州招聘网站有哪些？  南京网站制作费用,南京远驱官方网站？  学生网站制作软件,一个12岁的学生写小说，应该去什么样的网站？  武汉外贸网站制作公司,现在武汉外贸前景怎么样啊？  已有域名能否直接搭建网站？  如何在建站宝盒中设置产品搜索功能？  淘宝制作网站有哪些,淘宝网官网主页？  网站建设制作、微信公众号,公明人民医院怎么在网上预约？  建站之星安装步骤有哪些常见问题？  企业宣传片制作网站有哪些,传媒公司怎么找企业宣传片项目？  c# 在ASP.NET Core中管理和取消后台任务  建站之星3.0如何解决常见操作问题？  建站主机选购指南与交易推荐：核心配置解析  开封网站制作公司,网络用语开封是什么意思？  香港服务器建站指南：免备案优势与SEO优化技巧全解析  成都网站制作价格表,现在成都广电的单独网络宽带有多少的，资费是什么情况呢？  在线ppt制作网站有哪些,请推荐几个好的课件下载的网站？  香港服务器网站生成指南：免费资源整合与高速稳定配置方案  详解一款开源免费的.NET文档操作组件DocX（.NET组件介绍之一）  湖北网站制作公司有哪些,湖北清能集团官网？  如何用花生壳三步快速搭建专属网站？