springboot 使用 Spring Data Elasticsearch 高级查询文档

作者：马育民 • • 阅读：10750

上接：springboot 使用 Spring Data Elasticsearch 操作 Elasticsearch 7.9-创建工程

查询操作

service类

  1. package day1.service;
  3. import day1.entity.Toutiao;
  4. import org.elasticsearch.common.lucene.search.function.FieldValueFactorFunction;
  5. import org.elasticsearch.index.query.*;
  6. import org.elasticsearch.index.query.functionscore.FieldValueFactorFunctionBuilder;
  7. import org.elasticsearch.index.query.functionscore.FunctionScoreQueryBuilder;
  8. import org.elasticsearch.index.query.functionscore.ScoreFunctionBuilders;
  9. import org.elasticsearch.search.aggregations.AggregationBuilders;
  10. import org.elasticsearch.search.aggregations.Aggregations;
  11. import org.elasticsearch.search.aggregations.metrics.MaxAggregationBuilder;
  12. import org.elasticsearch.search.aggregations.metrics.ParsedMax;
  13. import org.elasticsearch.search.sort.SortBuilder;
  14. import org.elasticsearch.search.sort.SortBuilders;
  15. import org.elasticsearch.search.sort.SortOrder;
  16. import org.springframework.beans.factory.annotation.Autowired;
  17. import org.springframework.data.domain.PageRequest;
  18. import org.springframework.data.elasticsearch.core.ElasticsearchRestTemplate;
  19. import org.springframework.data.elasticsearch.core.SearchHits;
  20. import org.springframework.data.elasticsearch.core.query.NativeSearchQuery;
  21. import org.springframework.data.elasticsearch.core.query.NativeSearchQueryBuilder;
  22. import org.springframework.data.elasticsearch.core.query.Query;
  23. import org.springframework.stereotype.Service;
  25. @Service
  26. public class ESToutiaoSearchSrvImpl {
  27.     @Autowired
  28.     private ElasticsearchRestTemplate temp;
  29.     /**
  30.      * 短语查询
  31.      *
  32.      * @return
  33.      */
  34.     public SearchHits<Toutiao> matchPhraseQuery(String keyword) {
  35.         //1.查询title中含有 keyword 的内容的记录
  36.         MatchPhraseQueryBuilder builder = QueryBuilders.matchPhraseQuery("title", keyword);
  37.         //2.构建查询
  38.         NativeSearchQuery query=new NativeSearchQueryBuilder()
  39.                 .withQuery(builder)
  40.                 .build();
  41.         //3.执行查询
  42.         SearchHits<Toutiao> shs = temp.search(query,Toutiao.class);
  43.         return shs;
  44.     }
  45.     /**
  46.      * 通配符查询，使用keyword类型，区分大小写。按照like排序，且分页
  47.      *
  48.      * @return
  49.      */
  50.     public SearchHits<Toutiao> wildcardQuery(String keyword, Integer pageNum, Integer pageSize){
  51.         if(pageNum==null){
  52.             pageNum=1;
  53.         }
  54.         // 关键字查询
  55.         QueryBuilder wildcard=QueryBuilders.wildcardQuery("title.keyword",keyword);
  56.         //设置查询分页，从第一页开始，一页显示10条，从0开始
  57.         PageRequest page = PageRequest.of(pageNum, pageSize);
  58.         //按照点赞数量降序排序
  59.         SortBuilder sort=SortBuilders.fieldSort("like").order(SortOrder.DESC);
  61.         NativeSearchQuery nativeSearchQuery = new NativeSearchQueryBuilder()
  62.                 .withQuery(wildcard)
  63.                 .withPageable(page)
  64.                 .withSort(sort)
  65.                 .build();
  66.         SearchHits<Toutiao> shs=temp.search(nativeSearchQuery,Toutiao.class);
  67.         return shs;
  68.     }
  69.     /**
  70.      * 全文索引查询
  71.      *
  72.      * @return
  73.      */
  74.     public SearchHits<Toutiao> match(String text, Integer pageNum, Integer pageSize){
  75.         //match查询
  76.         QueryBuilder match=QueryBuilders.matchQuery("title",text);
  77.         //设置查询分页，从第一页开始，一页显示10条，从0开始
  78.         PageRequest page=PageRequest.of(pageNum,pageSize);
  79.         //按照点赞数量降序排序
  80.         SortBuilder sort=SortBuilders.fieldSort("like").order(SortOrder.DESC);
  81.         NativeSearchQuery query = new NativeSearchQueryBuilder()
  82.                 .withQuery(match)
  83.                 .withPageable(page)
  84.                 .withSort(sort)
  85.                 .build();
  86.         SearchHits<Toutiao> shs=temp.search(query,Toutiao.class);
  87.         return shs;
  88.     }
  89.     /**
  90.      * bool 查询
  91.      *
  92.      * @return
  93.      */
  94.     public SearchHits<Toutiao> boolQuery(String title, String content, Integer pageNum, Integer pageSize) {
  95.         //1.创建QueryBuilder(即设置查询条件)这儿创建的是组合查询(也叫多条件查询),后面会介绍更多的查询方法
  96.         /*组合查询BoolQueryBuilder
  97.          * must(QueryBuilders)   :AND
  98.          * mustNot(QueryBuilders):NOT
  99.          * should:               :OR
  100.          */
  101.         BoolQueryBuilder boolQueryBuilder = QueryBuilders.boolQuery();
  102.         //查询标题中必须有 入门，内容必须有 技巧
  103.         boolQueryBuilder.must(QueryBuilders.matchPhraseQuery("title",title));
  104.         boolQueryBuilder.must(QueryBuilders.matchPhraseQuery("content",content));
  105.         //设置查询分页，从第一页开始，一页显示10条，从0开始
  106.         PageRequest page = PageRequest.of(pageNum, pageSize);
  107.         //按照点赞数量降序排序
  108.         SortBuilder sort= SortBuilders.fieldSort("like").order(SortOrder.DESC);
  109.         //2.构建查询
  110.         NativeSearchQuery query=new NativeSearchQueryBuilder()
  111.                 .withQuery(boolQueryBuilder)
  112.                 .withPageable(page)
  113.                 .withSort(sort)
  114.                 .build();
  115.         //3.执行查询
  116.         SearchHits<Toutiao> shs = temp.search(query,Toutiao.class);
  117.         return shs;
  118.     }
  119.     /**
  120.      * 范围查询
  121.      *
  122.      * @return
  123.      */
  124.     public SearchHits<Toutiao> range(String text, int from, int to, Integer pageNum, Integer pageSize){
  125.         QueryBuilder match=QueryBuilders.matchQuery("content",text);
  127.         QueryBuilder range=QueryBuilders.rangeQuery("like").gt(from).lte(to);
  129.         BoolQueryBuilder boolQueryBuilder=QueryBuilders.boolQuery();
  130.         boolQueryBuilder.must(match);
  131.         boolQueryBuilder.filter(range);
  132.         //设置查询分页，从第一页开始，一页显示10条，从0开始
  133.         PageRequest page=PageRequest.of(pageNum,pageSize);
  134.         //按照点赞数量降序排序
  135.         SortBuilder sort=SortBuilders.fieldSort("like").order(SortOrder.DESC);
  136.         NativeSearchQuery query = new NativeSearchQueryBuilder()
  137.                 .withQuery(boolQueryBuilder)
  138.                 .withPageable(page)
  139.                 .withSort(sort)
  140.                 .build();
  141.         SearchHits<Toutiao> shs=temp.search(query,Toutiao.class);
  142.         return shs;
  143.     }
  144.     public double maxLike(){
  145.         MaxAggregationBuilder max= AggregationBuilders
  146.                 .max("max_like")
  147.                 .field("like");
  148.         PageRequest page=PageRequest.of(0,1);
  149.         NativeSearchQuery query = new NativeSearchQueryBuilder()
  150.                 .addAggregation(max)
  151.                 .withPageable(page)
  152.                 .build();
  153.         SearchHits<Toutiao> shs=temp.search(query,Toutiao.class);
  154.         Aggregations aggregations = shs.getAggregations();
  155.         ParsedMax agg=aggregations.get("max_like");
  156.         double val=agg.getValue();
  157.         return val;
  158.     }
  159.     /*
  160.     二次评分
  161.      */
  162.     public SearchHits<Toutiao> functionScore(String text, Integer pageNum, Integer pageSize){
  163.         MatchQueryBuilder mqb=QueryBuilders.matchQuery("title", text);
  164.         FieldValueFactorFunctionBuilder fffb=ScoreFunctionBuilders
  165.                 .fieldValueFactorFunction("like")
  166.                 .modifier(FieldValueFactorFunction.Modifier.LOG1P);
  167.         // 分数，并自动按分排序（对单个属性进行查询）
  168.         FunctionScoreQueryBuilder functionScoreQueryBuilder = QueryBuilders.
  169.                 functionScoreQuery(mqb,fffb);
  171.         PageRequest page=PageRequest.of(pageNum,pageSize);
  172.         Query query = new NativeSearchQueryBuilder()
  173.                 .withQuery(functionScoreQueryBuilder)
  174.                 .withPageable(page)
  175.                 .build();
  176.         SearchHits<Toutiao> shs = temp.search(query, Toutiao.class);
  177.         return shs;
  178.     }
  179. }

