4、solr的增删改查和高亮以及分组

冷不防 2022-09-18 15:56 57阅读 0赞

代码如下：

package com.hj.solr;
    
    import org.apache.solr.client.solrj.beans.Field;
    
    /**
     * 在变量的set方法上注解上lucene内部的字段名称
     */
    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;
    	}
    	
    	@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;
    	}
    }

package com.hj.solr;
    
    import java.io.IOException;
    import java.net.MalformedURLException;
    import java.util.ArrayList;
    import java.util.List;
    
    import org.apache.solr.client.solrj.SolrQuery;
    import org.apache.solr.client.solrj.SolrServerException;
    import org.apache.solr.client.solrj.impl.CommonsHttpSolrServer;
    import org.apache.solr.client.solrj.response.FacetField.Count;
    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.junit.Test;
    
    /**
     * @date 2013年12月4日
     * @author huangjie
     */
    @SuppressWarnings("deprecation")
    public class SolrTest {
    	//指定solr服务器的地址
    	private final static String URL = "http://localhost:8080/solr";
    	
    	@Test
    	public void test1(){
    		//1、创建SolrServer对象，该对象有两个可以使用，都是线程安全的
    //		CommonsHttpSolrServer：启动web服务器使用的，通过http请求的
    //		EmbeddedSolrServer：内嵌式的，导入solr的jar包就可以使用了
    		try {
    			CommonsHttpSolrServer server = new CommonsHttpSolrServer(URL);
    			//把查询出来的数据全部删除
    //			server.deleteByQuery("*:*");
    //			server.commit();
    			
    			SolrInputDocument doc = new SolrInputDocument();
    			//id是必填的，并且是String类型的
    			//<field name="id" type="string" indexed="true" stored="true" required="true" />
    			//id是唯一的主键，当多次添加的时候，最后添加的相同id会覆盖前面的域
    			doc.addField("id", "1");
    			doc.addField("msg_title", "这是我的第一个solrj程序");
    			doc.addField("msg_content", "solr程序的运行");
    			server.add(doc);
    			server.commit();
    		} catch (MalformedURLException e) {
    			e.printStackTrace();
    		} catch (SolrServerException e) {
    			e.printStackTrace();
    		} catch (IOException e) {
    			e.printStackTrace();
    		}
    	}
    	
    	/**
    	 * 基于列表的添加
    	 * @throws SolrServerException
    	 * @throws IOException
    	 */
    	@Test
    	public void test2() throws SolrServerException, IOException{
    		List<SolrInputDocument> docs = new ArrayList<SolrInputDocument>();
    		SolrInputDocument doc = new SolrInputDocument();
    		doc.addField("id", "2");
    		doc.addField("msg_title", "很好，solr可以工作了");
    		doc.addField("msg_content", "solr总算可以正式工作了");
    		
    		docs.add(doc);
    		
    		doc = new SolrInputDocument();
    		doc.addField("id", "3");
    		doc.addField("msg_title", "测试以下solr的添加");
    		doc.addField("msg_content", "看看能不能添加一个列表信息");
    		
    		docs.add(doc);
    		
    		CommonsHttpSolrServer server = new CommonsHttpSolrServer(URL);
    		server.add(docs);
    		server.commit();
    	}
    	
    	/**
    	 * 基于javabean的添加
    	 * @throws SolrServerException
    	 * @throws IOException
    	 */
    	@Test
    	public void test3() throws SolrServerException, IOException{
    		List<Message> msgs = new ArrayList<Message>();
    		//多值域的添加使用数组
    		msgs.add(new Message("4","基于javabean的添加", new String[]{"javabean的添加附件","javabean的添加附件1"}));
    		msgs.add(new Message("5","基于javabean的列表数据的添加", new String[]{"通过对象完成添加的附件","通过对象完成添加的附件1"}));
    		
    		CommonsHttpSolrServer server = new CommonsHttpSolrServer(URL);
    		server.addBeans(msgs);
    		server.commit();
    	}
    	
    	@Test
    	public void test4() throws SolrServerException, MalformedURLException{
    		CommonsHttpSolrServer server = new CommonsHttpSolrServer(URL);
    		//定义查询字符串
    		SolrQuery query = new SolrQuery("*:*");
    		//实现分页的查询
    		query.setStart(0);
    		query.setRows(3);
    		QueryResponse res = server.query(query);
    		//查询出来的结果都保存在SolrDocumentList中
    		SolrDocumentList sdl = res.getResults();
    		System.out.println("总数："+sdl.getNumFound());
    		for(SolrDocument sd : sdl){
    			System.out.println(sd.get("id")+"#"+sd.get("msg_title")+"#"+sd.get("msg_content"));
    		}
    	}
    	
    	@Test
    	public void test5() throws MalformedURLException, SolrServerException{
    		CommonsHttpSolrServer server = new CommonsHttpSolrServer(URL);
    		//相当于QueryParser
    		SolrQuery query = new SolrQuery("*:*");
    		query.setStart(1);
    		query.setRows(3);
    		QueryResponse res = server.query(query);
    		//可以直接查询相应的bean对象，但是不是很常用
    		//使用这种方式无法获取总数量
    		List<Message> list = res.getBeans(Message.class);
    		System.out.println("当前总数："+list.size());
    		for(Message msg : list){
    			System.out.println(msg.getId()+"#"+msg.getTitle()+"#"+msg.getContent());
    		}
    	}
    	
    	@Test
    	public void test6() throws SolrServerException, MalformedURLException{
    		CommonsHttpSolrServer server = new CommonsHttpSolrServer(URL);
    		SolrQuery query = new SolrQuery("msg_content:solr");
    		query.setHighlight(true).setHighlightSimplePre("<span class='red'>").setHighlightSimplePost("</span>")
    		.setStart(0).setRows(10);
    		//hl.fl表示高亮的field，也就是高亮的区域
    		query.setParam("hl.fl", "msg_title,msg_content");
    		QueryResponse res = server.query(query);
    		
    		SolrDocumentList sdl = res.getResults();
    		System.out.println("总数："+sdl.getNumFound());
    		for(SolrDocument sd : sdl){
    //			System.out.println(sd.get("id")+"#"+sd.get("msg_title")+"#"+sd.get("msg_content"));
    			String id = (String) sd.get("id");
    			//在solr这里对需要加高亮的字段必须要在索引中的store=true才行
    			System.out.println(id+"#"+res.getHighlighting().get(id).get("msg_content"));;
    			
    		}
    	}
    	
    	/**
    	 * 测试分组
    	 * @throws MalformedURLException 
    	 * @throws SolrServerException 
    	 */
    	@Test
    	public void test7() throws MalformedURLException, SolrServerException{
    		CommonsHttpSolrServer server = new CommonsHttpSolrServer(URL);
    		SolrQuery query = new SolrQuery("city:*");
    		query.setIncludeScore(false);
    		query.setFacet(true);
    		query.addFacetField("city");
    		query.setFacetSort(true);
    		QueryResponse res = server.query(query);
    		List<Count> countList = res.getFacetField("city").getValues();
    		
    		for(Count count : countList){
    			System.out.println(count.getName()+"#"+count.getCount());
    		}
    		/**
    		 * 输出结果：
    		 *  上海#5290
    			深圳#2763
    			广州#2504
    			北京#1962
    			东莞#1764
    			杭州#1713
    			苏州#1661
    			南京#1529
    		 */
    	}
    
    	
    	
    }

