Solr系列第02节之搜索和中文分词

package com.xuanyuv.solrj;
import java.util.ArrayList;
import java.util.List;
import org.apache.solr.client.solrj.SolrQuery;
import org.apache.solr.client.solrj.SolrServer;
import org.apache.solr.client.solrj.SolrServerException;
import org.apache.solr.client.solrj.impl.HttpSolrServer;
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 com.xuanyuv.model.MyMessage;

/**
 * Solr系列第02节之搜索和中文分词
 * Created by 玄玉<https://www.xuanyuv.com/> on 2013/08/20 14:12.
 */
public enum HelloSolrJ {
    INSTANCE;

    private SolrServer server;

    private HelloSolrJ(){
        //创建SolrServer对象，它有两个线程安全的子类EmbeddedSolrServer和CommonsHttpSolrServer
        //EmbeddedSolrServer-----内嵌式的，不需要启动Web服务器
        //CommonsHttpSolrServer--需要启动Web服务器，它是通过HTTP请求来获取数据的
        //CommonsHttpSolrServer已不推荐直接使用了，而是推荐使用HttpSolrServer类
        //同样的StreamingUpdateSolrServer也不推荐使用了，推荐使用ConcurrentUpdateSolrServer
        server = new HttpSolrServer("http://127.0.0.1:8088/solr");
    }

    /**
     * 删除全部索引
     */
    public void deleteAllIndex(){
        try {
            //删除所有的索引，删除完记得commit才能生效
            server.deleteByQuery("*:*");
            server.commit();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    /**
     * 添加索引
     * ----------------------------------------------------------------------
     * 这里之所以没有硬编码指定my_title和my_content的中文分词器
     * 是由于schema.xml中在定义这俩<field>时已通过type属性指定了中文分词器
     * ----------------------------------------------------------------------
     */
    public void addIndex(){
        List<SolrInputDocument> docs = new ArrayList<SolrInputDocument>();
        for(int i=1; i<4; i++){
            SolrInputDocument doc = new SolrInputDocument();
            //对于Solr而言，默认的id是唯一的主键（如果我们没有自定义主键的话），当多次添加的时候，最后添加的相同id的域会覆盖已有的
            doc.addField("id", Integer.toString(i));
            //这里使用的是自定义的Field，所以就需要在schema.xml中为其添加<field>描述，并通过type属性指定其使用的分词器
            doc.addField("my_title", "这是我的第" + i + "个SolrJ程序");
            doc.addField("my_content", "my第" + i + "个SolrJ程序的运行情况");
            docs.add(doc);
        }
        try {
            //可以只添加一个Document，也可以一次添加多个Document，这里添加多个
            server.add(docs);
            //commit后方能生效，否则搜索到的还是add前的索引信息
            server.commit();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    /**
     * 使用JavaBean来添加索引
     */
    public void addIndexByBean(){
        List<MyMessage> msgs = new ArrayList<MyMessage>();
        //由于MyMessage()的第三个参数允许传入多个值，此时就要在schema.xml中配置<field my_content multiValued="true"/>支持多值域
        //并且在MyMessage.java中要使用@Field注解标注，可以标注在属性或setter方法上，并通过注解的默认值来指定所要存储的Field的名字
        msgs.add(new MyMessage("4", "这是第4个SolrJ", new String[]{"第4个SolrJ程序诞生中国", "它是通过SolrJ OF JavaBean4添加"}));
        msgs.add(new MyMessage("5", "这是第5个SolrJ", new String[]{"第5个SolrJ程序诞生哈尔滨", "它是通过SolrJ OF JavaBean5添加"}));
        try {
            server.addBeans(msgs);
            server.commit();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    /**
     * 搜索文档
     */
    public void searchFile(){
        //设置查询字符串
        SolrQuery query = new SolrQuery("my_content:诞生");
        //设置了setStart和setRows就可以实现分页效果了
        query.setStart(0);
        query.setRows(3);
        QueryResponse resp = null;
        try {
            resp = server.query(query);
        } catch (SolrServerException e) {
            e.printStackTrace();
        }
        //搜索出来的结果都保存在SolrDocumentList中
        SolrDocumentList sdl = resp.getResults();
        //这里获取到的是搜索到的总数量，与query.setRows(3)无关
        System.out.println("本次搜索到[" + sdl.getNumFound() + "]条数据");
        for(SolrDocument sd : sdl){
            //System.out.println(sd)会输出下面的
            //SolrDocument[{id=4, my_title=这是第4个SolrJ, my_content=[第4个...中国, 它是...添加]}]
            System.out.println("my_title=" + sd.getFieldValue("my_title") + " my_content=" + sd.getFieldValue("my_content"));
        }
    }

    /**
     * 使用JavaBean接收文档搜索结果
     */
    public void searchFileByBean(){
        SolrQuery query = new SolrQuery("id:*");
        query.setStart(0);
        query.setRows(3);
        QueryResponse resp = null;
        try {
            resp = server.query(query);
        } catch (SolrServerException e) {
            e.printStackTrace();
        }
        List<MyMessage> list = resp.getBeans(MyMessage.class);
        //这里获取到的不是搜索到的总数量，而是与query.setRows(3)有关的值
        //若实际的SolrQuery("id:*")能搜索到5条，则这里得到的就是3，若实际能搜索到的是2条，则这里得到的就是2条
        System.out.println("本次搜索到[" + list.size() + "]条数据");
        for(MyMessage msg : list) {
            System.out.println(msg.getTitle() + "         " + msg.getContent()[0]);
        }
    }
}

package com.xuanyuv.model;
import org.apache.solr.client.solrj.beans.Field;

public class MyMessage {
    @Field
    private String id;
    @Field("my_title")
    private String title;
    @Field("my_content")
    private String[] content;

    /*-- 三个属性的setter和getter略 --*/

    public MyMessage() {}

    public MyMessage(String id, String title, String[] content) {
        this.id = id;
        this.title = title;
        this.content = content;
    }
}

package com.xuanyuv.test;
import org.junit.Test;
import com.xuanyuv.solrj.HelloSolrJ;

public class HelloSolrJTest {
    @Test
    public void deleteAllIndex(){
        HelloSolrJ.INSTANCE.deleteAllIndex();
    }

    @Test
    public void addIndexAndSearchFile(){
        HelloSolrJ.INSTANCE.addIndex();
        HelloSolrJ.INSTANCE.searchFile();
    }

    @Test
    public void addIndexAndSearchFileByBean(){
        HelloSolrJ.INSTANCE.addIndexByBean();
        HelloSolrJ.INSTANCE.searchFileByBean();
    }
}