SolrJava的使用

                                                                        SolrJava的使用

     SolrJ 是为java 准备的客户端,用于操作Solr服务器的工作.

见:http://wiki.apache.org/solr/Solrj  。

另示例如下:

package org.com.solr36.test001;

import java.io.IOException;
import java.net.MalformedURLException;
import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.sql.SQLException;
import java.sql.Types;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Map;

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 org.junit.Before;
import org.junit.Test;

public class SolrTest01 {

   private final static String URL = "http://localhost:8080/solr" ;
   
   SolrServer  httpSolrServer = null ;
   @Before
   public void init()
   {
	     httpSolrServer = new HttpSolrServer(URL); 
   }
	@Test
	 public void test01()
	 {
		//1.创建SolrServer对象(HttpSolrServer, CommonsHttpSolrServer,EmbeddedSolrServer)
		//在Solr3.6 以后:Deprecated. Use HttpSolrServer instead. 
		//new CommonsHttpSolrServer();
		try {
			
			//httpSolrServer.deleteByQuery("*:*");
			//httpSolrServer.commit() ;
			
			// 对solr 而言, 如果 id 相同,其它的属性值不相同的话,后者会更新前者。
			SolrInputDocument  sid = new SolrInputDocument();
			sid.addField("id", "4") ;  // id 是solr_home/conf/schema.xml 中的主键值,必须有,注意类型
			sid.addField("name", "我是一个外国人");
			sid.addField("msg_title", "这是我的第一个solr程序11111");
			sid.addField("msg_content", "我的第一个solr能否运行吗222?") ;
			httpSolrServer.add(sid) ;
			httpSolrServer.commit();
			
			
			
		} catch (SolrServerException e) {
			e.printStackTrace();
		} catch (IOException e) {
			e.printStackTrace();
		}
		
		
	 }
	
	@Test
	 public void test02()
	 {
		try {
			
			List<SolrInputDocument>  docs = new ArrayList<SolrInputDocument>();
			for(int i=5;i<10;i++)
			{
				SolrInputDocument  sid = new SolrInputDocument();
				sid.addField("id", i+"") ;  // id 是solr_home/conf/schema.xml 中的主键值,必须有,注意类型
				sid.addField("name", "我是一个外国人"+i);
				sid.addField("title", "这是我的第一个solr程序"+i);
				docs.add(sid);
			}
 			
			//sid.addField("content", "我的第一个solr能否运行吗?") ;
			httpSolrServer.add(docs);
			httpSolrServer.commit();
			
			
			//SolrParams params = new SolrParams();
			//httpSolrServer.query(params )
			
			
		} catch (SolrServerException e) {
			e.printStackTrace();
		} catch (IOException e) {
			e.printStackTrace();
		}
	 }
	
	@Test
	 public void test03()
	 {
		try {
			
			List<Message>  docs = new ArrayList<Message>();
			for(int i=5;i<10;i++)
			{
				Message  sid = new Message();
				sid.setId(i+"");
				sid.setTitle("这是我的第一个solr程序Bean"+i);
				sid.setName("我是一个外国人BEAN"+i);
				sid.setMsg_content(new String[]{sid.getTitle(),sid.getName()});
				docs.add(sid);
			}
			httpSolrServer.addBeans(docs);
			httpSolrServer.commit();
			
			
			//SolrParams params = new SolrParams();
			//httpSolrServer.query(params )
			
			
		} catch (SolrServerException e) {
			e.printStackTrace();
		} catch (IOException e) {
			e.printStackTrace();
		}
	 }
	
