ElasticSearch整合JavaApi代码

package com.itheima.day2; import com.alibaba.fastjson.JSON; import com.itheima.day2.domain.Goods; import com.itheima.day2.mapper.GoodsMapper; import org.elasticsearch.action.bulk.BulkRequest; import org.elasticsearch.action.bulk.BulkResponse; import org.elasticsearch.action.delete.DeleteRequest; import org.elasticsearch.action.index.IndexRequest; import org.elasticsearch.action.search.SearchRequest; import org.elasticsearch.action.search.SearchResponse; import org.elasticsearch.action.update.UpdateRequest; import org.elasticsearch.client.RequestOptions; import org.elasticsearch.client.RestHighLevelClient; import org.elasticsearch.common.text.Text; import org.elasticsearch.common.xcontent.XContentType; import org.elasticsearch.index.query.*; import org.elasticsearch.rest.RestStatus; import org.elasticsearch.search.SearchHit; import org.elasticsearch.search.SearchHits; import org.elasticsearch.search.aggregations.Aggregation; import org.elasticsearch.search.aggregations.AggregationBuilder; import org.elasticsearch.search.aggregations.AggregationBuilders; import org.elasticsearch.search.aggregations.Aggregations; import org.elasticsearch.search.aggregations.bucket.terms.Terms; 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.SortOrder; import org.junit.Test; import org.junit.runner.RunWith; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.boot.test.context.SpringBootTest; import org.springframework.test.context.junit4.SpringRunner; import java.io.IOException; import java.util.ArrayList; import java.util.HashMap; import java.util.List; import java.util.Map; @RunWith(SpringRunner.class) @SpringBootTest(classes = Day2Application.class) public class Day2ApplicationTests { @Autowired private RestHighLevelClient client; @Autowired private GoodsMapper goodsMapper; /* * 批量操作 * */ @Test public void testBulk() throws IOException { //创建bulkrequest对象，整合所有操作 BulkRequest bulkRequest = new BulkRequest(); //删除1号记录 DeleteRequest deleteRequest = new DeleteRequest(); bulkRequest.add(deleteRequest); // 添加6号记录 Map map = new HashMap(); map.put("name", "六号"); IndexRequest indexRequest = new IndexRequest("person"); bulkRequest.add(indexRequest); Map map2 = new HashMap(); map2.put("name", "三号"); //3. 修改3号记录 名称为 “三号” UpdateRequest updateReqeust = new UpdateRequest("person", "3").doc(map2); bulkRequest.add(updateReqeust); BulkResponse response = client.bulk(bulkRequest, RequestOptions.DEFAULT); RestStatus status = response.status(); System.out.println(status); } @Test public void getSql() { List<Goods> goodsList = goodsMapper.findAll(); System.out.println(goodsList.toString()); } //导入 @Test public void importData() throws IOException { //1.查询所有数据，mysql List<Goods> goodsList = goodsMapper.findAll(); //System.out.println(goodsList.size()); //2.bulk导入 BulkRequest bulkRequest = new BulkRequest(); //2.1 循环goodsList，创建IndexRequest添加数据 for (Goods goods : goodsList) { //2.2 设置spec规格信息 Map的数据 specStr:{} goods.setSpec(JSON.parseObject(goods.getSpecStr(), Map.class)); //将goods对象转换为json字符串 String data = JSON.toJSONString(goods);//map --> {} IndexRequest indexRequest = new IndexRequest("goods"); indexRequest.id(goods.getId() + "").source(data, XContentType.JSON); bulkRequest.add(indexRequest); } BulkResponse response = client.bulk(bulkRequest, RequestOptions.DEFAULT); System.out.println(response.status()); } //查询所有 matchAll @Test public void testMatchAll() throws IOException { //2.构建查询请求对象，指定查询的索引名称 SearchRequest searchRequest = new SearchRequest("goods"); //4.创建查询条件构造器builder SearchSourceBuilder sourceBuilder = new SearchSourceBuilder(); //6.查询条件 QueryBuilder query = QueryBuilders.matchAllQuery();//查询所有 //5,指定查询 sourceBuilder.query(query); //3.添加查询条件构造器 searchRequest.source(sourceBuilder); //8.添加分页信息 sourceBuilder.from(0); sourceBuilder.size(100); //1.查询，获取查询结果 SearchResponse response = client.search(searchRequest, RequestOptions.DEFAULT); //7.获取命中对象 SearchHits hits = response.getHits(); //7.1获取总记录数 long value = hits.getTotalHits().value; System.out.println("总记录数：" + value); List<Goods> goodsList = new ArrayList<>(); //7.2获取hits数据 数组 SearchHit[] hits1 = hits.getHits(); for (SearchHit hit : hits1) { //获取json字符串格式的数据 String sourceAsString = hit.getSourceAsString(); //转为java对象 Goods goods = JSON.parseObject(sourceAsString, Goods.class); goodsList.add(goods); } for (Goods goods : goodsList) { System.out.println(goods.toString()); } } //查询所有 term查询 @Test public void testTerm() throws IOException { //2.构建查询请求对象，指定查询的索引名称 SearchRequest searchRequest = new SearchRequest("goods"); //4.创建查询条件构造器builder SearchSourceBuilder sourceBuilder = new SearchSourceBuilder(); //6.查询条件，使用term查询！ QueryBuilder query = QueryBuilders.termQuery("title", "华为"); //5,指定查询 sourceBuilder.query(query); //3.添加查询条件构造器 searchRequest.source(sourceBuilder); //8.添加分页信息 sourceBuilder.from(0); sourceBuilder.size(100); //1.查询，获取查询结果 SearchResponse response = client.search(searchRequest, RequestOptions.DEFAULT); //7.获取命中对象 SearchHits hits = response.getHits(); //7.1获取总记录数 long value = hits.getTotalHits().value; System.out.println("总记录数：" + value); List<Goods> goodsList = new ArrayList<>(); //7.2获取hits数据 数组 SearchHit[] hits1 = hits.getHits(); for (SearchHit hit : hits1) { //获取json字符串格式的数据 String sourceAsString = hit.getSourceAsString(); //转为java对象 Goods goods = JSON.parseObject(sourceAsString, Goods.class); goodsList.add(goods); } for (Goods goods : goodsList) { System.out.println(goods.toString()); } } //matchQuery分词查询 @Test public void matchQuery() throws IOException { //2.构建查询请求对象，指定查询的索引名称 SearchRequest searchRequest = new SearchRequest("goods"); //4.创建查询条件构造器builder SearchSourceBuilder sourceBuilder = new SearchSourceBuilder(); //6.查询条件，使用term查询！ MatchQueryBuilder query = QueryBuilders.matchQuery("title", "华为手机"); query.operator(Operator.AND); //5,指定查询 sourceBuilder.query(query); //3.添加查询条件构造器 searchRequest.source(sourceBuilder); //8.添加分页信息 sourceBuilder.from(0);