springBoot整合solr

述：接上篇solr学习笔记https://blog.csdn.net/Dawn____Dawn/article/details/126230673

1. maven 依赖

<!--solr坐标-->
<dependency>
    <groupId>org.springframework.data</groupId>
    <artifactId>spring-data-solr</artifactId>
</dependency>

<!--操作solr的工具-->
<!--solrJ是java的客户端，使用它可以对solr的索引库进行增删改查等操作-->
<dependency>
    <groupId>org.apache.solr</groupId>
    <artifactId>solr-solrj</artifactId>
</dependency>

2. spring-data-solr对solrJ进行了封装。

import org.apache.http.auth.Credentials;
import org.apache.http.auth.UsernamePasswordCredentials;
import org.apache.http.client.params.AuthPolicy;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.data.solr.core.RequestMethod;
import org.springframework.data.solr.core.SolrTemplate;
import org.springframework.data.solr.server.support.HttpSolrClientFactory;

/**
 * SolrConfig
 *
 **/
@Configuration
public class SolrConfig {

    @Value("${spring.data.solr.name:}")
    private String name;

    @Value("${spring.data.solr.pwd:}")
    private String pwd;

    @Value("${spring.data.solr.host:}")
    private String serverUrl;

    @Bean
    public SolrTemplate solrTemplate(HttpSolrClientFactory httpSolrClientFactory) {
        return new SolrTemplate(httpSolrClientFactory, RequestMethod.POST);
    }

    @Bean
    public HttpSolrClientFactory httpSolrClientFactory() {
        EimsSolrClient solrClient = new EimsSolrClient(new EimsSolrClient.Builder(this.serverUrl));
        solrClient.setBasicAuthUser(this.name);
        solrClient.setBasicAuthPwd(this.pwd);
        Credentials defaultcreds = new UsernamePasswordCredentials(this.name, this.pwd);
        return new HttpSolrClientFactory(solrClient, defaultcreds, AuthPolicy.BASIC);
    }
}

3. 配置文件

# solr
spring.data.solr.host=http://localhost:8983/solr/
spring.data.solr.name=admin
spring.data.solr.pwd=

4. 调用示例

添加索引

@Autowired
private SolrTemplate solrTemplate;

@Autowired
private ProjectService projectService;

/**
 * 保存（新增、修改）
 *
 **/
@Override
public boolean save(long id) {
    ProjectModel model = this.projectService.getModelById(id);
    if(model == null) {
        logger.error("未找到要添加的项目！");
        return false;
    }

    if(!model.getMiniSale()) {
        return this.delById(model.getId());
    }
    ProjectOperationDto dto = new ProjectOperationDto();
    dto.setSolrId(String.valueOf(model.getId()));
    dto.setId(model.getId());
    dto.setName(model.getName());
    dto.setStatus(model.getStatus());
    dto.setSummary(model.getSummary());
    dto.setCatNamePath(model.getProjectCatNamePath());
    dto.setSpecifications(model.getSpecifications());
    dto.setPrice(model.getPrice() != null? model.getPrice().doubleValue() : Double.valueOf(0));
    dto.setMiniSale(model.getMiniSale());
    dto.setSaleTimeLimited(model.getSaleTimeLimited());
    dto.setBeginTime(model.getSaleBeginTime());
    dto.setEndTime(model.getSaleEndTime());

    UpdateResponse response = null;
    try {
        response = solrTemplate.saveBean(SolrCollectionsEnum.PROJECT_CORE.getCore(), dto);
        solrTemplate.commit(SolrCollectionsEnum.PROJECT_CORE.getCore());
    } catch (Exception e) {
        logger.error("索引保存异常", e);
    }

    if(response != null && response.getStatus() == 0) {
        return true;
    }
    return false;

}

删除索引

/**
 * 删除
 **/
@Override
public boolean delById(long id) {
    UpdateResponse response = null;
    try {
        response = solrTemplate.deleteByIds(SolrCollectionsEnum.PROJECT_CORE.getCore(), String.valueOf(id));
        solrTemplate.commit(SolrCollectionsEnum.PROJECT_CORE.getCore());
    } catch (Exception e) {
        logger.error("索引删除异常", e);
    }
    if(response != null && response.getStatus() == 0) {
        return true;
    }
    return false;
}

查询匹配的数据

    /**
     * 查询匹配的数据
     * HighlightPage<ArticleSolrQueryItemVo>
     * list：getContent()
     * 总页数：getTotalPages()
     * 总记录数：getTotalElements()
     * 是否有上一页：hasPrevious()
     * 是否有下一页：hasNext()
     *
     * @param queryDto
     * @return
     * @throws BusinessException
     */
    public HighlightPage<ArticleSolrItemVo> queryByDto(QuestionSolrQueryDto queryDto) throws BusinessException {
        if (StringUtils.isBlank(queryDto.getKeyWords())) {
            throw new BusinessException("使用搜索引擎查询，关键字必填");
        }

        //高亮查询
        HighlightQuery query = new SimpleHighlightQuery();
        HighlightOptions highlightOptions = new HighlightOptions().addField("title").addField("answer"); //设置高亮字段
        highlightOptions.setSimplePrefix("<em style='color:red'>");//设置高亮前缀
        highlightOptions.setSimplePostfix("</em>");//设置高亮后缀
        query.setHighlightOptions(highlightOptions);//设置高亮选项

        //分页
        query.setOffset(queryDto.getOffset());
        query.setRows(queryDto.getPageSize());

        //查询条件
        Criteria criteria = Criteria.where("keywords").is(queryDto.getKeyWords());
        query.addCriteria(criteria);

        //排序(SortType等于0时，无需处理。solr默认的排序就是：相关度降序)
        if (queryDto.getSortType() == 1) {
            Sort sort = Sort.by(Direction.DESC, "createTime");
            query.addSort(sort);
        }
        if (queryDto.getSortType() == 2) {
            Sort sort = Sort.by(Direction.DESC, "readCount");
            query.addSort(sort);
        }

        //高亮查询
        HighlightPage<ArticleSolrItemVo> page = solrTemplate.queryForHighlightPage(
                SolrCollectionsEnum.QUESTION_CORE.getCore(), query, ArticleSolrItemVo.class);
        for (HighlightEntry<ArticleSolrItemVo> h : page.getHighlighted()) {//循环高亮入口集合
            ArticleSolrItemVo item = h.getEntity();

            for (Highlight highlight : h.getHighlights()) {
                if ("title".equals(highlight.getField().getName())) {
                    item.setTitle(StrUtils.listToString(highlight.getSnipplets()));
                }
                if ("answer".equals(highlight.getField().getName())) {
                    item.setContent(StrUtils.listToString(highlight.getSnipplets()));
                }
            }
        }

        return page;
    }

5. 开启基本身份验证

创建一个security.json文件开启基本的身份验证。