solr学习三(测试类,含普通与ExtractingRequestHandler测试)

solr客户端基本是配置出来的,服务端可以对其进行测试,我使用的是solrj服务端。 
如果初学solr,先使用普通的测试类: 

Java代码  收藏代码
  1. import java.io.IOException;    
  2. import java.util.ArrayList;    
  3. import java.util.Collection;    
  4. import java.util.Date;  
  5. import org.apache.solr.client.solrj.SolrQuery;    
  6. import org.apache.solr.client.solrj.SolrServer;    
  7. import org.apache.solr.client.solrj.SolrServerException;    
  8. import org.apache.solr.client.solrj.impl.CommonsHttpSolrServer;    
  9. import org.apache.solr.client.solrj.request.AbstractUpdateRequest;    
  10. import org.apache.solr.client.solrj.request.UpdateRequest;    
  11. import org.apache.solr.client.solrj.response.QueryResponse;    
  12. import org.apache.solr.common.SolrInputDocument;    
  13.     
  14. public class SolrTest {  
  15.     
  16.     public static void main(String[] args) throws IOException,    
  17.             SolrServerException {    
  18.     
  19.         String urlString = " http://localhost:8393/keyPlace";    
  20.         SolrServer server = new CommonsHttpSolrServer(urlString);    
  21.         testAdd(server);  
  22.         testQuery(server);  
  23.           
  24.     }  
  25.     static void testAdd(SolrServer server) throws IOException,    
  26.        SolrServerException {  
  27.         SolrInputDocument doc1 = new SolrInputDocument();    
  28.         doc1.addField("id", 456);    
  29.         doc1.addField("orgId", "33030300310");   
  30.         doc1.addField("name", "张三");   
  31.         doc1.addField("key", "1");  
  32.         doc1.addField("createDate", new Date());  
  33.         SolrInputDocument doc2 = new SolrInputDocument();    
  34.         doc2.addField("id", 123);  
  35.         doc2.addField("orgId", "33030300310");   
  36.         doc2.addField("name", "李四");  
  37.         doc2.addField("key", "2");  
  38.         doc2.addField("createDate", new Date());  
  39.         Collection<SolrInputDocument> docs = new ArrayList<SolrInputDocument>();    
  40.         docs.add(doc1);    
  41.         docs.add( doc2 );    
  42.         server.add(docs);    
  43.         UpdateRequest req = new UpdateRequest();    
  44.         req.setAction(AbstractUpdateRequest.ACTION.COMMIT, false, false);    
  45.         req.add(docs);    
  46.         req.process(server);    
  47.     }  
  48.       
  49.     static void testQuery(SolrServer server) throws IOException,    
  50.        SolrServerException {  
  51.         SolrQuery query = new SolrQuery();    
  52.         
  53.         query.setQuery("name:张三");    
  54.         query.setHighlight(true).setHighlightSnippets(1);                                                       
  55.     
  56.         QueryResponse ret = server.query(query);    
  57.     
  58.         System.out.println(ret);    
  59.           
  60.     }  
  61.   
  62.    


如果服务端配置了ExtractingRequestHandler,可使用下面的类进行测试: 

Java代码  收藏代码
    1. import java.io.File;  
    2. import java.io.IOException;  
    3.   
    4. import org.apache.solr.client.solrj.SolrServer;  
    5. import org.apache.solr.client.solrj.SolrServerException;  
    6.   
    7. import org.apache.solr.client.solrj.request.AbstractUpdateRequest;  
    8. import org.apache.solr.client.solrj.response.QueryResponse;  
    9. import org.apache.solr.client.solrj.SolrQuery;  
    10. import org.apache.solr.client.solrj.impl.CommonsHttpSolrServer;  
    11. import org.apache.solr.client.solrj.request.ContentStreamUpdateRequest;  
    12. import org.apache.solr.common.params.ModifiableSolrParams;  
    13. import org.apache.solr.handler.extraction.ExtractingParams;  
    14.   
    15. public class ExtractTest {  
    16.   
    17.         public static void main(String[] args) {  
    18.             try {  
    19.               String urlString = "http://localhost:8393/sourcePool/";   
    20.               SolrServer solr = new CommonsHttpSolrServer(urlString);  
    21.               String fileName = "c:/slor1.doc";   
    22.               String solrId = "2";   
    23.               String resoucePoolid = "2";  
    24.                 
    25.               indexFilesSolrCell(fileName, solrId , resoucePoolid , solr);  
    26.               testQuery(solr);  
    27.             } catch (Exception ex) {  
    28.               System.out.println(ex.toString());  
    29.             }  
    30.           }  
    31.             
    32.           /** 
    33.            * Method to index all types of files into Solr.  
    34.            * @throws IOException 
    35.            * @throws SolrServerException 
    36.            */  
    37.           public static void indexFilesSolrCell(String fileName, String solrId , String resoucePoolid , SolrServer solr)   
    38.             throws IOException, SolrServerException {  
    39.               
    40.             ContentStreamUpdateRequest up   
    41.               = new ContentStreamUpdateRequest("/update/extract");  
    42.               
    43.             ModifiableSolrParams p = new ModifiableSolrParams();  
    44.             p.add(ExtractingParams.LITERALS_PREFIX + "orgids" , "33010033001");  
    45.             p.add(ExtractingParams.LITERALS_PREFIX + "orgids" , "33010033002");  
    46.             p.add(ExtractingParams.LITERALS_PREFIX + "orgids" , "33010033003");  
    47.             up.setParams(p);  
    48.               
    49.             up.addFile(new File(fileName));  
    50.             up.setParam(ExtractingParams.LITERALS_PREFIX + "id", solrId);  
    51.             up.setParam(ExtractingParams.LITERALS_PREFIX + "resoucepoolid", resoucePoolid);  
    52.             up.setParam(ExtractingParams.LITERALS_PREFIX + "orgid", "33010033001");  
    53.             up.setParam(ExtractingParams.LITERALS_PREFIX + "name", "33010033001");  
    54.             up.setParam(ExtractingParams.LITERALS_PREFIX + "releaseunit", "33010033001");  
    55.             up.setParam(ExtractingParams.LITERALS_PREFIX + "releasetime", "2011-02-12");  
    56.   
    57.             up.setParam(ExtractingParams.UNKNOWN_FIELD_PREFIX, "attr_");  
    58.               
    59.             //up.setParam("fmap.content", "filestream");  
    60.               
    61.             up.setAction(AbstractUpdateRequest.ACTION.COMMIT, true, true);  
    62.               
    63.             solr.request(up);  
    64.               
    65.               
    66.           }  
    67.             
    68.           static void testQuery(SolrServer solr)  throws IOException, SolrServerException {  
    69.               String fileStream = "filestream:老婆的老公";  
    70.               String field = "orgid:33010033001";  
    71.               QueryResponse rsp = solr.query(new SolrQuery(field));  
    72.                 System.out.println(rsp);   
    73.           }  

posted on 2017-10-30 15:23  cxhfuujust  阅读(167)  评论(0编辑  收藏  举报

导航