Elasticsearch6学习笔记——RestHighLevelClient连接

需要的JAR包

        <dependency>
			<groupId>org.elasticsearch</groupId>
			<artifactId>elasticsearch</artifactId>
			<version>6.4.2</version>
		</dependency>
		<dependency>
			<groupId>org.elasticsearch.client</groupId>
			<artifactId>elasticsearch-rest-high-level-client</artifactId>
			<version>6.4.2</version>
		</dependency>
		<dependency>
			<groupId>org.elasticsearch.client</groupId>
			<artifactId>elasticsearch-rest-client</artifactId>
			<version>6.4.2</version>
		</dependency>

实体类

import javax.persistence.Column;
import javax.persistence.Entity;
import javax.persistence.GeneratedValue;
import javax.persistence.Id;
import javax.persistence.Table;

import org.hibernate.annotations.GenericGenerator;

/**
 * @Description: 物料视图
 */
@Entity
@Table(name = "material_view", schema = "")
@SuppressWarnings("serial")
public class MaterialView implements java.io.Serializable {
	
	/** 主键 */
	private java.lang.String id;
	/** 物料名称 */
	private java.lang.String equipmentName;
	/** 品牌名称 */
	private java.lang.String brandName;
	private java.lang.String brandCode;
	/** 供应商 */
	private java.lang.String supplierName;
	private java.lang.String supplierId;
	/** 型号 */
	private java.lang.String model;
	/** 规格 */
	private java.lang.String type;
	/** 详细信息 */
	private java.lang.String paramInfo;
	/** 一级分类 */
	private java.lang.String levelOneName;
	/** 二级分类 */
	private java.lang.String levelTwoName;
	/** 三级分类 */
	private java.lang.String levelThreeName;
	private java.lang.String levelOneCode;
	private java.lang.String levelTwoCode;
	private java.lang.String levelThreeCode;
	private java.lang.Integer createDate;
	/** 市场价--面价 */
	private Double facePrice;
	/** 设计成本价 */
	private Double designCost;
	/** 采购成本价 */
	private Double purchaseCost;
	/** 备注 */
	private java.lang.String remake;
	/** 图片 */
	private java.lang.String img;
	
	public MaterialView() {
		// TODO Auto-generated constructor stub
	}
	
	public MaterialView(String id, String equipmentName, String brandName, String brandCode, String supplierName,
			String supplierId, String model, String type, String paramInfo, String levelOneName, String levelTwoName,
			String levelThreeName, String levelOneCode, String levelTwoCode, String levelThreeCode, Integer createDate,
			Double facePrice, Double designCost, Double purchaseCost, String remake, String img) {
		super();
		this.id = id;
		this.equipmentName = equipmentName;
		this.brandName = brandName;
		this.brandCode = brandCode;
		this.supplierName = supplierName;
		this.supplierId = supplierId;
		this.model = model;
		this.type = type;
		this.paramInfo = paramInfo;
		this.levelOneName = levelOneName;
		this.levelTwoName = levelTwoName;
		this.levelThreeName = levelThreeName;
		this.levelOneCode = levelOneCode;
		this.levelTwoCode = levelTwoCode;
		this.levelThreeCode = levelThreeCode;
		this.createDate = createDate;
		this.facePrice = facePrice;
		this.designCost = designCost;
		this.purchaseCost = purchaseCost;
		this.remake = remake;
		this.img = img;
	}

	/**
	 * 方法: 取得java.lang.String
	 * 
	 * @return: java.lang.String 主键
	 */
	@Id
	@GeneratedValue(generator = "paymentableGenerator")
	@GenericGenerator(name = "paymentableGenerator", strategy = "uuid")
	@Column(name = "ID", nullable = false, length = 36)
	public java.lang.String getId() {
		return this.id;
	}

	/**
	 * 方法: 设置java.lang.String
	 * 
	 * @param: java.lang.String 主键
	 */
	public void setId(java.lang.String id) {
		this.id = id;
	}

	@Column(name = "equipmentName", nullable = true)
	public java.lang.String getEquipmentName() {
		return equipmentName;
	}

	public void setEquipmentName(java.lang.String equipmentName) {
		this.equipmentName = equipmentName;
	}

	@Column(name = "brandName", nullable = true)
	public java.lang.String getBrandName() {
		return brandName;
	}

	public void setBrandName(java.lang.String brandName) {
		this.brandName = brandName;
	}

	@Column(name = "model", nullable = true)
	public java.lang.String getModel() {
		return model;
	}

	public void setModel(java.lang.String model) {
		this.model = model;
	}

	@Column(name = "type", nullable = true)
	public java.lang.String getType() {
		return type;
	}

	public void setType(java.lang.String type) {
		this.type = type;
	}

	@Column(name = "paramInfo", nullable = true)
	public java.lang.String getParamInfo() {
		return paramInfo;
	}

	public void setParamInfo(java.lang.String paramInfo) {
		this.paramInfo = paramInfo;
	}

