SpringBoot整合 solr

Myth丶恋晨 2022-05-24 07:38 230阅读 0赞

引言：solr想必我们都已经很熟悉了吧，solr服务器搭建好，数据导入之后。在我们实际项目开发中，该怎么整合solr呢？

接下来，介绍SpringBoot整合solr，那么闲话不多说，直接开始吧~

一：在SpringBoot项目中引入依赖关系，在pom.xml中加入：

<properties>
            <spring.data.solr.version>2.1.1.RELEASE</spring.data.solr.version>
        </properties>
    
        <dependencyManagement>
            <dependencies>
                <dependency>
                    <groupId>org.springframework.data</groupId>
                    <artifactId>spring-data-solr</artifactId>
                    <version>${spring.data.solr.version}</version>
                </dependency>
            </dependencies>
        </dependencyManagement>
    
        <dependencies>  
            <dependency>
                <groupId>org.springframework.data</groupId>
                <artifactId>spring-data-solr</artifactId>
            </dependency>
    
            
            <dependency>
                <groupId>org.apache.solr</groupId>
                <artifactId>solr-solrj</artifactId>
                <version>6.6.2</version>
            </dependency>
        </dependencies>

二：solr配置文件，application.properties（solr配置信息我是在这个配置文件，当然还有其他的配置方法，根据你们习惯）

spring.data.solr.host=http://10.0.2.42:8983/solr/

如果指定具体的Core（core我理解为集合）话，可以写成

spring.data.solr.host=http://10.0.2.42:8983/solr/collectionname

collectionname是core的名字，可以在core  admin选项中进行配置

我们可以把collectingname理解为数据库的概念，在操作时，如果有多个core,就可以切换core

三:solr的controller类

import org.apache.solr.client.solrj.SolrClient;
    import org.apache.solr.client.solrj.SolrQuery;
    import org.apache.solr.client.solrj.response.QueryResponse;
    import org.apache.solr.common.SolrDocument;
    import org.apache.solr.common.SolrDocumentList;
    import org.apache.solr.common.SolrInputDocument;
    import org.springframework.beans.factory.annotation.Autowired;
    import org.springframework.web.bind.annotation.RequestMapping;
    import org.springframework.web.bind.annotation.RestController;
    
    import java.util.List;
    import java.util.Map;
    import java.util.UUID;
    
    @RestController
    @RequestMapping("solr")
    public class SolrController {
    
        @Autowired
        private SolrClient client;
    
        /**
         * 新增/修改 索引
         * 当 id 存在的时候, 此方法是修改(当然, 我这里用的 uuid, 不会存在的), 如果 id 不存在, 则是新增
         * @return
         */
        @RequestMapping("add")
        public String add() {
            String uuid = UUID.randomUUID().toString().replaceAll("-", "");
            try {
                SolrInputDocument doc = new SolrInputDocument();
                doc.setField("id", uuid);
                doc.setField("content_ik", "我是中国人, 我爱中国");
    
                /* 如果spring.data.solr.host 里面配置到 core了, 那么这里就不需要传 collection1 这个参数
                 * 下面都是一样的
                 */
    
                client.add("collection1", doc);
                //client.commit();
                client.commit("collection1");
                return uuid;
            } catch (Exception e) {
                e.printStackTrace();
            }
    
            return "error";
        }
    
        /**
         * 根据id删除索引
         * @param id
         * @return
         */
        @RequestMapping("delete")
        public String delete(String id)  {
            try {
                client.deleteById("collection1",id);
                client.commit("collection1");
    
                return id;
            } catch (Exception e) {
                e.printStackTrace();
            }
    
    
            return "error";
        }
    
        /**
         * 删除所有的索引
         * @return
         */
        @RequestMapping("deleteAll")
        public String deleteAll(){
            try {
    
                client.deleteByQuery("collection1","*:*");
                client.commit("collection1");
    
                return "success";
            } catch (Exception e) {
                e.printStackTrace();
            }
            return "error";
        }
    
        /**
         * 根据id查询索引
         * @return
         * @throws Exception
         */
        @RequestMapping("getById")
        public String getById() throws Exception {
            SolrDocument document = client.getById("collection1", "536563");
            System.out.println(document);
            return document.toString();
        }
    
        /**
         * 综合查询: 在综合查询中, 有按条件查询, 条件过滤, 排序, 分页, 高亮显示, 获取部分域信息
         * @return
         */
        @RequestMapping("search")
        public Map<String, Map<String, List<String>>> search(){
    
            try {
                SolrQuery params = new SolrQuery();
    
                //查询条件, 这里的 q 对应 下面图片标红的地方
                params.set("q", "手机");
    
                //过滤条件
                params.set("fq", "product_price:[100 TO 100000]");
    
                //排序
                params.addSort("product_price", SolrQuery.ORDER.asc);
    
                //分页
                params.setStart(0);
                params.setRows(20);
    
                //默认域
                params.set("df", "product_title");
    
                //只查询指定域
                params.set("fl", "id,product_title,product_price");
    
                //高亮
                //打开开关
                params.setHighlight(true);
                //指定高亮域
                params.addHighlightField("product_title");
                //设置前缀
                params.setHighlightSimplePre("<span style='color:red'>");
                //设置后缀
                params.setHighlightSimplePost("</span>");
    
                QueryResponse queryResponse = client.query(params);
    
                SolrDocumentList results = queryResponse.getResults();
    
                long numFound = results.getNumFound();
    
                System.out.println(numFound);
    
    　　　　　　　//获取高亮显示的结果, 高亮显示的结果和查询结果是分开放的
                Map<String, Map<String, List<String>>> highlight = queryResponse.getHighlighting();
    
                for (SolrDocument result : results) {
                    System.out.println(result.get("id"));
                    System.out.println(result.get("product_title"));
                    //System.out.println(result.get("product_num"));
                    System.out.println(result.get("product_price"));
                    //System.out.println(result.get("product_image"));
    
                    Map<String, List<String>> map = highlight.get(result.get("id"));
                    List<String> list = map.get("product_title");
                    System.out.println(list.get(0));
    
                    System.out.println("------------------");
                    System.out.println();
                }
                return highlight;
            } catch (Exception e) {
                e.printStackTrace();
            }
            return null;
        }
    
    }

![70][]

希望能够帮到您，完！

[70]: /images/20220524/b100ad79ee074333a2173083102cc461.png