spring 操作elasticsearch查询使用方法

发布时间 - 2026-01-11 01:12:28    点击率:

最近学习了一下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 


相关栏目: 【 网站优化151355 】 【 网络推广146373 】 【 网络技术251813 】 【 AI营销90571


相关推荐: 阿里云高弹*务器配置方案|支持分布式架构与多节点部署  如何在Windows环境下新建FTP站点并设置权限?  Laravel Pest测试框架怎么用_从PHPUnit转向Pest的Laravel测试教程  活动邀请函制作网站有哪些,活动邀请函文案?  Laravel怎么处理异常_Laravel自定义异常处理与错误页面教程  Laravel怎么写单元测试_PHPUnit在Laravel项目中的基础测试入门  如何快速搭建FTP站点实现文件共享?  如何用已有域名快速搭建网站?  如何使用 jQuery 正确渲染 Instagram 风格的标签列表  如何在阿里云部署织梦网站?  详解Nginx + Tomcat 反向代理 负载均衡 集群 部署指南  手机钓鱼网站怎么制作视频,怎样拦截钓鱼网站。怎么办?  SQL查询语句优化的实用方法总结  微博html5版本怎么弄发语音微博_语音录制入口及时长限制操作【教程】  Swift中switch语句区间和元组模式匹配  智能起名网站制作软件有哪些,制作logo的软件?  rsync同步时出现rsync: failed to set times on “xxxx”: Operation not permitted  JavaScript模板引擎Template.js使用详解  网站图片在线制作软件,怎么在图片上做链接?  Laravel如何发送邮件_Laravel Mailables构建与发送邮件的简明教程  免费视频制作网站,更新又快又好的免费电影网站?  Laravel怎么实现软删除SoftDeletes_Laravel模型回收站功能与数据恢复【步骤】  Chrome浏览器标签页分组怎么用_谷歌浏览器整理标签页技巧【效率】  Win11怎么关闭资讯和兴趣_Windows11任务栏设置隐藏小组件  Laravel如何处理和验证JSON类型的数据库字段  Laravel用户密码怎么加密_Laravel Hash门面使用教程  Python文件流缓冲机制_IO性能解析【教程】  晋江文学城电脑版官网 晋江文学城网页版直接进入  佐糖AI抠图怎样调整抠图精度_佐糖AI精度调整与放大细化操作【攻略】  EditPlus中的正则表达式 实战(1)  使用spring连接及操作mongodb3.0实例  如何在建站之星网店版论坛获取技术支持?  如何用5美元大硬盘VPS安全高效搭建个人网站?  C++时间戳转换成日期时间的步骤和示例代码  详解jQuery停止动画——stop()方法的使用  网站页面设计需要考虑到这些问题  香港代理服务器配置指南:高匿IP选择、跨境加速与SEO优化技巧  Laravel如何从数据库删除数据_Laravel destroy和delete方法区别  Laravel怎么在Controller之外的地方验证数据  百度输入法ai组件怎么删除 百度输入法ai组件移除工具  Laravel API资源类怎么用_Laravel API Resource数据转换  头像制作网站在线观看,除了站酷,还有哪些比较好的设计网站?  Win11摄像头无法使用怎么办_Win11相机隐私权限开启教程【详解】  如何在局域网内绑定自建网站域名?  Laravel怎么实现微信登录_Laravel Socialite第三方登录集成  1688铺货到淘宝怎么操作 1688一键铺货到自己店铺详细步骤  laravel怎么通过契约(Contracts)编程_laravel契约(Contracts)编程方法  DeepSeek是免费使用的吗 DeepSeek收费模式与Pro版本功能详解  laravel怎么用DB facade执行原生SQL查询_laravel DB facade原生SQL执行方法  Laravel如何操作JSON类型的数据库字段?(Eloquent示例)