全文检索技术solr

  1 package cn.xjy.test ;
  2 
  3 import java.io.IOException ;
  4 import org.apache.solr.client.solrj.SolrQuery ;
  5 import org.apache.solr.client.solrj.SolrQuery.ORDER ;
  6 import org.apache.solr.client.solrj.SolrServer ;
  7 import org.apache.solr.client.solrj.impl.HttpSolrServer ;
  8 import org.apache.solr.client.solrj.response.QueryResponse ;
  9 import org.apache.solr.common.SolrDocument ;
 10 import org.apache.solr.common.SolrDocumentList ;
 11 import org.apache.solr.common.SolrInputDocument ;
 12 import org.junit.Before ;
 13 import org.junit.Test ;
 14 
 15 public class TestSolrJ {
 16     
 17     SolrServer    solrServer    = null ;
 18     
 19     @Before
 20     public void init() {
 21         // 1、 创建HttpSolrServer对象，通过它和Solr服务器建立连接。
 22         // 参数：solr服务器的访问地址
 23         solrServer = new HttpSolrServer("http://localhost:8080/solr/collection2") ;
 24     }
 25     
 26     /**
 27      * 更新或添加
 28      * 存在就更新,不存在就添加(根据id判断)
 29      * @throws Exception
 30      * @throws IOException
 31      */
 32     @Test
 33     public void testAddandUpdate() throws Exception, IOException {
 34         
 35         // 2、 创建SolrInputDocument对象，然后通过它来添加域。
 36         SolrInputDocument doc = new SolrInputDocument() ;
 37         
 38         /* 测试添加 */
 39         /* doc.addField("id","测试1"); doc.addField("product_name","测试name");
 40          * doc.addField("product_catalog_name","测试product_catalog_name");
 41          * doc.addField("product_price",22f);
 42          * doc.addField("product_description","测试product_description");
 43          * doc.addField("product_picture","测试product_picture"); */
 44         
 45         /* 测试更新 */
 46         
 47         // 第一个参数：域的名称，域的名称必须是在schema.xml中定义的
 48         // 第二个参数：域的值
 49         // 注意：id的域不能少
 50         doc.addField("id", "测试1") ;
 51         doc.addField("product_name", "测试更新name") ;
 52         doc.addField("product_catalog_name", "测试更新product_catalog_name") ;
 53         doc.addField("product_price", 23f) ;
 54         doc.addField("product_description", "测试更新product_description") ;
 55         doc.addField("product_picture", "测试更新product_picture") ;
 56         
 57         // 3、 通过HttpSolrServer对象将SolrInputDocument添加到索引库。
 58         solrServer.add(doc) ;
 59         
 60         // 4、 提交。
 61         solrServer.commit() ;
 62         
 63     }
 64     
 65     /**
 66      * 删除索引
 67      * @throws Exception
 68      */
 69     @Test
 70     public void testDelete() throws Exception {
 71         // 根据id删除
 72         /* solrServer.deleteById("测试1") ; solrServer.commit(); */
 73         
 74         // 根据条件删除
 75         solrServer.deleteByQuery("product_price:23") ;
 76         
 77         // 全部删除
 78         // solrServer.deleteByQuery("*:*");
 79         solrServer.commit() ;
 80     }
 81     
 82     /**
 83      * 简单查询
 84      * @throws Exception 
 85      */
 86     @Test
 87     public void testSimpleQuery() throws Exception {
 88         
 89         SolrQuery query = new SolrQuery() ;
 90         // q是固定的且必须 的
 91         query.set("q", "id:测试1") ;
 92         query.set("fl", "id,product_catalog_name") ;
 93         // query.setFields("id", "product_catalog_name");//效果同上
 94         
 95         QueryResponse queryRespOnse= solrServer.query(query) ;
 96         
 97         SolrDocumentList results = queryResponse.getResults() ;
 98         
 99         System.out.println("总数:" + results.getNumFound()) ;
100         
101         for (SolrDocument solrDocument : results) {
102             // System.out.println(solrDocument.get("id")) ;
103             System.out.println(solrDocument) ;
104         }
105     }
106     
107     /**
108      * 复杂查询
109      * @throws Exception 
110      */
111     @Test
112     public void testComplexQuery() throws Exception {
113         SolrServer solrServer2 = new HttpSolrServer("http://localhost:8080/solr") ;
114         
115         SolrQuery query = new SolrQuery() ;
116         
117         // 查询条件
118         query.setQuery("*:*") ;
119         
120         // 过滤条件
121         query.setFilterQueries("product_catalog_name:幽默杂货") ;
122         
123         // 按指定字段排序
124         query.setSort("id", ORDER.desc) ;
125         
126         // 分页
127         query.setStart(0) ;
128         query.setRows(100) ;
129         
130         // 设置要显示的字段
131         query.setFields("id,product_catalog_name") ;
132         
133         // 给特定字段设置样式
134         query.setHighlight(true) ;
135         query.addHighlightField("product_catalog_name") ;
136         query.setHighlightSimplePre("") ;
137         query.setHighlightSimplePost("") ;
138         
139         // 查询
140         QueryResponse queryRespOnse= solrServer2.query(query) ;
141         
142         // 获得结果集
143         SolrDocumentList results = queryResponse.getResults() ;
144         
145         System.out.println("总数:" + results.getNumFound()) ;
146         
147         // 遍历结果集
148         for (SolrDocument solrDocument : results) {
149             System.out.println(solrDocument) ;
150         }
151     }
152 }