common.search

package com.yueluo.mall.search.utils; import com.alibaba.fastjson.JSON; import com.alibaba.fastjson.JSONObject; import com.yueluo.mall.common.dto.es.esdata.EsBaseData; import com.yueluo.mall.search.esdata.EsField; import com.yueluo.mall.search.esdata.EsPage; import com.yueluo.mall.search.exception.BusinessException; import lombok.extern.slf4j.Slf4j; import org.elasticsearch.action.DocWriteResponse; import org.elasticsearch.action.admin.indices.delete.DeleteIndexRequest; import org.elasticsearch.action.bulk.BulkRequest; import org.elasticsearch.action.bulk.BulkResponse; import org.elasticsearch.action.delete.DeleteRequest; import org.elasticsearch.action.delete.DeleteResponse; import org.elasticsearch.action.get.GetRequest; import org.elasticsearch.action.get.GetResponse; import org.elasticsearch.action.index.IndexRequest; import org.elasticsearch.action.index.IndexResponse; import org.elasticsearch.action.search.SearchRequest; import org.elasticsearch.action.search.SearchResponse; import org.elasticsearch.action.support.WriteRequest; import org.elasticsearch.action.support.master.AcknowledgedResponse; import org.elasticsearch.action.update.UpdateRequest; import org.elasticsearch.action.update.UpdateResponse; import org.elasticsearch.client.IndicesClient; import org.elasticsearch.client.RequestOptions; import org.elasticsearch.client.RestHighLevelClient; import org.elasticsearch.client.indices.CreateIndexRequest; import org.elasticsearch.client.indices.CreateIndexResponse; import org.elasticsearch.common.unit.Fuzziness; import org.elasticsearch.common.unit.TimeValue; import org.elasticsearch.common.xcontent.XContentType; import org.elasticsearch.index.query.QueryBuilders; import org.elasticsearch.index.reindex.BulkByScrollResponse; import org.elasticsearch.index.reindex.DeleteByQueryRequest; import org.elasticsearch.index.reindex.UpdateByQueryRequest; import org.elasticsearch.script.Script; import org.elasticsearch.search.SearchHit; import org.elasticsearch.search.SearchHits; import org.elasticsearch.search.builder.SearchSourceBuilder; import org.elasticsearch.search.sort.SortOrder; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Component; import org.springframework.util.Assert; import org.springframework.util.CollectionUtils; import java.io.IOException; import java.util.ArrayList; import java.util.List; import java.util.Map; /** * es工具类型 * @author xjx * @email 15340655443@163.com * @date 2024/1/22 14:16 */ @Component @Slf4j public class ElasticsearchUtils { @Autowired private RestHighLevelClient restHighLevelClient; /** * 添加索引 * * @param indexName * @return * @throws IOException */ public boolean addIndex(String indexName) { Assert.hasLength(indexName, "Elasticsearch exception indexName null"); CreateIndexResponse createIndexResponse = null; try { //1.使用client获取操作索引对象 IndicesClient indices = restHighLevelClient.indices(); //2.具体操作获取返回值 //2.1 设置索引名称 CreateIndexRequest createIndexRequest = new CreateIndexRequest(indexName); createIndexResponse = indices.create(createIndexRequest, RequestOptions.DEFAULT); } catch (Exception e) { // e.printStackTrace(); log.error("elasticsearch addindex error , meassage = {}", e.getMessage()); //打印轨迹 log.error(e.getMessage(), e); //抛出异常 throw new BusinessException("elasticsearch addindex error , meassage=" + e.getMessage()); } //3.根据返回值判断结果 return createIndexResponse.isAcknowledged(); } /** * 删除索引 * * @param indexName * @return * @throws IOException */ public boolean deleteIndex(String indexName) { Assert.hasLength(indexName, "Elasticsearch exception indexName null"); AcknowledgedResponse deleteRespone = null; try { //1.使用client获取操作索引对象 IndicesClient indices = restHighLevelClient.indices(); //2.具体操作获取返回值 //2.1 设置索引名称 DeleteIndexRequest request = new DeleteIndexRequest("twitter_two");//指定要删除的索引名称 deleteRespone = indices.delete(request, RequestOptions.DEFAULT); } catch (Exception e) { // e.printStackTrace(); log.error("elasticsearch deleteIndex error , meassage = {}", e.getMessage()); //打印轨迹 log.error(e.getMessage(), e); throw new BusinessException("elasticsearch deleteIndex error , meassage=" + e.getMessage()); } //3.根据返回值判断结果 return deleteRespone.isAcknowledged(); } /** * 创建数据 * * @param indexName * @param id * @param data * @throws IOException */ public boolean addData(String indexName, String id, Object data) { Assert.notNull(data, "Elasticsearch exception data null"); Assert.hasLength(indexName, "Elasticsearch exception indexName null"); IndexResponse indexResponse = null; try { //准备文档 String jsonString = JSONObject.toJSONString(data); Map jsonMap = JSONObject.parseObject(jsonString, Map.class); //创建请求 IndexRequest indexRequest = new IndexRequest(indexName).id(id); //指定文档内容 indexRequest.source(jsonMap); //true 当存在相同的_id时，插入会出现异常； false 当存在相同_id时，插入会进行覆盖； indexRequest.create(true); //通过client进行http请求 indexResponse = restHighLevelClient.index(indexRequest, RequestOptions.DEFAULT); } catch (Exception e) { log.error("elasticsearch addOrUpdateDoc error , meassage = {}", e.getMessage()); //打印轨迹 log.error(e.getMessage(), e); throw new BusinessException("elasticsearch addOrUpdateDoc error , meassage=" + e.getMessage()); } return indexResponse.getResult().equals(DocWriteResponse.Result.CREATED); } /** * 创建文档id存在则更新文档 * * @param indexName * @param id * @param data * @throws IOException */ public boolean addOrUpdateData(String indexName, String id, Object data) { Assert.notNull(data, "Elasticsearch exception data null"); Assert.hasLength(indexName, "Elasticsearch exception indexName null"); IndexResponse indexResponse = null; try { //准备文档 String jsonString = JSONObject.toJSONString(data); Map jsonMap = JSONObject.parseObject(jsonString, Map.class); //创建请求 IndexRequest indexRequest = new IndexRequest(indexName).id(id); //指定文档内容 indexRequest.source(jsonMap); //通过client进行http请求 indexResponse = restHighLevelClient.index(indexRequest, RequestOptions.DEFAULT); } catch (Exception e) { log.error("elasticsearch addOrUpdateDoc error , meassage = {}", e.getMessage()); //打印轨迹 log.error(e.getMessage(), e); throw new BusinessException("elasticsearch addOrUpdateDoc error , meassage=" + e.getMessage()); } return indexResponse.getResult().equals(DocWriteResponse.Result.CREATED); } /** * 单条更新 * * @param indexName * @param id * @param data * @return