1、 ![SouthEast][]

2、![SouthEast 1][]

3、![SouthEast 2][]

4、![SouthEast 3][]

5、![SouthEast 4][]

6、![SouthEast 5][]

7、![SouthEast 6][]

8、![SouthEast 7][]

9、![SouthEast 8][]

10、![SouthEast 9][]

工程下载地址：[http://download.csdn.net/detail/wxwzy738/6665131][http_download.csdn.net_detail_wxwzy738_6665131]

[SouthEast]: /images/20220828/95b3c629759044b2ab5ec651584b57d1.png
[SouthEast 1]: /images/20220828/6539142c1d744f308b205e31e93f5a42.png
[SouthEast 2]: /images/20220828/2144d951554448409115adf5794e010e.png
[SouthEast 3]: /images/20220828/f97b2234d1ef4097afef9b31bd429eed.png
[SouthEast 4]: /images/20220828/62276cccb799401bb039d6d5fad897b8.png
[SouthEast 5]: /images/20220828/a1deff3de3284b61a0bb1e38f7f859ce.png
[SouthEast 6]: /images/20220828/81266ea743514aac9dc7f9777b805935.png
[SouthEast 7]: /images/20220828/14c2066919254ec88bd53f8c70eeac8c.png
[SouthEast 8]: /images/20220828/5c40cb1a3e6b485aaf45f4eb3b018073.png
[SouthEast 9]: /images/20220828/a8564eb2f7f8421880239286915a085c.png
[http_download.csdn.net_detail_wxwzy738_6665131]: http://download.csdn.net/detail/wxwzy738/6665131