spring 操作elasticsearch查詢(xún)使用方法
更新時(shí)間：2017年05月20日 15:41:43 作者：GOODDEEP
本篇文章主要介紹了spring 操作elasticsearch使用方法，小編覺(jué)得挺不錯(cuò)的，現(xiàn)在分享給大家，也給大家做個(gè)參考。一起跟隨小編過(guò)來(lái)看看吧
最近學(xué)習(xí)了一下elasticsearch使用，網(wǎng)上的資料又很少，真是一個(gè)頭兩個(gè)大。好歹最后終于了解了。留個(gè)筆記做日后查詢(xún)。
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ù)條件查詢(xún)
* @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()) {//字段查詢(xún)
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;
}
}
/**
* 條件查詢(xún)
*
* @param searchfields
* 查詢(xún)字段
* @param filedContentMap
* 字段和查詢(xún)內(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)//分頁(yè)
nsb.withPageable(new PageRequest(basePage.getPageNo(), basePage.getPageSize()));
BoolQueryBuilder qb=QueryBuilders. boolQuery();
for (String key : filedContentMap.keySet()) {//字段查詢(xún)
qb.must(QueryBuilders.matchQuery(key,filedContentMap.get(key)));
}
//userKey=78e48b85e94911e0d285f4eec990d556
//fa6e9c5bb24a21807c59e5fd3b609e12
nsb.withQuery(qb);
SearchQuery searchQuery = nsb.build();//查詢(xún)建立
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()) {//字段查詢(xún)
qb.must(QueryBuilders.matchQuery(key,filedContentMap.get(key)));
}
nsb.withQuery(qb);
SearchQuery searchQuery = nsb.build();//查詢(xún)建立
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();
}
/**
* 本類(lèi)查詢(xún)
*
* @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;
}
/**
* 得到類(lèi)型
*
* @return
*/
public abstract Class<T> getEntityClass();
/**
* 添加各自類(lèi)的影射
*/
public abstract void putClassMapping();
}
以上就是本文的全部?jī)?nèi)容，希望對(duì)大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。
相關(guān)文章
Sentinel結(jié)合Nacos實(shí)現(xiàn)數(shù)據(jù)持久化過(guò)程詳解
這篇文章主要介紹了Sentinel結(jié)合Nacos實(shí)現(xiàn)數(shù)據(jù)持久化過(guò)程，要持久化的原因是因?yàn)槊看螁?dòng)Sentinel都會(huì)使之前配置的規(guī)則就清空了，這樣每次都要再去設(shè)定規(guī)則顯得非常的麻煩，感興趣想要詳細(xì)了解可以參考下文2023-05-05
RestTemplate未使用線(xiàn)程池問(wèn)題的解決方法
今天給大家?guī)?lái)的是關(guān)于Springboot的相關(guān)知識(shí),文章圍繞著RestTemplate未使用線(xiàn)程池展開(kāi),文中有非常詳細(xì)的介紹及代碼示例,需要的朋友可以參考下2021-06-06
五分鐘解鎖springboot admin監(jiān)控新技巧
本文不會(huì)講如何搭建企業(yè)的運(yùn)維監(jiān)控系統(tǒng)，有興趣的可以去找找成熟的比如Zabbix、Prometheus,甚至比較簡(jiǎn)單的Wgcloud都能滿(mǎn)足一定的需求，不在此贅述。本文講解如何使用Springboot admin對(duì)spring boot項(xiàng)目進(jìn)行應(yīng)用監(jiān)控，感興趣的朋友一起看看吧2021-06-06
Spring Boot 整合 MyBatis 連接數(shù)據(jù)庫(kù)及常見(jiàn)問(wèn)題
MyBatis 是一個(gè)優(yōu)秀的持久層框架,支持定制化 SQL、存儲(chǔ)過(guò)程以及高級(jí)映射,下面詳細(xì)介紹如何在 Spring Boot 項(xiàng)目中整合 MyBatis 并連接數(shù)據(jù)庫(kù),感興趣的朋友一起看看吧2025-03-03
Mybatis plus中使用in查詢(xún)出錯(cuò)如何解決
這篇文章主要介紹了Mybatis plus中使用in查詢(xún)出錯(cuò)的問(wèn)題及解決方法，本文給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2020-08-08
這篇文章主要介紹了JAVA破壞單例模式的方式以及避免方法，文中講解非常細(xì)致，代碼幫助大家更好的理解和學(xué)習(xí)，感興趣的朋友可以了解下2020-06-06
用C++實(shí)現(xiàn)求N!中末尾0的個(gè)數(shù)的方法詳解
這篇文章主要介紹了用C++實(shí)現(xiàn)求N!中末尾0的個(gè)數(shù)的方法詳解，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2020-07-07
Spring框架接入單機(jī)Redis兩種實(shí)現(xiàn)方式解析
這篇文章主要介紹了Spring框架接入單機(jī)Redis兩種實(shí)現(xiàn)方式解析,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2020-09-09
這篇文章主要為大家介紹了kafka序列化器和攔截器使用示例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2023-03-03
java多線(xiàn)程加鎖以及Condition類(lèi)的使用實(shí)例解析
這篇文章主要介紹了java多線(xiàn)程加鎖以及Condition類(lèi)的使用實(shí)例解析,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2019-11-11