Solrj实现增删改查

淡淡的烟草味﹌ 2022-05-27 04:27 345阅读 0赞

### 一、准备 ###

在 `solrhome/collection1/conf/schema.xml` 配置文件中定义相关的字段。

<field name="product_name" type="text_ik" indexed="true" stored="true"/>
    <field name="product_price" type="float" indexed="true" stored="true"/>
    <field name="product_description" type="text_ik" indexed="true" stored="false" />
    <field name="product_catalog_name" type="string" indexed="true" stored="true" />
    
    
    <field name="product_keywords" type="text_ik" indexed="true" stored="false" multiValued="true"/>
    <copyField source="product_name" dest="product_keywords"/>
    <copyField source="product_description" dest="product_keywords"/>

这里通过新建 Maven 项目完成测试，在`pom.xml`中添加对应的依赖：

<dependency>
            <groupId>org.apache.solr</groupId>
            <artifactId>solr-solrj</artifactId>
            <version>7.2.1</version>
        </dependency>
        
        <dependency>
            <groupId>junit</groupId>
            <artifactId>junit</artifactId>
            <version>4.11</version>
            <scope>test</scope>
        </dependency>

### 二、测试增删改 ###

增删改相对来说比较简单，这里放在一起进行测试，Solr 中没有提供专门的修改方法，会根据 id 在文档中查找，如果没有找到就是添加，找到了就会覆盖原来的数据，即修改。

#### 2.1新增与修改 ####

编写 Java 代码：

@Test
        public void indexCreateTest() throws Exception {
             // 创建和Solr 服务端的连接，并指定 Solr 实例是 "collection1"
            SolrClient client = new HttpSolrClient
                .Builder("http://192.168.248.136:8080/solr/collection1").build();
            // 创建 Solr 文档对象
            SolrInputDocument document = new SolrInputDocument();
    
            // 向文档对象中添加指定字段对应的字段值，字段必须先定义后使用，且必须要有 id 字段
            // 如果是修改，只需要固定 id 字段的值，修改其他的字段即可
            document.addField("id", "a1");
            document.addField("product_catalog_name", "幽默杂货");
            document.addField("product_price", "20");
            document.addField("product_name", "小王子");
    
             // 将文档添加到 client 对象中
            client.add(document);
            // 提交
            client.commit();
        }

在浏览器端进行验证：  
![70][]

#### 2.2删除 ####

编写 Java 代码：

@Test
        public void indexDelTest() throws Exception {
            SolrClient client = new HttpSolrClient
                    .Builder("http://192.168.248.136:8080/solr/collection1").build();
    
            // 根据 id 删除
           // client.deleteById("a1");
            // 根据查询删除, *:* 表示删除所有
            client.deleteByQuery("product_name:小王子");
    
            client.commit();
        }

这里就不在浏览器验证了。

### 三、测试查询 ###

相对于增删改操作，往往需要我们做的是查询操作。查询操作也相对比较复杂，有关查询的设置，可以在浏览器端查看，只要在代码中设置即可。

查询高亮的结果比较麻烦，这里将浏览器端的查询结果贴出来，希望能帮助大家更好理解高亮数据的获取方式。需要高亮显示的字段是：`product_name`，查询关键字是：`小王子`。  
![这里写图片描述][70 1]

编写 Java 代码：

@Test
        public void indexSearchTest() throws Exception {
            SolrClient client = new HttpSolrClient
                    .Builder("http://192.168.248.136:8080/solr/collection1").build();
            // 创建查询对象
            SolrQuery query = new SolrQuery();
    
            // 设置默认搜索字段，如果不指定搜索的字段，则从默认字段中搜索
            query.set("df", "product_keywords");
            // 指定搜索域与搜索的关键字
            query.setQuery("product_name:小王子");
            // 如果不定搜索域，则从默认搜索域中搜索，如下
            //query.setQuery("手机");
    
            // 设置搜索的过滤器，只搜索"product_price" 在 10-20 之间的
            query.addFilterQuery("product_price:[15 TO 30]");
            // 设置起始的条数，默认是 0
            query.setStart(0);
            // 设置查询的条数，默认是 10 
            query.setRows(5);
            // 设置高亮
            query.setHighlight(true);
            // 设置显示高亮的字段
            query.addHighlightField("product_name");
            // 设置高亮字段值的前缀
            query.setHighlightSimplePre("<span style=\"color:red\">");
            // 设置高亮字段值的后缀
            query.setHighlightSimplePost("</span>");
    
            // 获得查询结果的响应对象
            QueryResponse response = client.query(query);
            // 从响应对象中获得结果集对象
            SolrDocumentList list = response.getResults();
    
            System.out.println("查询到的总记录数：" + list.getNumFound());
    
            // 遍历结果集
            for (SolrDocument document : list){
                System.out.println("product_price : " + document.get("product_price"));
                System.out.println("product_name : " + document.get("product_name"));
    
                // 从响应对象中获得高亮，并处理
                Map<String, Map<String, List<String>>> map = response.getHighlighting();
                List<String> lightList = map.get(document.get("id")).get("product_name");
    
                if(lightList != null && lightList.size() > 0) {
                    System.out.println("high lighting product_name : " + lightList.get(0));
                }
    
                System.out.println("==================================");
            }
        }

查询结果输出：  
![这里写图片描述][70 2]

(完)

[70]: /images/20220527/08ea62a08ccf4fe9ba90b0fbd5628db6.png
[70 1]: /images/20220527/58796d6df0ce4073bba2ea1fb65fbc1e.png
[70 2]: /images/20220527/9cb96522bcf143398d3329da5f2b6d8d.png