spring 操作elasticsearch查詢使用方法
最近學(xué)習(xí)了一下elasticsearch使用,網(wǎng)上的資料又很少,真是一個(gè)頭兩個(gè)大。好歹最后終于了解了。留個(gè)筆記做日后查詢。
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; } } /** * 根據(jù)條件查詢 * @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; } } /** * 檢查健康狀態(tài) * * @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 * 字段和查詢內(nèi)容 * @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) {//如果設(shè)置高亮 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//如果不設(shè)置高亮 { 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) {//如果設(shè)置高亮 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//如果不設(shè)置高亮 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(); }
以上就是本文的全部內(nèi)容,希望對大家的學(xué)習(xí)有所幫助,也希望大家多多支持腳本之家。
相關(guān)文章
解決Java?結(jié)構(gòu)化數(shù)據(jù)處理開源庫?SPL的問題
這篇文章主要介紹了Java?結(jié)構(gòu)化數(shù)據(jù)處理開源庫?SPL的問題,Scala提供了較豐富的結(jié)構(gòu)化數(shù)據(jù)計(jì)算函數(shù),但編譯型語言的特點(diǎn),也使它不能成為理想的結(jié)構(gòu)化數(shù)據(jù)計(jì)算類庫,對此內(nèi)容感興趣的朋友一起看看吧2022-03-03java+mysql實(shí)現(xiàn)圖書館管理系統(tǒng)實(shí)戰(zhàn)
這篇文章主要為大家詳細(xì)介紹了java+mysql實(shí)現(xiàn)圖書館管理系統(tǒng)實(shí)戰(zhàn),文中示例代碼介紹的非常詳細(xì),具有一定的參考價(jià)值,感興趣的小伙伴們可以參考一下2020-12-12基于Spring?Boot的線程池監(jiān)控問題及解決方案
這篇文章主要介紹了基于Spring?Boot的線程池監(jiān)控方案,本文給大家介紹的非常詳細(xì),對大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2022-03-03IDEA下使用Spring Boot熱加載的實(shí)現(xiàn)
本文主要介紹了IDEA下使用Spring Boot熱加載的實(shí)現(xiàn),文中通過示例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2023-06-06SpringBoot2整合Drools規(guī)則引擎及案例詳解
這篇文章主要介紹了SpringBoot2整合Drools規(guī)則引擎及案例詳解,文中通過示例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2019-10-10Spring Boot的listener(監(jiān)聽器)簡單使用實(shí)例詳解
監(jiān)聽器(Listener)的注冊方法和 Servlet 一樣,有兩種方式:代碼注冊或者注解注冊。接下來通過本文給大家介紹Spring Boot的listener(監(jiān)聽器)簡單使用,需要的朋友可以參考下2017-04-04SpringBoot整合JDBC的實(shí)現(xiàn)
這篇文章主要介紹了SpringBoot整合JDBC的實(shí)現(xiàn),文中通過示例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2021-01-01