测试类

  1. package test;
  3. import day1.Main;
  4. import day1.entity.Toutiao;
  5. import day1.service.ESToutiaoSearchSrvImpl;
  6. import org.junit.Test;
  7. import org.junit.runner.RunWith;
  8. import org.springframework.beans.factory.annotation.Autowired;
  9. import org.springframework.boot.test.context.SpringBootTest;
  10. import org.springframework.data.elasticsearch.core.SearchHit;
  11. import org.springframework.data.elasticsearch.core.SearchHits;
  12. import org.springframework.test.context.junit4.SpringRunner;
  14. import java.util.List;
  16. @RunWith(SpringRunner.class)
  17. @SpringBootTest(classes = Main.class)
  18. public class TestESToutiaoSearchSrvImpl {
  19.     @Autowired
  20.     private ESToutiaoSearchSrvImpl service;
  22.     @Test
  23.     public void matchPhraseQuery(){
  24.         SearchHits<Toutiao> shs = service.matchPhraseQuery("mysql java");
  25.         //4.获取总条数(用于前端分页)
  26.         System.out.println("共："+shs.getTotalHits()+" 条记录！");
  27.         //5.获取查询到的数据内容（返回给前端）
  28.         for (SearchHit<Toutiao> item : shs) {
  29.             Toutiao toutiao=item.getContent();
  30.             System.out.println(toutiao);
  31.         }
  32.     }
  33.     @Test
  34.     public void wildcardQuery(){
  35.         // 查询 title是java 开头记录
  36.         SearchHits<Toutiao> shs = service.wildcardQuery("java*", 0, 5);
  38.         System.out.println("共："+shs.getTotalHits()+" 条记录！");
  39.         for(SearchHit<Toutiao> item : shs){
  40.             Toutiao toutiao=item.getContent();
  41.             System.out.println(toutiao);
  42.         }
  43.     }
  45.     @Test
  46.     public void match(){
  47.         SearchHits<Toutiao> shs = service.match("java mysql",1,5);
  48.         for(SearchHit<Toutiao> item : shs){
  49.             Toutiao toutiao=item.getContent();
  50.             System.out.println(toutiao);
  51.         }
  52.     }
  53.     @Test
  54.     public void boolQuery(){
  55.         SearchHits<Toutiao> shs = service.boolQuery("入门", "技巧", 0, 5);
  57.         //4.获取总条数(用于前端分页)
  58.         System.out.println("共："+shs.getTotalHits()+" 条记录！");
  59.         //5.获取查询到的数据内容（返回给前端）
  60.         for (SearchHit<Toutiao> item : shs) {
  61.             Toutiao toutiao=item.getContent();
  62.             System.out.println(toutiao);
  63.         }
  64.     }
  65.     @Test
  66.     public void range(){
  67.         SearchHits<Toutiao> shs = service.range("java", 229, 983, 0, 5);
  69.         //4.获取总条数(用于前端分页)
  70.         System.out.println("共："+shs.getTotalHits()+" 条记录！");
  71.         for(SearchHit<Toutiao> item : shs){
  72.             Toutiao toutiao=item.getContent();
  73.             System.out.println(toutiao);
  74.         }
  76.     }
  78.     @Test
  79.     public void maxLike(){
  80.         double val = service.maxLike();
  81.         System.out.println(val);
  82.     }
  83.     @Test
  84.     public void functionScore(){
  85.         SearchHits<Toutiao> shs = service.functionScore("java", 0, 5);
  87.         List<SearchHit<Toutiao>> list=shs.getSearchHits();
  88.         for(SearchHit<Toutiao> item:list){
  89.             Toutiao toutiao=item.getContent();
  90.             System.out.println("score:"+item.getScore()+"--"+toutiao);
  91.         }
  92.     }
  93. }

原文出处：http://malaoshi.top/show_1IX1tOTe226Q.html

吉ICP备15003043号