GOODDEEP 通过本文主要向大家介绍了spring elasticsearch,elastic spring,elasticsearch使用,spring opensearch,spring boot使用等相关知识,希望对您有所帮助,也希望大家支持linkedu.com www.linkedu.com
最近学习了一下elasticsearch使用，网上的资料又很少，真是一个头两个大。好歹最后终于了解了。留个笔记做日后查询。
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;
}
}
/**
* 根据条件查询
* @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()