springboot springdata 整合es

实体类

package com.fwz.tproject.bean;

import java.io.Serializable;
import java.math.BigDecimal;

import org.springframework.data.annotation.Id;
import org.springframework.data.elasticsearch.annotations.Document;
import org.springframework.data.elasticsearch.annotations.Field;
import org.springframework.data.elasticsearch.annotations.FieldType;

import lombok.Data;
import lombok.experimental.Accessors;

@Data
@Document(indexName = "product_index", shards = 3, replicas = 1)
@Accessors(chain = true)
public class Product implements Serializable {
    /**
     * 
     */
    private static final long serialVersionUID = 1L;
    @Id
    Long id;
    @Field(type = FieldType.Text, analyzer = "ik_smart")
    String title; // 标题
    @Field(type = FieldType.Text, analyzer = "ik_max_word")
    String description; // 描述
    @Field(type = FieldType.Keyword)
    String category;// 分类
    @Field(type = FieldType.Keyword)
    String brand; // 品牌
    @Field(type = FieldType.Double)
    BigDecimal price; // 价格
    @Field(index = false, type = FieldType.Keyword)
    String images; // 图片地址
}

controller

package com.fwz.tproject.controller;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.elasticsearch.core.SearchHits;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import com.fwz.tproject.bean.Product;
import com.fwz.tproject.serviceImpl.SpringESService;

/**
 * r
 * 
 * @author 冯文哲
 * @version 2018-06-11
 */
@RestController
@RequestMapping(value = "/ses")
public class SpringESController {

    @Autowired
    private SpringESService springESService;

    @RequestMapping(value = "/cindex")
    public Boolean testCreate() { //
        return springESService.createIndex();
    }

    @RequestMapping(value = "/search")
    public SearchHits<Product> search() { //
        return springESService.search();
    }

}

Service

package com.fwz.tproject.serviceImpl;

import java.math.BigDecimal;
import java.util.ArrayList;
import java.util.List;

import org.elasticsearch.action.search.SearchResponse;
import org.elasticsearch.index.query.MultiMatchQueryBuilder;
import org.elasticsearch.index.query.QueryBuilders;
import org.elasticsearch.search.aggregations.AggregationBuilders;
import org.elasticsearch.search.fetch.subphase.highlight.HighlightBuilder;
import org.elasticsearch.search.sort.SortBuilders;
import org.elasticsearch.search.sort.SortOrder;
import org.elasticsearch.search.suggest.SuggestBuilder;
import org.junit.jupiter.api.Test;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.data.domain.PageRequest;
import org.springframework.data.elasticsearch.core.ElasticsearchOperations;
import org.springframework.data.elasticsearch.core.IndexOperations;
import org.springframework.data.elasticsearch.core.SearchHits;
import org.springframework.data.elasticsearch.core.SearchHitsIterator;
import org.springframework.data.elasticsearch.core.convert.ElasticsearchConverter;
import org.springframework.data.elasticsearch.core.document.Document;
import org.springframework.data.elasticsearch.core.mapping.IndexCoordinates;
import org.springframework.data.elasticsearch.core.query.AliasQuery;
import org.springframework.data.elasticsearch.core.query.BulkOptions;
import org.springframework.data.elasticsearch.core.query.DeleteQuery;
import org.springframework.data.elasticsearch.core.query.GetQuery;
import org.springframework.data.elasticsearch.core.query.IndexQuery;
import org.springframework.data.elasticsearch.core.query.IndexQueryBuilder;
import org.springframework.data.elasticsearch.core.query.MoreLikeThisQuery;
import org.springframework.data.elasticsearch.core.query.NativeSearchQueryBuilder;
import org.springframework.data.elasticsearch.core.query.Query;
import org.springframework.data.elasticsearch.core.query.UpdateQuery;
import org.springframework.data.elasticsearch.core.query.UpdateResponse;
import org.springframework.data.util.CloseableIterator;

import com.fwz.tproject.bean.Product;
import com.fwz.tproject.service.ElasticSearchUtils;
import com.fwz.tproject.service.IdGeneratorSnowflake;
import com.fwz.tproject.service.Serv;

@SpringBootTest
class SpringESServiceTest implements Serv {

