elasticsearch query related

最新推荐文章于 2025-01-07 10:15:13 发布
c_zyer
最新推荐文章于 2025-01-07 10:15:13 发布
阅读量544
点赞数
分类专栏： ElasticSearch 文章标签： java elasticsearch
ElasticSearch 专栏收录该内容
10 篇文章
订阅专栏
本文介绍了一个针对Elasticsearch实现的抽象基类EsBaseDaoImpl，该类提供了包括增删改查在内的多种操作方法，并支持批量操作、条件查询等功能。通过具体的代码示例，展示了如何使用此基类来简化Elasticsearch的应用开发。
摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >
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();  
      
  
  
      
  
}