	@Column(name = "levelOneName", nullable = true)
	public java.lang.String getLevelOneName() {
		return levelOneName;
	}

	public void setLevelOneName(java.lang.String levelOneName) {
		this.levelOneName = levelOneName;
	}

	@Column(name = "levelTwoName", nullable = true)
	public java.lang.String getLevelTwoName() {
		return levelTwoName;
	}

	public void setLevelTwoName(java.lang.String levelTwoName) {
		this.levelTwoName = levelTwoName;
	}

	@Column(name = "levelThreeName", nullable = true)
	public java.lang.String getLevelThreeName() {
		return levelThreeName;
	}

	public void setLevelThreeName(java.lang.String levelThreeName) {
		this.levelThreeName = levelThreeName;
	}

	@Column(name = "brandCode", nullable = true)
	public java.lang.String getBrandCode() {
		return brandCode;
	}

	public void setBrandCode(java.lang.String brandCode) {
		this.brandCode = brandCode;
	}

	@Column(name = "levelOneCode", nullable = true)
	public java.lang.String getLevelOneCode() {
		return levelOneCode;
	}

	public void setLevelOneCode(java.lang.String levelOneCode) {
		this.levelOneCode = levelOneCode;
	}

	@Column(name = "levelTwoCode", nullable = true)
	public java.lang.String getLevelTwoCode() {
		return levelTwoCode;
	}

	public void setLevelTwoCode(java.lang.String levelTwoCode) {
		this.levelTwoCode = levelTwoCode;
	}

	@Column(name = "levelThreeCode", nullable = true)
	public java.lang.String getLevelThreeCode() {
		return levelThreeCode;
	}

	public void setLevelThreeCode(java.lang.String levelThreeCode) {
		this.levelThreeCode = levelThreeCode;
	}

	@Column(name = "supplierName", nullable = true)
	public java.lang.String getSupplierName() {
		return supplierName;
	}

	public void setSupplierName(java.lang.String supplierName) {
		this.supplierName = supplierName;
	}

	@Column(name = "createDate", nullable = true)
	public Integer getCreateDate() {
		return createDate;
	}

	public void setCreateDate(Integer createDate) {
		this.createDate = createDate;
	}

	@Column(name = "supplierId", nullable = true)
	public java.lang.String getSupplierId() {
		return supplierId;
	}

	public void setSupplierId(java.lang.String supplierId) {
		this.supplierId = supplierId;
	}

	@Column(name = "facePrice", nullable = true)
	public Double getFacePrice() {
		return facePrice;
	}

	public void setFacePrice(Double facePrice) {
		this.facePrice = facePrice;
	}

	@Column(name = "designCost", nullable = true)
	public Double getDesignCost() {
		return designCost;
	}

	public void setDesignCost(Double designCost) {
		this.designCost = designCost;
	}

	@Column(name = "purchaseCost", nullable = true)
	public Double getPurchaseCost() {
		return purchaseCost;
	}

	public void setPurchaseCost(Double purchaseCost) {
		this.purchaseCost = purchaseCost;
	}

	@Column(name = "remake", nullable = true)
	public java.lang.String getRemake() {
		return remake;
	}

	public void setRemake(java.lang.String remake) {
		this.remake = remake;
	}

	@Column(name = "img", nullable = true)
	public java.lang.String getImg() {
		return img;
	}

	public void setImg(java.lang.String img) {
		this.img = img;
	}
	
}

获取连接的工具类

import org.apache.http.HttpHost;
import org.elasticsearch.client.RestClient;
import org.elasticsearch.client.RestHighLevelClient;

/**
 * @author jins
 * 需要时获取用完需要关闭 .close()
 *
 */
public class ESRestClientUtil {

	//集群名称
	//private static final String CLUSTERNAME = "es-cluster";
	//ES服务器地址
	private static final String IP = "192.168.0.119";
	//ES服务器端口--elastic search默认tcp端口9300,http端口9200
	private static final Integer PORT = 9200;

	public static RestHighLevelClient getClient() {
		RestHighLevelClient client = null;
		try {
			client = new RestHighLevelClient(
					RestClient.builder(
							new HttpHost(IP, PORT, "http")
							)
					);
		} catch (Exception e) {
			e.printStackTrace();
		}
		return client;
	}

}

CRUD操作