    Logger logger = LoggerFactory.getLogger(ElasticSearchUtils.class.getName());
    @Autowired
    public ElasticsearchOperations elasticsearchOperations;
    @Autowired
    public IdGeneratorSnowflake f;

    /**
     * 创建索引
     * 
     * @author fwzz
     * @version 创建时间：2021年1月29日 下午5:53:58
     *
     */
    @Test
    void testCreateIndex() {
        IndexOperations indexOperations = elasticsearchOperations.indexOps(Product.class);
        Document d = indexOperations.createMapping(Product.class);
        logger.info(d.toJson());
        logger.info("start=====================================");

        Boolean b = indexOperations.create();
        logger.info("end=====================================");
        indexOperations.putMapping(d);
        logger.info(b.toString());

    }

    /**
     * 设置别名
     * 
     * @author fwzz
     * @version 创建时间：2021年1月29日 下午5:54:05
     *
     */
    @Test
    void testSetAlias() {

        IndexOperations indexOperations = elasticsearchOperations.indexOps(Product.class);
        // 设置别名
        AliasQuery aq = new AliasQuery("my_product");
        indexOperations.addAlias(aq);
    }

    /**
     * 删除别名
     * 
     * @author fwzz
     * @version 创建时间：2021年1月29日 下午5:54:05
     *
     */
    @Test
    void testRemoveAlias() {

        IndexOperations indexOperations = elasticsearchOperations.indexOps(Product.class);
        // 删除别名
        AliasQuery aq = new AliasQuery("my_product");
        indexOperations.removeAlias(aq);
    }

    /**
     * 单条数据插入
     * 
     * @author fwzz
     * @version 创建时间：2021年1月29日 下午7:48:39
     *
     */
    @Test
    void testSaveDocuments() {
        Product product = new Product().setBrand(String.valueOf(32 / 5)).setCategory(String.valueOf(65 / 16))
                .setDescription("红色的轿车").setPrice(new BigDecimal(568)).setTitle("比亚迪轿车")
                .setImages("http://www.xymapp.cn").setId(f.snowflakeId());
        IndexQuery indexQuery = new IndexQueryBuilder().withId(product.getId().toString()).withObject(product).build();
        elasticsearchOperations.index(indexQuery, IndexCoordinates.of("my_product"));
    }

    /**
     * 单条数据插入
     * 
     * @author fwzz
     * @version 创建时间：2021年1月29日 下午7:48:39
     *
     */
    @Test
    void testSave() {
        Product product = new Product().setBrand(String.valueOf(32 / 5)).setCategory(String.valueOf(65 / 16))
                .setDescription("红色的轿车").setPrice(new BigDecimal(568)).setTitle("比亚迪轿车")
                .setImages("http://www.xymapp.cn").setId(f.snowflakeId());
        System.out.println(product.getId());
        elasticsearchOperations.save(product);
    }

    /**
     * 测试修改
     * 
     * @author fwzz
     * @version 创建时间：2021年1月29日 下午7:48:39
     *
     */
    @Test
    void testUpdate() {
        Product product = new Product().setId(1355127539135283200L).setBrand("update brand");
        elasticsearchOperations.save(product);
    }

    /**
     * 测试查询
     * 
     * @author fwzz
     * @version 创建时间：2021年1月29日 下午7:48:39
     *
     */
    @Test
    void testSearch() {
        // 分词字段/高亮字段
        String fieId = "title";
        String fieId1 = "description";
        NativeSearchQueryBuilder queryBuilder = new NativeSearchQueryBuilder();
        // match类型查询，会把查询条件进行分词，然后进行查询,多个词条之间是or的关系
        MultiMatchQueryBuilder multiMatchQueryBuilder = QueryBuilders.multiMatchQuery("比亚迪红色", fieId, fieId1);
        multiMatchQueryBuilder.type(MultiMatchQueryBuilder.Type.BEST_FIELDS);
        multiMatchQueryBuilder.tieBreaker(0.3F);
        // 应该匹配的分词的最少数量
        multiMatchQueryBuilder.minimumShouldMatch("30%");
        queryBuilder.withQuery(multiMatchQueryBuilder);
        // 排序
        queryBuilder.withSort(SortBuilders.fieldSort("price").order(SortOrder.DESC));

        HighlightBuilder.Field allHighLight = new HighlightBuilder.Field(fieId).preTags("<span style="color:red">")
                .postTags("</span>").requireFieldMatch(false);

        HighlightBuilder.Field allHighLight1 = new HighlightBuilder.Field(fieId1).requireFieldMatch(false)
                .preTags("<span style="color:red">").postTags("</span>");

        HighlightBuilder.Field[] ary = new HighlightBuilder.Field[2];
        ary[0] = allHighLight;
        ary[1] = allHighLight1;
        queryBuilder.withHighlightFields(ary);//高亮
        queryBuilder.addAggregation(AggregationBuilders.terms("4").field("category"));//聚合
        queryBuilder.withPageable(PageRequest.of(0, 10));//分页

        SearchHits<Product> hit = elasticsearchOperations.search(queryBuilder.build(), Product.class);
        logger.info(String.valueOf(hit.getSearchHits().size()));

    }

