六、spring boot 集成elasticsearch

作者: 茶铺里的水 | 来源:发表于2017-11-27 18:06 被阅读625次

1. 新增依赖

我本地使用的spring boot版本为1.5.8.RELEASE

compile("org.springframework.boot:spring-boot-starter-data-elasticsearch")

2. 新增配置

application.yml新增关于es的一些配置

# elasticsearch
spring:
    data:
        elasticsearch:
            cluster-name: elasticsearch
            cluster-nodes: 127.0.0.1:9300

3. 新增model

package com.mk.model;

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.FieldIndex;
import org.springframework.data.elasticsearch.annotations.FieldType;

import java.util.Date;

@Document(indexName = "mk_spring_es", type = "article", shards = 1, replicas = 0, refreshInterval = "-1")
public class ArticleInfo {

    @Id
    private String id;

    @Field(type = FieldType.String, store = true, analyzer = "ik", index = FieldIndex.analyzed)
    private String articleAuthor;

    @Field(type = FieldType.String, store = true, analyzer = "ik", index = FieldIndex.analyzed)
    private String articleTitle;

    @Field(type = FieldType.String, store = true, analyzer = "ik", index = FieldIndex.analyzed)
    private String articleContent;

    @Field
    private Date createTime;

    public String getId() {
        return id;
    }

    public void setId(String id) {
        this.id = id;
    }

    public String getArticleAuthor() {
        return articleAuthor;
    }

    public void setArticleAuthor(String articleAuthor) {
        this.articleAuthor = articleAuthor;
    }

    public String getArticleTitle() {
        return articleTitle;
    }

    public void setArticleTitle(String articleTitle) {
        this.articleTitle = articleTitle;
    }

    public String getArticleContent() {
        return articleContent;
    }

    public void setArticleContent(String articleContent) {
        this.articleContent = articleContent;
    }

    public Date getCreateTime() {
        return createTime;
    }

    public void setCreateTime(Date createTime) {
        this.createTime = createTime;
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;

        ArticleInfo that = (ArticleInfo) o;

        if (id != null ? !id.equals(that.id) : that.id != null) return false;
        if (articleAuthor != null ? !articleAuthor.equals(that.articleAuthor) : that.articleAuthor != null)
            return false;
        if (articleTitle != null ? !articleTitle.equals(that.articleTitle) : that.articleTitle != null) return false;
        if (articleContent != null ? !articleContent.equals(that.articleContent) : that.articleContent != null)
            return false;
        return createTime != null ? createTime.equals(that.createTime) : that.createTime == null;
    }

    @Override
    public int hashCode() {
        int result = id != null ? id.hashCode() : 0;
        result = 31 * result + (articleAuthor != null ? articleAuthor.hashCode() : 0);
        result = 31 * result + (articleTitle != null ? articleTitle.hashCode() : 0);
        result = 31 * result + (articleContent != null ? articleContent.hashCode() : 0);
        result = 31 * result + (createTime != null ? createTime.hashCode() : 0);
        return result;
    }

    @Override
    public String toString() {
        return "ArticleInfo{" +
                "id='" + id + '\'' +
                ", articleAuthor='" + articleAuthor + '\'' +
                ", articleTitle='" + articleTitle + '\'' +
                ", articleContent='" + articleContent + '\'' +
                ", createTime=" + createTime +
                '}';
    }
}

4. 新增dao

package com.mk.dao;

import com.mk.model.ArticleInfo;
import org.springframework.data.elasticsearch.repository.ElasticsearchRepository;
import org.springframework.stereotype.Component;

@Component("elasticsearchArticleDao")
public interface ElasticsearchArticleDao extends ElasticsearchRepository<ArticleInfo, String> {

}

5. 新增service

package com.mk.service;

import com.mk.entity.Page;
import com.mk.model.ArticleInfo;

import java.util.List;

public interface ElasticsearchArticleService {

    void createMapping() throws Exception;

    // 保存一条数据
    void saveArticle(ArticleInfo info);

    // 保存多条数据
    void saveArticles(List<ArticleInfo> list);

    // 根据ID获取文章
    ArticleInfo findArticleInfoById(String id);

    // 根据关键字查询
    List<ArticleInfo> findAllArticle();

    // 根据关键字查询
    List<ArticleInfo> findArticleByKeyword(String keyword);

    // 根据关键字查询（分页）
    Page<ArticleInfo> findArticleByKeywordWithPage(String keyword, int pageNo, int pageSize);
}

6. 新增service实现

package com.mk.service.impl;