import java.io.IOException;
import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.apache.commons.lang3.reflect.FieldUtils;
import org.elasticsearch.action.admin.indices.create.CreateIndexRequest;
import org.elasticsearch.action.admin.indices.create.CreateIndexResponse;
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.ClearScrollRequest;
import org.elasticsearch.action.search.ClearScrollResponse;
import org.elasticsearch.action.search.SearchRequest;
import org.elasticsearch.action.search.SearchResponse;
import org.elasticsearch.action.search.SearchScrollRequest;
import org.elasticsearch.action.update.UpdateRequest;
import org.elasticsearch.action.update.UpdateResponse;
import org.elasticsearch.client.RequestOptions;
import org.elasticsearch.client.RestHighLevelClient;
import org.elasticsearch.common.settings.Settings;
import org.elasticsearch.common.text.Text;
import org.elasticsearch.common.unit.TimeValue;
import org.elasticsearch.common.xcontent.XContentBuilder;
import org.elasticsearch.common.xcontent.XContentFactory;
import org.elasticsearch.index.get.GetResult;
import org.elasticsearch.index.query.BoolQueryBuilder;
import org.elasticsearch.index.query.MultiMatchQueryBuilder;
import org.elasticsearch.index.query.MultiMatchQueryBuilder.Type;
import org.elasticsearch.index.query.Operator;
import org.elasticsearch.index.query.QueryBuilder;
import org.elasticsearch.index.query.QueryBuilders;
import org.elasticsearch.search.Scroll;
import org.elasticsearch.search.SearchHit;
import org.elasticsearch.search.SearchHits;
import org.elasticsearch.search.builder.SearchSourceBuilder;
import org.elasticsearch.search.fetch.subphase.highlight.HighlightBuilder;
import org.elasticsearch.search.fetch.subphase.highlight.HighlightField;
import org.elasticsearch.search.sort.FieldSortBuilder;
import org.elasticsearch.search.sort.ScoreSortBuilder;
import org.elasticsearch.search.sort.SortOrder;
import org.jeecgframework.core.common.model.json.AjaxJson;
import org.jeecgframework.core.util.PinyinUtil;
import org.jeecgframework.core.util.StringUtil;

import com.alibaba.fastjson.JSONObject;
import com.jeecg.yw.entity.MaterialView;

import cn.hutool.json.JSONUtil;

/**
 * 	单个操作
 * @author jins
 *
 */
public class ESRestMaterialUtil {

	//索引名称--数据库名
	private static final String INDEX_NAME = "cos";
	//类型名称--表名
	private static final String TYPE_NAME = "material";

	/**
	 * 	创建index
	 *
	 * @return
	 * @throws Exception 
	 */
	public static AjaxJson createIndex() throws Exception {
		AjaxJson j = new AjaxJson();

		RestHighLevelClient client = ESRestClientUtil.getClient();

		XContentBuilder builder = XContentFactory.jsonBuilder()
				.startObject()
				.startObject("properties")
				.startObject("id")
				.field("type", "text")
				.field("store", true)
				.endObject()
				.startObject("equipmentName")
				.field("type", "text")
				.field("store", true)
				.field("analyzer", "ik_max_word")
				.field("search_analyzer", "ik_max_word")
				.endObject()
				.startObject("model")
				.field("type", "text")
				.field("store", true)
				.field("analyzer", "ik_max_word")
				.field("search_analyzer", "ik_max_word")
				.endObject()
				.startObject("type")
				.field("type", "text")
				.field("store", true)
				.field("analyzer", "ik_max_word")
				.field("search_analyzer", "ik_max_word")
				.endObject()
				.startObject("brandName")
				.field("type", "text")
				.field("store", true)
				.field("analyzer", "ik_max_word")
				.field("search_analyzer", "ik_max_word")
				.endObject()
				.startObject("brandCode")
				.field("type", "text")
				.field("store", true)
				.endObject()
				.startObject("supplierName")
				.field("type", "text")
				.field("store", true)
				.field("analyzer", "ik_max_word")
				.field("search_analyzer", "ik_max_word")
				.endObject()
				.startObject("supplierId")
				.field("type", "text")
				.field("store", true)
				.endObject()
				.startObject("levelOneName")
				.field("type", "text")
				.field("store", true)
				.field("analyzer", "ik_max_word")
				.field("search_analyzer", "ik_max_word")
				.endObject()
				.startObject("levelTwoName")
				.field("type", "text")
				.field("store", true)
				.field("analyzer", "ik_max_word")
				.field("search_analyzer", "ik_max_word")
				.endObject()
				.startObject("levelThreeName")
				.field("type", "text")
				.field("store", true)
				.field("analyzer", "ik_max_word")
				.field("search_analyzer", "ik_max_word")
				.endObject()
				.startObject("levelOneCode")
				.field("type", "text")
				.field("store", true)
				.endObject()
				.startObject("levelTwoCode")
				.field("type", "text")
				.field("store", true)
				.endObject()
				.startObject("levelThreeCode")
				.field("type", "text")
				.field("store", true)
				.endObject()
				.startObject("paramInfo")
				.field("type", "text")
				.field("store", true)
				.field("analyzer", "ik_max_word")
				.field("search_analyzer", "ik_max_word")
				.endObject()
				.startObject("createDate")
				.field("type", "integer")
				.field("store", true)
				.endObject()
				.startObject("remake")
				.field("type", "text")
				.field("store", true)
				.field("analyzer", "ik_max_word")
				.field("search_analyzer", "ik_max_word")
				.endObject()
				.startObject("facePrice")
				.field("type", "double")
				.field("store", true)
				.endObject()
				.startObject("designCost")
				.field("type", "double")
				.field("store", true)
				.endObject()
				.startObject("purchaseCost")
				.field("type", "double")
				.field("store", true)
				.endObject()
				.startObject("img")
				.field("type", "text")
				.field("store", true)
				.endObject()
				.endObject()
				.endObject();

		//创建索引请求
		CreateIndexRequest request = new CreateIndexRequest(INDEX_NAME);
		//设置分片和副本数
		request.settings(Settings.builder()
				.put("number_of_shards", 3)
				.put("number_of_replicas", 0)
				//修改查询最大条数
				.put("max_result_window", 10000000)
				);
		//创建映射,中间填写需要存到ES上的JavaDTO对象对应的JSON数据。
		request.mapping(TYPE_NAME, builder);
		//创建索引
		CreateIndexResponse createIndexResponse = client.indices().create(request, RequestOptions.DEFAULT);

		System.err.println("--------->"+createIndexResponse.isAcknowledged());
		
		client.close();

		j.setObj(null);
		return j;
	}