    /**
     * 批量插入
     * 
     * @author fwzz
     * @version 创建时间：2021年1月29日 下午7:48:39
     *
     */
    @Test
    void testPutBulkDocuments() {
        List<IndexQuery> queries = new ArrayList<IndexQuery>();
        for (int i = 0; i < 100; i++) {
            Product p = new Product();
            p.setBrand(String.valueOf(i / 5)).setCategory(String.valueOf(i / 16)).setDescription("红色的轿车")
                    .setPrice(new BigDecimal(568)).setTitle("比亚迪轿车").setImages("http://www.xymapp.cn")
                    .setId(f.snowflakeId());
            IndexQuery q = new IndexQueryBuilder().withId(p.getId().toString()).withObject(p).build();

            queries.add(q);
        }

        elasticsearchOperations.bulkIndex(queries, IndexCoordinates.of("my_product"));
        // elasticsearchOperations.bulkIndex(queries, index);
    }

}

ID工具类：

package com.fwz.tproject.testfunction.service;

import javax.annotation.PostConstruct;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;

import cn.hutool.core.lang.Snowflake;
import cn.hutool.core.net.NetUtil;
import cn.hutool.core.util.IdUtil;

@Component
public class IdGeneratorSnowflake {
    private long workerId = 0;
    private long datacenterId = 1;
    private Snowflake snowflake = IdUtil.createSnowflake(workerId, datacenterId);

    private static final Logger log = LoggerFactory.getLogger(IdGeneratorSnowflake.class.getName());

    // 依赖注入完成后执行该方法，进行一些初始化工作
    @PostConstruct
    public void init() {
        try {
            workerId = NetUtil.ipv4ToLong(NetUtil.getLocalhostStr());
            log.info("当前机器的workerId: {}", workerId);
        } catch (Exception e) {
            e.printStackTrace();
            log.warn("当前机器的workerId获取失败", e);
            // 释放ID
            workerId = NetUtil.getLocalhostStr().hashCode();
        }
    }

    // 使用默认机房号获取ID
    public synchronized long snowflakeId() {
        return snowflake.nextId();
    }

    // 自己制定机房号获取ID
    public synchronized long snowflakeId(long workerId, long datacenterId) {
        Snowflake snowflake = IdUtil.createSnowflake(workerId, datacenterId);

        return snowflake.nextId();
    }

    /**
     * 生成的是不带-的字符审，类似于: 73a64edf935d4952a287739a66f96e06
     * 
     * @return
     */
    public String simpleUUID() {
        return IdUtil.simpleUUID();
    }

    /**
     * 生成的UUID是带-的字符串，类似于: b12b6401-6f9c-4351-b2b6-d8afc9ab9272
     * 
     * @return
     */
    public String randomUUID() {
        return IdUtil.randomUUID();
    }

    public static void main(String[] args) {
        IdGeneratorSnowflake f = new IdGeneratorSnowflake();
        for (int i = 0; i < 1000; i++) {

            System.out.println(f.snowflakeId(0, 0));
        }

    }
}

要使用工具类pom.xml需加入依赖：

 <dependency>
            <groupId>cn.hutool</groupId>
            <artifactId>hutool-captcha</artifactId>
            <version>5.2.0</version>
        </dependency>

<dependency>
<groupId>org.projectlombok</groupId>
<artifactId>lombok</artifactId>
<optional>true</optional>
</dependency>