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;
}
}
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