	/**
	 * 	同步物料信息
	 *
	 * @return
	 * @throws Exception 
	 */
	public static AjaxJson synchronizeData(List<MaterialView> list) throws Exception {
		AjaxJson j = new AjaxJson();
		j.setMsg("同步物料数量:"+doAddList(list));
		return j;
	}
	
	/**
	 * 增,插入记录
	 * 插入操作有四种方式,分同步异步操作,可选参数设置,结果返回IndexResponse,抛出异常
	 * @return 
	 * @return 
	 * @throws Exception
	 */
	public static int doAddList(List<MaterialView> list) throws Exception{

		RestHighLevelClient client = ESRestClientUtil.getClient();

		//第三种方式: XContentBuilder automatically converted to JSON
		
		int num = 0;
		for (MaterialView m : list) {

			XContentBuilder builder = XContentFactory.jsonBuilder().startObject()
					.field("brandCode", m.getBrandCode())
					.field("brandName", m.getBrandName())
					.field("equipmentName", m.getEquipmentName())
					.field("id", m.getId())
					.field("levelOneName", m.getLevelOneName())
					.field("levelOneCode", m.getLevelOneCode())
					.field("levelThreeName", m.getLevelThreeName())
					.field("levelThreeCode", m.getLevelThreeCode())
					.field("levelTwoName", m.getLevelTwoName())
					.field("levelTwoCode", m.getLevelTwoCode())
					.field("model", m.getModel())
					.field("paramInfo", m.getParamInfo())
					.field("supplierName", m.getSupplierName())
					.field("supplierId", m.getSupplierId())
					.field("type", m.getType())
					.field("remake", m.getRemake())
					.field("facePrice", m.getFacePrice())
					.field("designCost", m.getDesignCost())
					.field("purchaseCost", m.getPurchaseCost())
					.field("img", m.getImg())
					.field("createDate", m.getCreateDate())
					.endObject();

			IndexRequest indexRequest1 = new IndexRequest(INDEX_NAME, TYPE_NAME, m.getId())
					.source(builder);

			//同步执行
			IndexResponse indexResponse = client.index(indexRequest1, RequestOptions.DEFAULT);

			System.err.println(num+++"-------response.status()----->"+indexResponse.status());

		}
		client.close();
		
		return num;
	}

	/**
	 * 增,插入记录
	 * 插入操作有四种方式,分同步异步操作,可选参数设置,结果返回IndexResponse,抛出异常
	 * @return 
	 * @throws Exception
	 */
	public static String doAdd(MaterialView m) throws Exception{

		RestHighLevelClient client = ESRestClientUtil.getClient();

		//第三种方式: XContentBuilder automatically converted to JSON

		XContentBuilder builder = XContentFactory.jsonBuilder().startObject()
				.field("brandCode", m.getBrandCode())
				.field("brandName", m.getBrandName())
				.field("equipmentName", m.getEquipmentName())
				.field("id", m.getId())
				.field("levelOneName", m.getLevelOneName())
				.field("levelOneCode", m.getLevelOneCode())
				.field("levelThreeName", m.getLevelThreeName())
				.field("levelThreeCode", m.getLevelThreeCode())
				.field("levelTwoName", m.getLevelTwoName())
				.field("levelTwoCode", m.getLevelTwoCode())
				.field("model", m.getModel())
				.field("paramInfo", m.getParamInfo())
				.field("supplierName", m.getSupplierName())
				.field("supplierId", m.getSupplierId())
				.field("type", m.getType())
				.field("remake", m.getRemake())
				.field("facePrice", m.getFacePrice())
				.field("designCost", m.getDesignCost())
				.field("purchaseCost", m.getPurchaseCost())
				.field("img", m.getImg())
				.field("createDate", m.getCreateDate())
				.endObject();

		IndexRequest indexRequest1 = new IndexRequest(INDEX_NAME, TYPE_NAME, m.getId())
				.source(builder);

		//同步执行
		IndexResponse indexResponse = client.index(indexRequest1, RequestOptions.DEFAULT);

		System.err.println("response.status()----->"+indexResponse.status());

		client.close();
		return indexResponse.status().toString();
	}