import com.mk.dao.ElasticsearchArticleDao;
import com.mk.entity.Page;
import com.mk.model.ArticleInfo;
import com.mk.service.ElasticsearchArticleService;
import org.elasticsearch.action.admin.indices.mapping.put.PutMappingRequest;
import org.elasticsearch.client.Requests;
import org.elasticsearch.common.xcontent.XContentBuilder;
import org.elasticsearch.common.xcontent.XContentFactory;
import org.elasticsearch.index.query.QueryBuilder;
import org.elasticsearch.index.query.QueryBuilders;
import org.elasticsearch.search.highlight.HighlightBuilder;
import org.elasticsearch.search.sort.FieldSortBuilder;
import org.elasticsearch.search.sort.SortBuilder;
import org.elasticsearch.search.sort.SortOrder;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
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.query.NativeSearchQueryBuilder;
import org.springframework.data.elasticsearch.core.query.SearchQuery;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;

@Service("elasticsearchArticleService")
public class ElasticsearchArticleServiceImpl implements ElasticsearchArticleService{

    private static final Logger logger = LoggerFactory.getLogger(ElasticsearchArticleServiceImpl.class);


    @Resource
    private ElasticsearchArticleDao elasticsearchArticleDao;

    @Resource
    private ElasticsearchTemplate elasticsearchTemplate;

    @Override
    public void createMapping() throws Exception{
        String index = "mk_spring_es";
        String type = "article";
        elasticsearchTemplate.getClient().admin().indices().prepareCreate(index).execute().actionGet();
        XContentBuilder builder= XContentFactory.jsonBuilder()
                .startObject()
                .startObject(type)
                .startObject("properties")
                .startObject("articleTitle").field("type", "string").field("store", "yes").field("analyzer","ik").field("index","analyzed").endObject()
                .startObject("articleContent").field("type", "string").field("store", "yes").field("analyzer","ik").field("index","analyzed").endObject()
                .endObject()
                .endObject()
                .endObject();
        PutMappingRequest mapping = Requests.putMappingRequest(index).type(type).source(builder);
        elasticsearchTemplate.getClient().admin().indices().putMapping(mapping).actionGet();

    }

    @Override
    public void saveArticle(ArticleInfo info) {
        info.setCreateTime(new Date());
        elasticsearchArticleDao.save(info);
    }

    @Override
    public void saveArticles(List<ArticleInfo> list) {
        for (ArticleInfo info : list) {
            info.setCreateTime(new Date());
        }
        elasticsearchArticleDao.save(list);
    }

    @Override
    public ArticleInfo findArticleInfoById(String id) {
        return elasticsearchArticleDao.findOne(id);
    }

    @Override
    public List<ArticleInfo> findAllArticle() {
        Iterable<ArticleInfo> iterable = elasticsearchArticleDao.search(QueryBuilders.matchAllQuery());
        List<ArticleInfo> list = new ArrayList<>();
        if(iterable != null){
            iterable.forEach(list::add);
        }
        return list;
    }

    @Override
    public List<ArticleInfo> findArticleByKeyword(String keyword) {
        logger.info("query by keyword:{}",keyword);
        HighlightBuilder.Field field = new HighlightBuilder.Field("articleTitle");
        field.preTags("<span>");
        field.postTags("</span>");
        HighlightBuilder.Field[] fields = {field};
        QueryBuilder queryBuilder = QueryBuilders.multiMatchQuery(keyword, "articleAuthor", "articleTitle", "articleContent");
        SortBuilder sortBuilder = new FieldSortBuilder("createTime");
        sortBuilder.order(SortOrder.DESC);
        SearchQuery searchQuery = new NativeSearchQueryBuilder()
                .withQuery(queryBuilder)
                .withHighlightFields(fields)
                .withSort(sortBuilder)
                .build();
        List<ArticleInfo> list = new ArrayList<>();
        Iterable<ArticleInfo> iterable = elasticsearchArticleDao.search(searchQuery);
        if(iterable != null){
            iterable.forEach(list::add);
        }
        return list;
    }

    @Override
    public Page<ArticleInfo> findArticleByKeywordWithPage(String keyword, int pageNo, int pageSize) {
        Page<ArticleInfo> result = new Page<>();
        result.setPageNo(pageNo);
        result.setPageSize(pageSize);
        // 查询
        QueryBuilder queryBuilder = QueryBuilders.multiMatchQuery(keyword, "articleAuthor", "articleTitle", "articleContent");
        // 分页
        Pageable pageable = new PageRequest(pageNo-1, pageSize);
        // 排序
        SortBuilder sortBuilder = new FieldSortBuilder("createTime");
        sortBuilder.order(SortOrder.DESC);
        SearchQuery searchQuery = new NativeSearchQueryBuilder()
                .withQuery(queryBuilder)
                .withPageable(pageable)
                .withSort(sortBuilder)
                .build();
        org.springframework.data.domain.Page<ArticleInfo> queryResult = elasticsearchArticleDao.search(searchQuery);
        if(queryResult.getTotalElements() > 0){
            List<ArticleInfo> list = new ArrayList<>();
            queryResult.forEach(list::add);
            result.setTotal(Long.valueOf(queryResult.getTotalElements()).intValue());
            result.setList(list);
        }
        return result;
    }
}

