trzxycx 发表于 2016-12-16 08:53:02

二十六、solr的基本使用

  public class Message {
    private String id;
    private String title;
    private String[] content;
   
   
    public Message() {
        super();
    }

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

    public String getId() {
        return id;
    }
   
    @Field
    public void setId(String id) {
        this.id = id;
    }
    public String getTitle() {
        return title;
    }
   
       //使用solr的注解
    @Field("msg_title")
    public void setTitle(String title) {
        this.title = title;
    }
    public String[] getContent() {
        return content;
    }
   
    @Field("msg_content")
    public void setContent(String[] content) {
        this.content = content;
    }
}
  

  

  public class SolrTest {
    private final static String        URL        = "http://localhost:8080/solr";
    private CommonsHttpSolrServer    server    = null;
   
    @Before
    public void init() {
        try {
            server = new CommonsHttpSolrServer(URL);
        } catch (MalformedURLException e) {
            e.printStackTrace();
        }
    }
   
    @Test
    public void test01() {
        try {
            SolrInputDocument doc = new SolrInputDocument();
            //id是唯一的主键,当多次添加的时候,最后添加的相同id的域会覆盖前面的域
            doc.addField("id", "1");
                 //具体的Field需要在schema.xml文件中通过field标签定义,并且定义好分词器
            doc.addField("msg_title", "这是我的第一个solrj的程序");
            doc.addField("msg_content", "我的solrj的程序究竟能不能跑得起来呢?");
            server.add(doc);
            server.commit();
        } catch (MalformedURLException e) {
            e.printStackTrace();
        } catch (SolrServerException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
   
    @Test
    public void test02() {
        try {
            List<SolrInputDocument> docs = new ArrayList<SolrInputDocument>();
            SolrInputDocument doc = new SolrInputDocument();
            doc.addField("id", "2");
            doc.addField("msg_title", "很好!solr可以工作了");
            doc.addField("msg_content", "slor总算可以正式工作了");
            docs.add(doc);
            doc = new SolrInputDocument();
            doc.addField("id", "3");
            doc.addField("msg_title", "测试一下solr的添加");
            doc.addField("msg_content", "看看能不能添加一个列表信息");
            docs.add(doc);
            server.add(docs);
            server.commit();
        } catch (SolrServerException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
   
    @Test
    public void test03() {
        try {
            List<Message> msgs = new ArrayList<Message>();
            msgs.add(new Message("4", "基于java bean的添加", new String[] { "通过java bean完成添加",
                    "java bean的添加附件" }));
                 //添加的域如果是个数组,需要将schema.xml中field的mutilValue设置成true
            msgs.add(new Message("5", "基于java bean的列表数据的添加", new String[] { "测试如何通过一个对象完成添加",
                    "通过对象完成添加的附件" }));
            server.addBeans(msgs);
            server.commit();
        } catch (SolrServerException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
   
    @Test
    public void test04() {
        try {
            //定义查询字符串
            SolrQuery query = new SolrQuery("*");
                 //分页
            query.setStart(0);
            query.setRows(3);
            QueryResponse resp = server.query(query);
            //查询出来的结果都保存在SolrDocumentList中
            SolrDocumentList sdl = resp.getResults();
            System.out.println(sdl.getNumFound());
            for (SolrDocument sd : sdl) {
                //                System.out.println(sd);
                System.out.println(sd.getFieldValue("msg_title") + ","
                                    + sd.getFieldValue("msg_content"));
            }
        } catch (SolrServerException e) {
            e.printStackTrace();
        }
    }
   
    @Test
    public void test05() {
        try {
            SolrQuery query = new SolrQuery("*");
            query.setStart(0);
            query.setRows(3);
            QueryResponse resp = server.query(query);
            //可以直接查询相应的bean对象,但是不是很常用
            List<Message> list = resp.getBeans(Message.class);
            System.out.println(list.size());
            for (Message msg : list) {
                System.out.println(msg.getTitle());
            }
        } catch (SolrServerException e) {
            e.printStackTrace();
        }
    }
   
    @Test
    public void test06() {
        try {
            SolrQuery query = new SolrQuery("测试");
            query.setHighlight(true).setHighlightSimplePre("<span class='highligter'>")
                .setHighlightSimplePost("</span>").setStart(0).setRows(5);
              //设置哪些字段高亮
            query.setParam("hl.fl", "msg_title,msg_content");
            QueryResponse resp = server.query(query);
            //查询出来的结果都保存在SolrDocumentList中
            SolrDocumentList sdl = resp.getResults();
            System.out.println(sdl.getNumFound());
            for (SolrDocument sd : sdl) {
                String id = (String) sd.getFieldValue("id");
                      //取出高亮字段
                System.out.println(resp.getHighlighting().get(id).get("msg_content"));
            }
        } catch (SolrServerException e) {
            e.printStackTrace();
        }
    }
}
页: [1]
查看完整版本: 二十六、solr的基本使用