	/**
	 * 根据 id 获取数据
	 * @throws Exception
	 */
	public static void get(String id) throws Exception{

		RestHighLevelClient client = ESRestClientUtil.getClient();

		GetRequest request = new GetRequest(INDEX_NAME, TYPE_NAME, id);

		//同步执行
		GetResponse getResponse = client.get(request, RequestOptions.DEFAULT);

		System.err.println("----->"+getResponse.getSourceAsString());

		client.close();

		//Get Response 获取信息

		//抛出异常
	}

	/**
	 * 根据id删除
	 * @throws Exception
	 */
	public static void doDelete(String id) throws Exception{

		RestHighLevelClient client = ESRestClientUtil.getClient();
		DeleteRequest request = new DeleteRequest(INDEX_NAME, TYPE_NAME, id);

		//同步执行
		DeleteResponse deleteResponse = client.delete(request, RequestOptions.DEFAULT);

		//Delete Response
		System.err.println("----->"+deleteResponse.status());

		client.close();

	}

	/**
	 * 根据id更新
	 * @throws Exception
	 */
	public static void doUpdate(MaterialView m) throws Exception{

		RestHighLevelClient client = ESRestClientUtil.getClient();
		System.err.println("m------------>"+JSONUtil.toJsonStr(m));
		UpdateRequest request = new UpdateRequest(INDEX_NAME, TYPE_NAME, m.getId());

		request.doc(
				XContentFactory.jsonBuilder().startObject()
				.field("brandCode", m.getBrandCode())
				.field("brandName", m.getBrandName())
				.field("equipmentName", m.getEquipmentName())
				.field("id", m.getId())
				.field("levelOneName", m.getLevelOneName())
				.field("levelOneCode", m.getLevelOneCode())
				.field("levelThreeName", m.getLevelThreeName())
				.field("levelThreeCode", m.getLevelThreeCode())
				.field("levelTwoName", m.getLevelTwoName())
				.field("levelTwoCode", m.getLevelTwoCode())
				.field("model", m.getModel())
				.field("paramInfo", m.getParamInfo())
				.field("supplierName", m.getSupplierName())
				.field("supplierId", m.getSupplierId())
				.field("type", m.getType())
				.field("remake", m.getRemake())
				.field("facePrice", m.getFacePrice())
				.field("designCost", m.getDesignCost())
				.field("purchaseCost", m.getPurchaseCost())
				.field("img", m.getImg())
				.field("createDate", m.getCreateDate())
				.endObject()
				);

		//同步执行
		UpdateResponse updateResponse = client.update(request, RequestOptions.DEFAULT);
		//update Response
		GetResult result = updateResponse.getGetResult();

		System.err.println("----->"+result);

		client.close();

	}

	/**
	 * 	分页获取数据
	 * @param page				分页
	 * @param keyword			搜索关键词-物料名称、详细参数、型号、规格、备注
	 * @param brandCodes		品牌CODES
	 * @param supplierIds		供应商IDS
	 * @param levelOneCode		一级分类CODE
	 * @param levelTwoCode		二级分类CODE
	 * @param levelThreeCode	三级分类CODE
	 * @throws IOException 
	 */

