devbox
从前慢现在也慢
这个屌丝很懒，什么也没留下！
关注作者
热门标签
热门文章
当前位置:   article > 正文

SpringBoot基于ElasticSearch7.9.2和ElasticsearchRestTemplate的一些通用接口（可复用）_elasticsearchresttemplate.bulkupdate

作者：从前慢现在也慢 | 2024-03-27 17:12:41

赞

踩

elasticsearchresttemplate.bulkupdate

之前写过一篇SpringBoot集成ElasticSearch 7.9.2 教程和简单增删改查案例icon-default.png?t=M276https://blog.csdn.net/weixin_43753812/article/details/117733551

这里提供一个通用的操作接口和文档的方法类，大多数的使用场景都已经覆盖，话不多说，开整：

  1. 首先，创建一个自定义注解，结合elasticsearch提供的注解一起，配合实体类操作文档和索引。
  1. package com.cqvip.innocence.common.annotation;
  2.  
  3. import java.lang.annotation.*;
  4.  
  5. /**
  6.  * @ClassName DocumentId
  7.  * @Description es文档数据的id标识
  8.  * @Author Innocence
  9.  * @Date 2021/6/16 11:34
  10.  * @Version 1.0
  11.  */
  12. @Target(ElementType.FIELD)
  13. @Retention(RetentionPolicy.RUNTIME)
  14. @Documented
  15. @Inherited
  16. public @interface DocumentId {
  17. }