	@Test
	public void test04()
	{
		
		//SolrParams 
		SolrQuery query = new SolrQuery("*"); // msg_all:222
		query.setStart(0);  //分页 起始
		query.setRows(4) ; // 每页查多少条记录
		try {
			QueryResponse qresponse = httpSolrServer.query(query) ;
			SolrDocumentList  sorlList = qresponse.getResults();
			System.out.println(sorlList.getNumFound());
			for(SolrDocument sd :sorlList)
			{
				System.out.println(sd);
				System.out.println(sd.getClass());
				System.out.println(sd.getFieldValue("msg_title"));
			}
		} catch (SolrServerException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
		
	}
	 
	/**
	 * 不常 用
	 */
	@Test
	public void test05()
	{
		
		//SolrParams 
		SolrQuery query = new SolrQuery("*"); // msg_all:222
		query.setStart(0);  //分页 起始
		query.setRows(4) ; // 每页查多少条记录
		try {
			QueryResponse qresponse = httpSolrServer.query(query) ;
			List<Message> beans = qresponse.getBeans(Message.class);
			for(Message sd :beans)
			{
				System.out.println(sd);
				System.out.println(sd.getClass());
				System.out.println(sd.getName());
			}
		} catch (SolrServerException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
		
	}
	
	/**
	 * 高亮显示
	 */
	@Test
	public void test06()
	{
		
		//SolrParams 
		SolrQuery query = new SolrQuery("msg_all:程序"); // msg_all:222
		query.setHighlight(true).setHighlightSimplePre("<span class='highlighter'>")
		     .setHighlightSimplePost("</span>")
		     .setStart(0)
		     .setRows(5) ;
		query.setParam("hl.fl", "msg_all,msg_name,msg_content");  // hl.fl  这是一个固定值 
		//query.setStart(0);  //分页 起始
		//query.setRows(4) ; // 每页查多少条记录
		try {
			QueryResponse qresponse = httpSolrServer.query(query) ;
			SolrDocumentList  sorlList = qresponse.getResults();
			System.out.println(sorlList.getNumFound());
			for(SolrDocument sd :sorlList)
			{
				System.out.println(sd);
				System.out.println(sd.getClass());
				String id =sd.getFieldValue("id")+"" ;
				
				//System.out.println(sd.getFieldValue("msg_title"));
				Map<String,List<String>> maphighlighter = qresponse.getHighlighting().get(id);
				if(maphighlighter!= null)
				{
					System.out.println(maphighlighter.get("msg_content"));  // 这个msg_content一定要 store 才能查得到,否则查不到高亮。
					System.out.println(maphighlighter.get("msg_all")); 
				}
					
			}
		} catch (SolrServerException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
	}
		
		    private static int fetchSize = 1000;
		    private static HttpSolrServer solrCore;

		    public void TestDataBase() throws MalformedURLException
		    {
		         solrCore = new HttpSolrServer(URL);
		    }

		    /**
		     * Takes an SQL ResultSet and adds the documents to solr. Does it in batches
		     * of fetchSize.
		     * 
		     * @param rs
		     *            A ResultSet from the database.
		     * @return The number of documents added to solr.
		     * @throws SQLException
		     * @throws SolrServerException
		     * @throws IOException
		     */
		    public long addResultSet(ResultSet rs) throws SQLException,
		            SolrServerException, IOException
		    {
		        long count = 0;
		        int innerCount = 0;
		        Collection<SolrInputDocument> docs = new ArrayList<SolrInputDocument>();
		        ResultSetMetaData rsm = rs.getMetaData();
		        int numColumns = rsm.getColumnCount();
		        String[] colNames = new String[numColumns + 1];

		        /**
		         * JDBC numbers the columns starting at 1, so the normal java convention
		         * of starting at zero won't work.
		         */
		        for (int i = 1; i < (numColumns + 1); i++)
		        {
		            colNames[i] = rsm.getColumnName(i);
		            /**
		             * If there are fields that you want to handle manually, check for
		             * them here and change that entry in colNames to null. This will
		             * cause the loop in the next section to skip that database column.
		             */
		            // //Example:
		            // if (rsm.getColumnName(i) == "db_id")
		            // {
		            // colNames[i] = null;
		            // }
		        }

		        while (rs.next())
		        {
		            count++;
		            innerCount++;

		            SolrInputDocument doc = new SolrInputDocument();

		            /**
		             * At this point, take care of manual document field assignments for
		             * which you previously assigned the colNames entry to null.
		             */
		            // //Example:
		            // doc.addField("solr_db_id", rs.getLong("db_id"));

		            for (int j = 1; j < (numColumns + 1); j++)
		            {
		                if (colNames[j] != null)
		                {
		                    Object f;
		                    switch (rsm.getColumnType(j))
		                    {
		                        case Types.BIGINT:
		                        {
		                            f = rs.getLong(j);
		                            break;
		                        }
		                        case Types.INTEGER:
		                        {
		                            f = rs.getInt(j);
		                            break;
		                        }
		                        case Types.DATE:
		                        {
		                            f = rs.getDate(j);
		                            break;
		                        }
		                        case Types.FLOAT:
		                        {
		                            f = rs.getFloat(j);
		                            break;
		                        }
		                        case Types.DOUBLE:
		                        {
		                            f = rs.getDouble(j);
		                            break;
		                        }
		                        case Types.TIME:
		                        {
		                            f = rs.getDate(j);
		                            break;
		                        }
		                        case Types.BOOLEAN:
		                        {
		                            f = rs.getBoolean(j);
		                            break;
		                        }
		                        default:
		                        {
		                            f = rs.getString(j);
		                        }
		                    }
		                    doc.addField(colNames[j], f);
		                }
		            }
		            docs.add(doc);

		            /**
		             * When we reach fetchSize, index the documents and reset the inner
		             * counter.
		             */
		            if (innerCount == fetchSize)
		            {
		                solrCore.add(docs);
		                docs.clear();
		                innerCount = 0;
		            }
		        }

		        /**
		         * If the outer loop ended before the inner loop reset, index the
		         * remaining documents.
		         */
		        if (innerCount != 0)
		        {
		            solrCore.add(docs);
		        }
		        return count;
		    }
		    
		    
		    
}

 

 

你可能感兴趣的:(solrj)