	public static PageUtil doPage(PageUtil page, String keyword, String brandCodes, String supplierIds,
			String levelOneCode, String levelTwoCode, String levelThreeCode) throws IOException {

		RestHighLevelClient client = ESRestClientUtil.getClient();
		//1,构建SearchRequest请求对象,指定索引库
		SearchRequest searchRequest = new SearchRequest(INDEX_NAME);
		//2,构建SearchSourceBuilder查询对象
		SearchSourceBuilder sourceBuilder = new SearchSourceBuilder();
		//3,构建QueryBuilder对象指定查询方式和查询条件
		BoolQueryBuilder query = setBoolQueryBuilder(keyword, brandCodes, supplierIds, levelOneCode, levelTwoCode, levelThreeCode);
		//4,将QuseryBuilder对象设置到SearchSourceBuilder对象中
		sourceBuilder.query(query);

		//字段过滤
		// sourceBuilder.fetchSource(new String[]{"title","price","band","category","id"},new String[]{"images"});
		//字段过滤
		sourceBuilder.fetchSource(
				new String[]{"brandName","equipmentName","id","levelOneName","levelTwoName","levelThreeName","model",
						"paramInfo","supplierName","type","remake","facePrice","designCost","purchaseCost","img"},
				new String[]{"brandCode", "levelOneCode", "levelTwoCode", "levelThreeCode", "supplierId", "createDate"}
				);
		//排序
		//sourceBuilder.sort("createDate", SortOrder.ASC);
		//Specifying Sorting 指定排序
		sourceBuilder.sort(new ScoreSortBuilder().order(SortOrder.DESC));
		sourceBuilder.sort(new FieldSortBuilder("_uid").order(SortOrder.DESC));

		int pageSize = page.getRows();
		int currentPageNo = page.getPage();

		List<MaterialView> result = new ArrayList<>(pageSize);

		//分页
		sourceBuilder.from((currentPageNo - 1) * pageSize);
		sourceBuilder.size(pageSize);

		//5,将SearchSourceBuilder设置到SearchRequest中
		searchRequest.source(sourceBuilder);

		// 设置高亮属性
		// 高亮设置
		HighlightBuilder highlightBuilder = new HighlightBuilder();
		highlightBuilder.requireFieldMatch(false)
		.field("equipmentName")
		.field("paramInfo")
		.field("model")
		.field("remake")
		.field("type")
		.field("supplierName")
		.field("brandName")
		//开启匹配指定字段高亮--默认false
		.requireFieldMatch(true)
		.preTags("<strong class=\"highlight\">").postTags("</strong>");

		sourceBuilder.highlighter(highlightBuilder);

		try {
			//6,调用方法查询数据
			SearchResponse searchResponse = client.search(searchRequest, RequestOptions.DEFAULT);

			// 获取查询总数量
			Long totalCount = searchResponse.getHits().getTotalHits();
			// 设置分页数据总数
			page.setTotal(totalCount.intValue());

			// 查询总数量为0,直接返回
			if (totalCount == 0) {
				page.setResults(result);
				return page;
			}
			// 获取分页结果
			SearchHits hits = searchResponse.getHits();
			MaterialView entityEsDto= null;
			// 遍历转换结果对象
			for (SearchHit searchHit : hits) {

				//entityEsDto = (MaterialView) ReflectUtils.newInstance(MaterialView.class);
				entityEsDto = JSONObject.parseObject(searchHit.getSourceAsString(), MaterialView.class);

				try {
					FieldUtils.writeField(entityEsDto, "id", searchHit.getId(), true);
				} catch (IllegalAccessException e) {
					e.printStackTrace();
				}

				// 非高亮字段的数据写入
				for (Map.Entry<String, Object> entry : searchHit.getSourceAsMap().entrySet()) {

					Field field = FieldUtils.getField(MaterialView.class, entry.getKey(), true);
					// 判断字段不存在
					if (null == field) {
						continue;
					}

					try {
						FieldUtils.writeField(entityEsDto, entry.getKey(), entry.getValue(), true);
					} catch (IllegalAccessException e) {
						e.printStackTrace();
					}
				}

				for (Map.Entry<String, HighlightField> entry : searchHit.getHighlightFields().entrySet()) {
					StringBuilder sb = new StringBuilder();
					Text[] fragments = entry.getValue().getFragments();
					for (Text fragment : fragments) {
						sb.append(fragment.toString());
					}

					// 写入高亮的内容
					try {
						String str = sb.toString();
						if("equipmentName".equals(entry.getKey())) {//替换物料名称的class属性
							str = str.replaceAll("class=\"highlight\"", "class=\"highlight2\"");
							FieldUtils.writeField(entityEsDto, entry.getKey(), str, true);
						}else {
							FieldUtils.writeField(entityEsDto, entry.getKey(), str, true);
						}
					} catch (IllegalAccessException e) {
						e.printStackTrace();
					}
				}

				result.add(entityEsDto);
			}

		} catch (IOException e) {
			e.printStackTrace();
		}

		//封装结果
		List<Map<String, Object>> resultList = new ArrayList<>();
		for (MaterialView mat : result) {
			Map<String, Object> material = new HashMap<String, Object>();
			material.put("id", mat.getId());
			material.put("equipment_name", mat.getEquipmentName());
			material.put("sys_level_one", mat.getLevelOneCode());
			material.put("img", mat.getImg());
			material.put("brand", mat.getBrandCode());
			material.put("model", mat.getModel());
			material.put("remake", mat.getRemake());
			material.put("param_info", mat.getParamInfo());
			material.put("sys_level_two", mat.getLevelTwoCode());
			material.put("sys_level_three", mat.getLevelThreeCode());
			material.put("supplier", mat.getSupplierName());
			material.put("type", mat.getType());
			material.put("brandName", mat.getBrandName());


			Map<String, Object> resultMap = new HashMap<>();
			resultMap.put("material", material);

			resultMap.put("getSysLevelOne", mat.getLevelOneName());
			resultMap.put("getSysLevelTwo", mat.getLevelTwoName());
			resultMap.put("getSysLevelThree", mat.getLevelThreeName());

			resultList.add(resultMap);
		}

		page.setPage(currentPageNo);
		page.setResults(resultList);

		client.close();

		return page;
	}