7. 新增controller

package com.mk.controller;

import com.mk.entity.Page;
import com.mk.entity.Response;
import com.mk.model.ArticleInfo;
import com.mk.model.User;
import com.mk.service.ElasticsearchArticleService;
import io.swagger.annotations.ApiImplicitParam;
import io.swagger.annotations.ApiImplicitParams;
import io.swagger.annotations.ApiOperation;
import org.springframework.web.bind.annotation.*;

import javax.annotation.Resource;
import java.util.List;

@RestController
@RequestMapping("/article")
public class ArticleController {

    @Resource
    private ElasticsearchArticleService elasticsearchArticleService;

    @ApiOperation(value = "创建mapping",notes = "创建mapping")
    @RequestMapping(value = "/createMapping",method = {RequestMethod.GET})
    public Response createMapping(){
        Response response = new Response();
        try {
            elasticsearchArticleService.createMapping();
        } catch (Exception e){
            response.setResult(false);
            response.setMessage(e.getMessage());
        }
        return response;
    }

    @ApiOperation(value = "保存一篇文章",notes = "保存一篇文章")
    @ApiImplicitParam(name = "articleInfo",value = "文章信息", required = true, dataType = "ArticleInfo")
    @RequestMapping(value = "/save",method = {RequestMethod.GET,RequestMethod.POST})
    public Response save(@ModelAttribute ArticleInfo articleInfo){
        Response response = new Response();
        try {
            elasticsearchArticleService.saveArticle(articleInfo);
        } catch (Exception e){
            response.setResult(false);
            response.setMessage(e.getMessage());
        }
        return response;
    }

    @ApiOperation(value = "根据ID查询一篇文章",notes = "根据ID查询一篇文章")
    @ApiImplicitParam(name = "id",value = "文章id", required = true, dataType = "String")
    @RequestMapping(value = "/findById",method = {RequestMethod.GET})
    public Response findById(@RequestParam String id){
        Response<ArticleInfo> response = new Response<>();
        try {
            ArticleInfo info = elasticsearchArticleService.findArticleInfoById(id);
            response.setData(info);
        } catch (Exception e){
            response.setResult(false);
            response.setMessage(e.getMessage());
        }
        return response;
    }

    @ApiOperation(value = "查询所有文章",notes = "查询所有文章")
    @RequestMapping(value = "/findAll",method = {RequestMethod.GET})
    public Response findAll(){
        Response<List> response = new Response<>();
        try {
            List<ArticleInfo> list = elasticsearchArticleService.findAllArticle();
            response.setData(list);
        } catch (Exception e){
            response.setResult(false);
            response.setMessage(e.getMessage());
        }
        return response;
    }

    @ApiOperation(value = "根据关键字查询文章",notes = "根据关键字查询文章")
    @ApiImplicitParam(name = "keyword",value = "关键字", required = true, dataType = "String")
    @RequestMapping(value = "/findByKeyword",method = {RequestMethod.GET})
    public Response findByKeyword(@RequestParam String keyword){
        Response<List> response = new Response<>();
        try {
            List<ArticleInfo> list = elasticsearchArticleService.findArticleByKeyword(keyword);
            response.setData(list);
        } catch (Exception e){
            response.setResult(false);
            response.setMessage(e.getMessage());
        }
        return response;
    }

    @ApiOperation(value = "根据关键字查询文章（带分页）",notes = "根据关键字查询文章（带分页）")
    @ApiImplicitParam(name = "keyword",value = "关键字", required = true, dataType = "String")
    @RequestMapping(value = "/findByKeywordWithPage",method = {RequestMethod.GET})
    public Response findByKeywordWithPage(@RequestParam("keyword") String keyword,
                                          @RequestParam("pageNo") int pageNo,
                                          @RequestParam("pageSize") int pageSize){
        Response<Page> response = new Response<>();
        try {
            Page<ArticleInfo> list = elasticsearchArticleService.findArticleByKeywordWithPage(keyword,pageNo,pageSize);
            response.setData(list);
        } catch (Exception e){
            response.setResult(false);
            response.setMessage(e.getMessage());
        }
        return response;
    }
}

8. 测试

save

save

findById

findById

findByKeywordWithPage

findByKeywordWithPage

六、spring boot 集成elasticsearch

1. 新增依赖

2. 新增配置

3. 新增model

4. 新增dao

5. 新增service

6. 新增service实现

7. 新增controller

8. 测试

相关文章

网友评论

延伸阅读

深度阅读

栏目导航

热点阅读