2.写一个与文档对应的DTO（这里只是示例，具体的实体根据对应需求建立）

  1. package com.cqvip.innocence.project.model.entity;
  2.  
  3. import com.cqvip.innocence.common.annotation.DocumentId;
  4. import lombok.Data;
  5. import org.springframework.data.elasticsearch.annotations.Document;
  6. import org.springframework.data.elasticsearch.annotations.Field;
  7. import org.springframework.data.elasticsearch.annotations.FieldType;
  8.  
  9. import java.io.Serializable;
  10.  
  11. /**
  12.  * @ClassName EsTestInfo
  13.  * @Description es测试数据
  14.  * @Author Innocence
  15.  * @Date 2021/6/16 13:57
  16.  * @Version 1.0
  17.  */
  18. @Data
  19. @Document(indexName = "test_info")
  20. public class EsTestInfo implements Serializable {
  21.     private static final long serialVersionUID = 6735913456541605819L;
  22.  
  23.     @DocumentId
  24.     @Field(type = FieldType.Keyword)
  25.     private String infoId;
  26.  
  27.     private String name;
  28.  
  29.     @Field(type = FieldType.Integer)
  30.     private Integer age;
  31.  
  32.     @Field(type = FieldType.Keyword)
  33.     private String sex;
  34.  
  35.     @Field(type = FieldType.Keyword)
  36.     private String[] roles;
  37. }

 3.索引操作的接口和实现

  1. package com.cqvip.innocence.project.esservice;
  2.  
  3. /**
  4.  * @ClassName IndexService
  5.  * @Description 索引的增删改查
  6.  * @Author Innocence
  7.  * @Date 2021/3/1 14:06
  8.  * @Version 1.0
  9.  */
  10. public interface IndexService<T>{
  11.  
  12.     /**
  13.      * 根据类创建索引
  14.      * @author Innocence
  15.      * @date 2021/6/16
  16.      * @param clazz
  17.      * @return java.lang.Boolean
  18.      */
  19.     Boolean createIndexByClass(Class<T> clazz);
  20.  
  21.     /**
  22.      * 根据名称创建索引
  23.      * 不建议使用，因为很多设置需要自己写配置文件
  24.      * @author Innocence
  25.      * @date 2021/3/1
  26.      * @param indexName
  27.      * @return java.lang.Boolean
  28.      */
  29.     Boolean createIndexByName(String indexName);
  30.  
  31.     /**
  32.      * 根据索引名判断索引是否存在
  33.      * @author Innocence
  34.      * @date 2021/6/16
  35.      * @param indexName 索引名
  36.      * @return java.lang.Boolean
  37.      */
  38.     Boolean isIndexExist(String indexName);
  39.  
  40.     /**
  41.      * 根据索引名删除索引
  42.      * @author Innocence
  43.      * @date 2021/6/16
  44.      * @param indexName 索引名
  45.      * @return java.lang.Boolean
  46.      */
  47.     Boolean deleteIndexByName(String indexName);
  48.  
  49.     /**
  50.      * 获取被标记的索引名
  51.      * @author Innocence
  52.      * @date 2021/6/16
  53.      * @param clazz
  54.      * @return java.lang.String
  55.      */
  56.     String getIndexName(Class<T> clazz);
  57. }
  1. package com.cqvip.innocence.project.esservice.impl;
  2.  
  3. import cn.hutool.core.util.StrUtil;
  4. import com.cqvip.innocence.project.esservice.IndexService;
  5. import org.springframework.beans.factory.annotation.Autowired;
  6. import org.springframework.data.elasticsearch.annotations.Document;
  7. import org.springframework.data.elasticsearch.core.ElasticsearchRestTemplate;
  8. import org.springframework.data.elasticsearch.core.IndexOperations;
  9. import org.springframework.data.elasticsearch.core.mapping.IndexCoordinates;
  10. import org.springframework.stereotype.Service;
  11.  
  12. import java.lang.annotation.Annotation;
  13.  
  14. /**
  15.  * @ClassName IndexServiceImpl
  16.  * @Description  抽象
  17.  * @Author Innocence
  18.  * @Date 2021/3/1 14:24
  19.  * @Version 1.0
  20.  */
  21. @Service
  22. public class IndexServiceImpl<T> implements IndexService<T> {
  23.  
  24.     @Autowired
  25.     private ElasticsearchRestTemplate restTemplate;
  26.  
  27.     @Override
  28.     public String getIndexName(Class<T> clazz){
  29.         Annotation documentAnnotation = clazz.getDeclaredAnnotation(Document.class);
  30.         if(documentAnnotation==null){
  31.             return null;
  32.         }
  33.         String indexName = ((Document) documentAnnotation).indexName();
  34.         if (StrUtil.isNotBlank(indexName)){
  35.             return indexName;
  36.         }
  37.         return null;
  38.     }
  39.  
  40.     @Override
  41.     public Boolean createIndexByClass(Class<T> clazz) {
  42.         Boolean indexExist = isIndexExist(getIndexName(clazz));
  43.         if (indexExist){
  44.             return false;
  45.         }
  46.         IndexOperations indexOps = restTemplate.indexOps(clazz);
  47.         boolean result1 = indexOps.create();
  48.         boolean result2 = indexOps.putMapping(indexOps.createMapping(clazz));
  49.         return result1&result2;
  50.     }
  51.  
  52.     @Override
  53.     public Boolean createIndexByName(String indexName) {
  54.         Boolean indexExist = isIndexExist(indexName);
  55.         if (indexExist){
  56.             return false;
  57.         }
  58.         IndexOperations indexOps = restTemplate.indexOps(IndexCoordinates.of(indexName));
  59.         return indexOps.create();
  60.  
  61.     }
  62.  
  63.     @Override
  64.     public Boolean isIndexExist(String indexName) {
  65.         IndexOperations indexOps = restTemplate.indexOps(IndexCoordinates.of(indexName));
  66.         return indexOps.exists();
  67.     }
  68.  
  69.     @Override
  70.     public Boolean deleteIndexByName(String indexName) {
  71.         IndexOperations indexOps = restTemplate.indexOps(IndexCoordinates.of(indexName));
  72.         return indexOps.delete();
  73.  
  74.     }
  75. }