	/**
	 * 	根据条件获取全部信息
	 * @param keyword			搜索关键词-物料名称、详细参数、型号、规格、备注
	 * @param brandCodes		品牌CODES
	 * @param supplierIds		供应商IDS
	 * @param levelOneCode		一级分类CODE
	 * @param levelTwoCode		二级分类CODE
	 * @param levelThreeCode	三级分类CODE
	 * @return
	 * @throws IOException 
	 */
	private static List<MaterialView> doScroll(String keyword, String brandCodes, String supplierIds,
			String levelOneCode, String levelTwoCode, String levelThreeCode) throws IOException{

		RestHighLevelClient client = ESRestClientUtil.getClient();
		//初始化scroll
		final Scroll scroll = new Scroll(TimeValue.timeValueMinutes(1L)); //设定滚动时间间隔
		SearchRequest searchRequest = new SearchRequest();
		searchRequest.scroll(scroll);
		SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder();

		// 组织查询条件
		BoolQueryBuilder queryBuilder = setBoolQueryBuilder(keyword, brandCodes, supplierIds, levelOneCode, levelTwoCode, levelThreeCode);

		searchSourceBuilder.query(queryBuilder);
		searchSourceBuilder.size(1000); //设定每次返回多少条数据
		searchRequest.source(searchSourceBuilder);

		SearchResponse response = null;
		try {
			response = client.search(searchRequest, RequestOptions.DEFAULT);
		} catch (IOException e) {
			e.printStackTrace();
		}

		List<MaterialView> materialViews = new ArrayList<MaterialView>();
		while(true){
			for (SearchHit hit : response.getHits().getHits()) {
				MaterialView dto = JSONObject.parseObject(hit.getSourceAsString(), MaterialView.class);
				materialViews.add(dto);
			}
			//	        重置searchResponse 和ID
			String scrollId = response.getScrollId();

			SearchScrollRequest scrollRequest = new SearchScrollRequest(scrollId);
			scrollRequest.scroll(scroll);
			try {
				response = client.scroll(scrollRequest, RequestOptions.DEFAULT);
			} catch (IOException e) {
				e.printStackTrace();
			}

			if (response.getHits().getHits().length == 0) {
				break;
			}
		}
		//清除滚屏
		ClearScrollRequest clearScrollRequest = new ClearScrollRequest();
		clearScrollRequest.addScrollId(response.getScrollId());//也可以选择setScrollIds()将多个scrollId一起使用
		ClearScrollResponse clearScrollResponse = null;
		try {
			clearScrollResponse = client.clearScroll(clearScrollRequest,RequestOptions.DEFAULT);
		} catch (IOException e) {
			e.printStackTrace();
		}
		boolean succeeded = clearScrollResponse.isSucceeded();
		System.out.println("succeeded:" + succeeded);

		client.close();

		return materialViews;
	}

	public static void main(String[] args) throws IOException {
		//		System.err.println("begin----------->"+DateUtil.formatDateTime(new Date()));
		//		getSuppliers(null, null, null, null, null);
		//		getBrands(null, null, null, null, null);
		//		System.err.println("end----------->"+DateUtil.formatDateTime(new Date()));

		try {
			get("4028f3817c91038b017c910a23760004");
		} catch (Exception e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
	}

	/**
	 * 	获取搜索结果相关的供应商信息
	 * @param keyword			搜索关键词-物料名称、详细参数、型号、规格、备注
	 * @param brandCodes		品牌CODES
	 * @param levelOneCode		一级分类CODE
	 * @param levelTwoCode		二级分类CODE
	 * @param levelThreeCode	三级分类CODE
	 * @return
	 * @throws IOException 
	 */
	public static List<Map<String, Object>> getSuppliers(String keyword, String brandCodes, 
			String levelOneCode, String levelTwoCode, String levelThreeCode) throws IOException{

		List<MaterialView> materialViews = doScroll(keyword, brandCodes, null, levelOneCode, levelTwoCode, levelThreeCode);

		Map<String, MaterialView> map = new HashMap<String, MaterialView>();
		for (MaterialView mv : materialViews) {
			if(!map.containsKey(mv.getSupplierId())) {
				map.put(mv.getSupplierId(), mv);
			}
		}

		List<Map<String, Object>> supps = new ArrayList<Map<String,Object>>();

		//保存结果
		for (String sid : map.keySet()) {
			Map<String, Object> resultMap = new HashMap<String, Object>();

			String name_str = map.get(sid).getSupplierName();
			if(!StringUtil.isNotEmpty(name_str)) {
				continue;
			}

			resultMap.put("value", name_str);
			resultMap.put("id", sid);

			String name_pinyin = "";
			try {

				String name_trim = StringUtil.replaceBlank(name_str);
				if(name_trim.length()>0) {
					name_pinyin = PinyinUtil.converterToFirstSpell(name_trim.substring(0,1));
				}else {
					name_pinyin = name_trim;
				}

			} catch (Exception e) {
				// TODO: handle exception
				System.err.println("供应商名称----------》"+map.get("value").toString().trim());
				e.printStackTrace();
			}

			resultMap.put("py", name_pinyin);

			supps.add(resultMap);
		}
		System.err.println("size---------->"+supps.size());
		return supps;
	}

	/**
	 * 	获取搜索结果相关的品牌
	 * @param keyword			搜索关键词-物料名称、详细参数、型号、规格、备注
	 * @param supplierIds		供应商IDS
	 * @param levelOneCode		一级分类CODE
	 * @param levelTwoCode		二级分类CODE
	 * @param levelThreeCode	三级分类CODE
	 * @return
	 * @throws IOException 
	 */
	public static List<Map<String, Object>> getBrands(String keyword, String supplierIds,
			String levelOneCode, String levelTwoCode, String levelThreeCode) throws IOException{

		List<MaterialView> materialViews = doScroll(keyword, null, supplierIds, levelOneCode, levelTwoCode, levelThreeCode);

		Map<String, MaterialView> map = new HashMap<String, MaterialView>();
		for (MaterialView mv : materialViews) {
			if(!map.containsKey(mv.getBrandCode())) {
				map.put(mv.getBrandCode(), mv);
			}
		}

		List<Map<String, Object>> brands = new ArrayList<Map<String,Object>>();
		//保存结果
		for (String bcode : map.keySet()) {
			Map<String, Object> resultMap = new HashMap<String, Object>();

			String name_str = map.get(bcode).getBrandName();
			if(!StringUtil.isNotEmpty(name_str)) {
				continue;
			}

			resultMap.put("typeName", name_str);
			resultMap.put("typeCode", bcode);

			String name_pinyin = "";
			try {

				String name_trim = StringUtil.replaceBlank(name_str);
				if(name_trim.length()>0) {
					name_pinyin = PinyinUtil.converterToFirstSpell(name_trim.substring(0,1));
				}else {
					name_pinyin = name_trim;
				}

			} catch (Exception e) {
				// TODO: handle exception
				System.err.println("供应商名称----------》"+map.get("value").toString().trim());
				e.printStackTrace();
			}

			resultMap.put("py", name_pinyin);

			brands.add(resultMap);
		}
		System.err.println("size---------->"+brands.size());
		return brands;
	}

	/**
	 * 
	 * @param keyword			搜索关键词-物料名称、详细参数、型号、规格、备注
	 * @param brandCodes		品牌CODES
	 * @param supplierIds		供应商IDS
	 * @param levelOneCode		一级分类CODE
	 * @param levelTwoCode		二级分类CODE
	 * @param levelThreeCode	三级分类CODE
	 * @return
	 */
	public static BoolQueryBuilder setBoolQueryBuilder(String keyword, String brandCodes, String supplierIds, 
			String levelOneCode, String levelTwoCode, String levelThreeCode) {

		BoolQueryBuilder query = QueryBuilders.boolQuery();

		if (StringUtil.isNotEmpty(keyword)) {
			// 条件  字段    字段.....
			// 物料名称、详细参数、型号、规格、备注
			MultiMatchQueryBuilder queryBuilder = QueryBuilders.multiMatchQuery(keyword, 
					"equipmentName",
					"paramInfo", 
					"model", 
					"remake", 
					"supplierName", 
					"brandName", 
					"type")
					//minmum_should_match为百分之百的时候,效果等同于operator为and的情况
					.operator(Operator.OR).type(Type.CROSS_FIELDS).minimumShouldMatch("95%");

			query.should(queryBuilder);
		}

		// 添加 关键字模糊匹配条件 不分词实现sql like模糊匹配效果,使用es wildcard 通配符搜索
		//        if (StringUtil.isNotEmpty(key)) {
		//            QueryBuilder query1 = QueryBuilders.wildcardQuery("key", "*" + key + "*");
		//            query.must(query1);
		//        }

		//添加 状态条件
		if (StringUtil.isNotEmpty(levelOneCode)) {
			QueryBuilder query2 = QueryBuilders.termQuery("levelOneCode", levelOneCode);
			query.must(query2);
		}
		//添加 状态条件
		if (StringUtil.isNotEmpty(levelTwoCode)) {
			QueryBuilder query2 = QueryBuilders.termQuery("levelTwoCode", levelTwoCode);
			query.must(query2);
		}
		// termQuery不会对搜索词进行分词处理,而是作为一个整体与目标字段进行匹配,若完全匹配,则可查询到。
		if (StringUtil.isNotEmpty(levelThreeCode)) {
			QueryBuilder query2 = QueryBuilders.termQuery("levelThreeCode", levelThreeCode);
			query.must(query2);
		}

		//添加 状态条件
		if (StringUtil.isNotEmpty(brandCodes)) {
			QueryBuilder query2 = QueryBuilders.matchQuery("brandCode", brandCodes);
			query.must(query2);
		}

		// matchQuery-会将搜索词分词,再与目标查询字段进行匹配,若分词中的任意一个词与目标字段匹配上,则可查询到。
		if (StringUtil.isNotEmpty(supplierIds)) {
			QueryBuilder query2 = QueryBuilders.matchQuery("supplierId", supplierIds);
			query.must(query2);
		}

		// 添加 开始时间 结束时间范围条件
		//        if (begin != null && end != null) {
		//            QueryBuilder query4 = QueryBuilders.rangeQuery("createTime").from(begin.getTime()).to(end.getTime());
		//            query.must(query4);
		//        }
		return query;
	}

}

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值