4.文档操作的接口和实现类

  1. package com.cqvip.innocence.project.esservice;
  2.  
  3. import org.elasticsearch.search.aggregations.bucket.terms.Terms;
  4. import org.elasticsearch.search.builder.SearchSourceBuilder;
  5. import org.elasticsearch.search.fetch.subphase.highlight.HighlightBuilder;
  6. import org.springframework.data.domain.PageRequest;
  7. import org.springframework.data.elasticsearch.core.query.Query;
  8.  
  9. import java.io.IOException;
  10. import java.util.List;
  11. import java.util.Map;
  12.  
  13. /**
  14.  * @ClassName DocumentService
  15.  * @Description 适用于本项目的数据的增删改查
  16.  * @Author Innocence
  17.  * @Date 2021/3/1 15:43
  18.  * @Version 1.0
  19.  */
  20. public interface DocumentService<T> {
  21.  
  22.     /**
  23.      * 设置高亮字段
  24.      * @author Innocence
  25.      * @date 2021/3/2
  26.      * @param fields 需要设置的高亮字段
  27.      * @return org.elasticsearch.search.fetch.subphase.highlight.HighlightBuilder
  28.      */
  29.     HighlightBuilder getHighlightBuilder(String[] fields);
  30.  
  31.     /**
  32.      * 根据id判断文档是否存在于指定索引中
  33.      * @author Innocence
  34.      * @date 2021/3/1
  35.      * @param id
  36.      * @param clazz
  37.      * @return java.lang.Boolean
  38.      */
  39.     Boolean isExist(String id, Class<T> clazz);
  40.  
  41.     /**
  42.      *  单条数据插入
  43.      * @author Innocence
  44.      * @date 2021/3/1
  45.      * @param t 需插入的数据
  46.      * @return java.lang.String 返回文档id
  47.      */
  48.     String saveByEntity(T t);
  49.  
  50.     /**
  51.      * 批量插入
  52.      * @author Innocence
  53.      * @date 2021/3/1
  54.      * @param entities 待插入的数据实体集合
  55.      * @return java.util.List<java.lang.String> 返回idList
  56.      */
  57.     List<String> saveBatchByEntities(List<T> entities) throws Exception;
  58.  
  59.     /**
  60.      * 单条数据更新
  61.      * @author Innocence
  62.      * @date 2021/3/1
  63.      * @param t 需要更新的实体数据
  64.      * @return void
  65.      */
  66.     void updateByEntity(T t);
  67.  
  68.     /**
  69.      * 批量更新
  70.      * @author Innocence
  71.      * @date 2021/3/1
  72.      * @param entities 待更新的数据实体集合
  73.      * @return void
  74.      */
  75.     void updateByEntities(List<T> entities);
  76.  
  77.     /**
  78.      * 根据id删除数据
  79.      * @author Innocence
  80.      * @date 2021/3/1
  81.      * @param id
  82.      * @param clazz
  83.      * @return java.lang.String 被删除的id
  84.      */
  85.     String deleteById(String id, Class<T> clazz);
  86.  
  87.     /**
  88.      * 根据id批量删除数据
  89.      * @author Innocence
  90.      * @date 2021/3/12
  91.      * @param ids
  92.      * @param clazz
  93.      * @return void
  94.      */
  95.     void deleteByIds(List<String> ids, Class<T> clazz);
  96.  
  97.     /**
  98.      * 根据条件删除数据
  99.      * @author Innocence
  100.      * @date 2021/3/1
  101.      * @param query 条件构造器
  102.      * @param clazz
  103.      * @return void
  104.      */
  105.     void deleteByQuery(Query query, Class<T> clazz);
  106.  
  107.     /**
  108.      * 根据id查询数据 (基于注解形式设置了索引mapping)
  109.      * @author Innocence
  110.      * @date 2021/3/1
  111.      * @param clazz
  112.      * @return T
  113.      */
  114.     T getEntityById(String id,Class<T> clazz);
  115.  
  116.     /**
  117.      * 根据id批量查询数据 (基于注解形式设置了索引mapping)
  118.      * @author Innocence
  119.      * @date 2021/8/24
  120.      * @return java.util.List<T>
  121.      */
  122.     List<T> getEntityByIds(List<String> ids, Class<T> clazz);
  123.  
  124.     /**
  125.      * 查询符合条件的总条数
  126.      * @author Innocence
  127.      * @date 2021/3/2
  128.      * @return java.lang.Long
  129.      */
  130.     Long getCount(Query query,Class<T> clazz);
  131.  
  132.     /**
  133.      * 查询符合条件的实体list
  134.      * @author Innocence
  135.      * @date 2021/3/2
  136.      * @param query 构建的查询条件 主要使用NativeSearchQuery 来进行构造
  137.      * @param isHighLight 是否映射高亮（true映射高亮，否则不映射）
  138.      * @return java.util.List<T>
  139.      */
  140.     List<T> getInfoList(Query query, Class<T> clazz,Boolean isHighLight);
  141.  
  142.     /**
  143.      * 查询符合条件的分页
  144.      * @author Innocence
  145.      * @date 2021/3/2
  146.      * @param query 构建的查询条件 主要使用NativeSearchQuery 来进行构造
  147.      * @return org.springframework.data.domain.Page<T>
  148.      */
  149.     Map<String,Object> getPageList(Query query, PageRequest pageRequest,Class<T> clazz);
  150.  
  151.     /**
  152.      * 根据条件获取聚类信息
  153.      * springboot-data 封装的聚类查询速度很慢，这里直接用client操作
  154.      * @author Innocence
  155.      * @date 2021/3/19
  156.      * @param query, clazz
  157.      * @return java.util.Map<java.lang.String,java.util.List<? extends org.elasticsearch.search.aggregations.bucket.terms.Terms.Bucket>>
  158.      */
  159.     Map<String, List<? extends Terms.Bucket>> getFacetByQuery(SearchSourceBuilder query, Class<T> clazz) throws IOException;
  160. }

 

  1. package com.cqvip.innocence.project.esservice.impl;
  2.  
  3. import cn.hutool.core.util.StrUtil;
  4. import com.alibaba.fastjson.JSON;
  5. import com.cqvip.innocence.common.annotation.DocumentId;
  6. import com.cqvip.innocence.common.exception.ElasticServiceException;
  7. import com.cqvip.innocence.project.esservice.DocumentService;
  8. import com.cqvip.innocence.project.esservice.IndexService;
  9. import org.apache.commons.beanutils.PropertyUtils;
  10. import org.apache.lucene.search.IndexSearcher;
  11. import org.elasticsearch.action.search.SearchRequest;
  12. import org.elasticsearch.action.search.SearchResponse;
  13. import org.elasticsearch.client.RequestOptions;
  14. import org.elasticsearch.client.RestHighLevelClient;
  15. import org.elasticsearch.index.query.QueryBuilders;
  16. import org.elasticsearch.search.aggregations.Aggregation;
  17. import org.elasticsearch.search.aggregations.Aggregations;
  18. import org.elasticsearch.search.aggregations.bucket.terms.ParsedStringTerms;
  19. import org.elasticsearch.search.aggregations.bucket.terms.Terms;
  20. import org.elasticsearch.search.builder.SearchSourceBuilder;
  21. import org.elasticsearch.search.fetch.subphase.highlight.HighlightBuilder;
  22. import org.springframework.beans.factory.annotation.Autowired;
  23. import org.springframework.beans.factory.annotation.Qualifier;
  24. import org.springframework.data.domain.Page;
  25. import org.springframework.data.domain.PageImpl;
  26. import org.springframework.data.domain.PageRequest;
  27. import org.springframework.data.elasticsearch.core.*;
  28. import org.springframework.data.elasticsearch.core.document.Document;
  29. import org.springframework.data.elasticsearch.core.mapping.IndexCoordinates;
  30. import org.springframework.data.elasticsearch.core.query.*;
  31. import org.springframework.stereotype.Service;
  32.  
  33. import java.io.IOException;
  34. import java.lang.reflect.*;
  35. import java.lang.reflect.Field;
  36. import java.util.ArrayList;
  37. import java.util.HashMap;
  38. import java.util.List;
  39. import java.util.Map;
  40.  
  41.  
  42. /**
  43.  * @ClassName DocumentServiceImpl
  44.  * @Description TODO
  45.  * @Author Innocence
  46.  * @Date 2021/3/1 16:27
  47.  * @Version 1.0
  48.  */
  49. @Service
  50. public class DocumentServiceImpl<T> implements DocumentService<T> {
  51.  
  52.     @Autowired
  53.     private ElasticsearchRestTemplate restTemplate;
  54.  
  55.     @Qualifier("highLevelClient")
  56.     @Autowired
  57.     private RestHighLevelClient highLevelClient;
  58.  
  59.     @Autowired
  60.     private IndexService<T> indexService;
  61.  
  62.     /**
  63.      * 反射获取es实体代表id的字段名
  64.      * @author Innocence
  65.      * @date 2021/6/17
  66.      * @param clazz
  67.      * @return java.lang.String
  68.      */
  69.     private String getPrimaryNameByClass(Class<T> clazz){
  70.         Field[] fields = clazz.getDeclaredFields();
  71.         for (Field f:fields) {
  72.             DocumentId id = f.getAnnotation(DocumentId.class);
  73.             if (id != null){
  74.                 return f.getName();
  75.             }
  76.         }
  77.         return null;
  78.     }
  79.  
  80.     /**
  81.      * 根据注解获取实体类关联es属性的id值
  82.      * @author Innocence
  83.      * @date 2021/6/16
  84.      * @return java.lang.String
  85.      */
  86.     private String getPrimaryValueByEntity(T t){
  87.         Class<T> clazz = (Class<T>) t.getClass();
  88.         Field[] fields = clazz.getDeclaredFields();
  89.         for (Field f:fields) {
  90.             Class<?> type = f.getType();
  91.             if (type.getTypeName().equals(String.class.getTypeName())){
  92.                 DocumentId id = f.getAnnotation(DocumentId.class);
  93.                 if (id != null){
  94.                     try {
  95.                         String replace = f.getName()
  96.                                 .replace(f.getName().substring(0, 1), f.getName().substring(0, 1).toUpperCase());
  97.                         Method method = clazz.getMethod("get" + replace);
  98.                         String invoke = (String) method.invoke(t);
  99.                         if (StrUtil.isNotBlank(invoke)){
  100.                             return invoke;
  101.                         }else {
  102.                             return null;
  103.                         }
  104.                     } catch (NoSuchMethodException e) {
  105.                         e.printStackTrace();
  106.                     } catch (IllegalAccessException e) {
  107.                         e.printStackTrace();
  108.                     } catch (InvocationTargetException e) {
  109.                         e.printStackTrace();
  110.                     }
  111.                 }
  112.             }
  113.         }
  114.         return null;
  115.     }
  116.  
  117.     /**
  118.      * 类上无索引名的统一异常处理
  119.      * @author Innocence
  120.      * @date 2021/8/6
  121.      * @param clazz
  122.      * @return void
  123.      */
  124.     private String indexNameExceptionHandler(Class<T> clazz){
  125.         String indexName = indexService.getIndexName(clazz);
  126.         if (StrUtil.isBlank(indexName)){
  127.             throw new ElasticServiceException("The index name on the current class does not exist！");
  128.         }else {
  129.             return indexName;
  130.         }
  131.     }
  132.  
  133.     /**
  134.      * 映射高亮字段到原生属性
  135.      * @author Innocence
  136.      * @date 2021/3/2
  137.      * @param searchHits
  138.      * @return java.util.List<T>
  139.      */
  140.     private <T> List<T> mappingHighlight(List<SearchHit<T>> searchHits){
  141.         List<T> infoList = new ArrayList<>();
  142.         for (SearchHit<T> searchHit : searchHits) {
  143.             T content = searchHit.getContent();
  144.             Map<String, List<String>> highlightFields = searchHit.getHighlightFields();
  145.             for (Map.Entry<String, List<String>> entry : highlightFields.entrySet()) {
  146.                 try {
  147.                     PropertyUtils.setProperty(content,entry.getKey(),entry.getValue().get(0));
  148.                 } catch (IllegalAccessException | InvocationTargetException | NoSuchMethodException e) {
  149.                     e.printStackTrace();
  150.                 }
  151.             }
  152.             infoList.add(content);
  153.         }
  154.         return infoList;
  155.     }
  156.  
  157.     @Override
  158.     public HighlightBuilder getHighlightBuilder(String[] fields) {
  159.         HighlightBuilder highlightBuilder = new HighlightBuilder();
  160.         for (String field : fields) {
  161.             highlightBuilder.field(field);
  162.         }
  163.         highlightBuilder.requireFieldMatch(false);     //如果要多个字段高亮,这项要为false
  164.         highlightBuilder.preTags("<span style=\"color:red\">");
  165.         highlightBuilder.postTags("</span>");
  166.         //下面这两项,如果你要高亮如文字内容等有很多字的字段,必须配置,不然会导致高亮不全,文章内容缺失等
  167.         highlightBuilder.fragmentSize(800000); //最大高亮分片数
  168.         highlightBuilder.numOfFragments(0); //从第一个分片获取高亮片段
  169.  
  170.         return highlightBuilder;
  171.     }
  172.  
  173.  
  174.     @Override
  175.     public Boolean isExist(String id,Class<T> clazz) {
  176.         String name = indexNameExceptionHandler(clazz);
  177.         return restTemplate.exists(id, IndexCoordinates.of(name));
  178.     }
  179.  
  180.     @Override
  181.     public  String saveByEntity(T t) {
  182.         String id = getPrimaryValueByEntity(t);
  183.         String indexName = indexNameExceptionHandler((Class<T>) t.getClass());
  184.         if (StrUtil.isBlank(id)){
  185.             throw new ElasticServiceException("Document id cannot be empty！");
  186.         }
  187.         IndexQuery build = new IndexQueryBuilder().withId(id).withObject(t).build();
  188.         String index = restTemplate.index(build, IndexCoordinates.of(indexName));
  189.         //业务需要，新增后马上刷新，ElasticsearchRestTemplate是默认不立即刷新（立即刷新会影响性能）
  190.         restTemplate.indexOps(t.getClass()).refresh();
  191.         return index;
  192.     }
  193.  
  194.     @Override
  195.     public List<String> saveBatchByEntities(List<T> entities){
  196.         Class<T> clazz =(Class<T>) entities.get(0).getClass();
  197.         String indexName = indexNameExceptionHandler(clazz);
  198.         List<IndexQuery> queryList = new ArrayList<>();
  199.         for (T item:entities){
  200.             String id = getPrimaryValueByEntity(item);
  201.             if (StrUtil.isBlank(id)){
  202.                 throw new ElasticServiceException("Document id cannot be empty！");
  203.             }
  204.             IndexQuery build = new IndexQueryBuilder().withId(id).withObject(item).build();
  205.             queryList.add(build);
  206.         }
  207.         List<String> idList = restTemplate.bulkIndex(queryList, IndexCoordinates.of(indexName));
  208.         restTemplate.indexOps(IndexCoordinates.of(indexName)).refresh();
  209.         return idList;
  210.     }
  211.  
  212.     @Override
  213.     public void updateByEntity(T t) {
  214.         String indexName = indexNameExceptionHandler((Class<T>) t.getClass());
  215.         Document document = Document.parse(JSON.toJSONString(t));
  216.         document.setId(getPrimaryValueByEntity(t));
  217.         UpdateQuery build = UpdateQuery.builder(document.getId())
  218.                 .withRefresh(UpdateQuery.Refresh.Wait_For) //更新后立即刷新可见（影响性能，按需注释或使用）
  219.                 .withDocument(document)
  220.                 .build();
  221.         restTemplate.update(build, IndexCoordinates.of(indexName));
  222.     }
  223.  
  224.     @Override
  225.     public void updateByEntities(List<T> entities) {
  226.         Class<T> clazz =(Class<T>) entities.get(0).getClass();
  227.         String indexName = indexNameExceptionHandler(clazz);
  228.         List<UpdateQuery> updateQueries = new ArrayList<>();
  229.         entities.forEach(item->{
  230.             Document document = Document.parse(JSON.toJSONString(item));
  231.             document.setId(getPrimaryValueByEntity(item));
  232.             UpdateQuery build = UpdateQuery.builder(document.getId())
  233.                     .withRefresh(UpdateQuery.Refresh.Wait_For)
  234. //                    .withDocAsUpsert(true) //不加默认false。true表示更新时不存在就插入
  235.                     .withDocument(document)
  236.                     .build();
  237.             updateQueries.add(build);
  238.         });
  239.         restTemplate.bulkUpdate(updateQueries,IndexCoordinates.of(indexName));
  240.     }
  241.  
  242.     @Override
  243.     public String deleteById(String id, Class<T> clazz) {
  244.         String indexName = indexNameExceptionHandler(clazz);
  245.         return restTemplate.delete(id,IndexCoordinates.of(indexName));
  246.     }
  247.  
  248.     @Override
  249.     public void deleteByIds(List<String> ids, Class<T> clazz) {
  250.         String indexName = indexNameExceptionHandler(clazz);
  251.         StringQuery query = new StringQuery(QueryBuilders.termsQuery(getPrimaryNameByClass(clazz), ids).toString());
  252.         restTemplate.delete(query,clazz,IndexCoordinates.of(indexName));
  253.     }
  254.  
  255.     @Override
  256.     public void deleteByQuery(Query query,Class<T> clazz) {
  257.         String indexName = indexNameExceptionHandler(clazz);
  258.         restTemplate.delete(query,clazz,IndexCoordinates.of(indexName));
  259.     }
  260.  
  261.     @Override
  262.     public T getEntityById(String id,Class<T> clazz) {
  263.         return restTemplate.get(id,clazz);
  264.     }
  265.  
  266.     @Override
  267.     public List<T> getEntityByIds(List<String> ids, Class<T> clazz) {
  268.         String indexName = indexNameExceptionHandler(clazz);
  269.         NativeSearchQuery build = new NativeSearchQueryBuilder().withIds(ids).build();
  270.         return restTemplate.multiGet(build, clazz, IndexCoordinates.of(indexName));
  271.     }
  272.  
  273.     @Override
  274.     public Long getCount(Query query,Class<T> clazz) {
  275.         return restTemplate.count(query,clazz);
  276.     }
  277.  
  278.     @Override
  279.     public List<T> getInfoList(Query query,Class<T> clazz,Boolean isHighLight) {
  280.         query.setTrackTotalHits(true);
  281.         SearchHits<T> search = restTemplate.search(query,clazz);
  282.         if (isHighLight){
  283.             return mappingHighlight(search.getSearchHits());
  284.         }
  285.         List<T> infoList = new ArrayList<>();
  286.         for (SearchHit<T> searchHit : search){
  287.             infoList.add(searchHit.getContent());
  288.         }
  289.         return infoList;
  290.     }
  291.  
  292.     @Override
  293.     public Map<String, Object> getPageList(Query query, PageRequest pageRequest,Class<T> clazz) {
  294.         query.setTrackTotalHits(true);
  295.         query.setPageable(pageRequest);
  296.         SearchHits<T> search = restTemplate.search(query,clazz);
  297.         Aggregations aggregations = search.getAggregations();
  298.         List<SearchHit<T>> searchHits = search.getSearchHits();
  299.         List<T> esSourceInfos = mappingHighlight(searchHits);
  300.         Page infos = new PageImpl(
  301.                 esSourceInfos,
  302.                 pageRequest,
  303.                 search.getTotalHits());
  304.         Map<String, Object> map = new HashMap<>();
  305.         map.put("page",infos);
  306.         map.put("ag",formatFacet(aggregations));
  307.         return map;
  308.     }
  309.  
  310.     @Override
  311.     public Map<String, List<? extends Terms.Bucket>> getFacetByQuery(SearchSourceBuilder query, Class<T> clazz) throws IOException {
  312.         String indexName = indexNameExceptionHandler(clazz);
  313.         SearchRequest request = new SearchRequest(indexName);
  314.         SearchSourceBuilder builder = query;
  315.         request.source(builder);
  316.         SearchResponse response = highLevelClient.search(request, RequestOptions.DEFAULT);
  317.         Aggregations aggregations = response.getAggregations();
  318.         return formatFacet(aggregations);
  319.     }
  320.  
  321.  
  322.     /**
  323.      * 格式化聚类数据
  324.      * @author Innocence
  325.      * @date 2021/3/18
  326.      * @param aggregations
  327.      * @return java.util.Map<java.lang.String,java.util.List<java.util.Map<java.lang.String,java.lang.Object>>>
  328.      */
  329.     private Map<String, List<? extends Terms.Bucket>> formatFacet(Aggregations aggregations){
  330.         if (aggregations == null){
  331.             return null;
  332.         }
  333.         Map<String, List<? extends Terms.Bucket>> map = new HashMap<>();
  334.         List<Aggregation> list = aggregations.asList();
  335.         list.forEach(item->{
  336.             ParsedStringTerms newItem = (ParsedStringTerms) item;
  337.             String name = newItem.getName();
  338.             List<? extends Terms.Bucket> buckets = newItem.getBuckets();
  339.             map.put(name,buckets);
  340.         });
  341.         return map;
  342.     }
  343. }

 5.使用示例，使用@Autowired自动注入，并且带上泛型类

  1. package com.cqvip.innocence.tests;
  2.  
  3. import com.cqvip.innocence.project.esservice.DocumentService;
  4. import com.cqvip.innocence.project.esservice.IndexService;
  5. import com.cqvip.innocence.project.model.entity.EsTestInfo;
  6. import org.apache.commons.lang3.builder.ToStringExclude;
  7. import org.elasticsearch.index.query.BoolQueryBuilder;
  8. import org.elasticsearch.index.query.QueryBuilders;
  9. import org.junit.jupiter.api.Test;
  10. import org.springframework.beans.factory.annotation.Autowired;
  11. import org.springframework.boot.test.context.SpringBootTest;
  12. import org.springframework.data.elasticsearch.core.query.NativeSearchQuery;
  13. import org.springframework.data.elasticsearch.core.query.NativeSearchQueryBuilder;
  14.  
  15. import java.util.ArrayList;
  16. import java.util.List;
  17.  
  18. /**
  19.  * @ClassName EsTest
  20.  * @Description TODO
  21.  * @Author Innocence
  22.  * @Date 2021/6/16 14:05
  23.  * @Version 1.0
  24.  */
  25. @SpringBootTest
  26. public class EsTest {
  27.  
  28.     @Autowired
  29.     private IndexService<EsTestInfo> indexService;
  30.  
  31.     @Autowired
  32.     private DocumentService<EsTestInfo> documentService;
  33.  
  34.     @Test
  35.     public void indexOperation(){
  36.         Boolean index = indexService.createIndexByClass(EsTestInfo.class);
  37.         System.out.println("index = " + index);
  38.     }
  39.  
  40.     @Test
  41.     public void saveByEntity(){
  42.         EsTestInfo esTestInfo = new EsTestInfo();
  43.         esTestInfo.setAge(20);
  44.         esTestInfo.setInfoId("test_02");
  45.         esTestInfo.setName("一库一库");
  46.         esTestInfo.setSex("男");
  47.         String entity = null;
  48.         try {
  49.             entity = documentService.saveByEntity(esTestInfo);
  50.         } catch (Exception e) {
  51.             e.printStackTrace();
  52.         }
  53.         System.out.println("entity = " + entity);
  54.     }
  55.  
  56.     @Test
  57.     public void isExist(){
  58.         Boolean test_01 = documentService.isExist("test_01", EsTestInfo.class);
  59.         System.out.println("test_01 = " + test_01);
  60.     }
  61.  
  62.     @Test
  63.     public void updateByEntity(){
  64.         EsTestInfo info = documentService.getEntityById("test_01", EsTestInfo.class);
  65.         info.setName(info.getName()+"_update");
  66.         documentService.updateByEntity(info);
  67.     }
  68.  
  69.     @Test
  70.     public void deleteById(){
  71.         String deleteById = documentService.deleteById("test_01", EsTestInfo.class);
  72.         System.out.println("deleteById = " + deleteById);
  73.     }
  74.  
  75.     @Test
  76.     public void saveBatchByEntities(){
  77.         List<EsTestInfo> list = new ArrayList<>();
  78.         for (int i = 0; i < 10; i++) {
  79.             EsTestInfo info = new EsTestInfo();
  80.             info.setInfoId("test_batch_"+(i+1)+"");
  81.             info.setName("name_batch_"+(i+1)+"");
  82.             info.setSex("N");
  83.             info.setAge((i+1)*10);
  84.             info.setRoles(new String[]{"管理员","超级管理员"});
  85.             list.add(info);
  86.         }
  87.         try {
  88.             documentService.saveBatchByEntities(list);
  89.         } catch (Exception e) {
  90.             e.printStackTrace();
  91.         }
  92.     }
  93.  
  94.     @Test
  95.     public void getInfoList(){
  96.         NativeSearchQueryBuilder queryBuilder = new NativeSearchQueryBuilder();
  97.         BoolQueryBuilder boolQuery = QueryBuilders.boolQuery()
  98.                 .must(QueryBuilders.wildcardQuery("infoId","test_batch_*"));
  99.         NativeSearchQuery build = queryBuilder.withQuery(boolQuery).build();
  100.         List<EsTestInfo> infoList = documentService.getInfoList(build,EsTestInfo.class,true);
  101.     }
  102.  
  103.     @Test
  104.     public void count(){
  105.         Long count = documentService.getCount(new NativeSearchQueryBuilder().build(),EsTestInfo.class);
  106.         System.out.println("count = " + count);
  107.     }
  108.  
  109.     @Test
  110.     public void deleteByIds(){
  111.         ArrayList<String> strings = new ArrayList<>();
  112.         strings.add("test_01");
  113.         strings.add("test_02");
  114.         Class<EsTestInfo> clazz = EsTestInfo.class;
  115.         documentService.deleteByIds(strings, clazz);
  116.     }
  117.  
  118.     @Test
  119.     public void update() {
  120.         EsTestInfo info = new EsTestInfo();
  121.         info.setName("wait_for");
  122.         info.setInfoId("test_batch_wait_for_2");
  123.         documentService.saveByEntity(info);
  124.         NativeSearchQueryBuilder queryBuilder = new NativeSearchQueryBuilder();
  125.         BoolQueryBuilder boolQuery = QueryBuilders.boolQuery()
  126.                 .must(QueryBuilders.wildcardQuery("infoId","test_batch_*"));
  127.         NativeSearchQuery build = queryBuilder.withQuery(boolQuery).build();
  128.         List<EsTestInfo> infoList = documentService.getInfoList(build,EsTestInfo.class);
  129.         System.out.println("infoList = " + infoList);
  130.     }
  131. }

 

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/从前慢现在也慢/article/detail/325740
推荐阅读
相关标签

Copyright © 2003-2013 www.wpsshop.cn 版权所有，并保留所有权利。

  

闽ICP备14008679号