package com.jadyer.solrj;
import java.util.ArrayList;
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.SolrInputDocument;
/**
* 【Lucene3.6.2入门系列】第15节_SolrJ高亮
* @create Aug 20, 2013 1:50:38 PM
* @author 玄玉<http://blog.csdn.net/jadyer>
*/
public enum HelloSolrJHighLighter {
INSTANCE;
private SolrServer server;
private HelloSolrJHighLighter(){
//创建SolrServer对象,它有两个线程安全的子类EmbeddedSolrServer和CommonsHttpSolrServer
//EmbeddedSolrServer-----内嵌式的,不需要启动Web服务器
//CommonsHttpSolrServer--需要启动Web服务器,它是通过HTTP请求来获取数据的
//CommonsHttpSolrServer已不推荐直接使用了,而是推荐使用HttpSolrServer类
//同样的StreamingUpdateSolrServer也不推荐使用了,推荐使用ConcurrentUpdateSolrServer
server = new HttpSolrServer("http://127.0.0.1:8088/solr");
//添加索引
this.addIndex();
}
/**
* 添加索引
* @see 这里之所以没有硬编码指定my_title和my_content的中文分词器
* @see 是由于schema.xml中在定义这俩<field>时已通过type属性指定了中文分词器
* @see 关于指定中文分词器的具体方法,详见http://blog.csdn.net/jadyer/article/details/10106011
*/
private void addIndex(){
try {
//添加之前先删除全部索引,删除完记得commit才能生效
server.deleteByQuery("*:*");
server.commit();
} catch (Exception e) {
e.printStackTrace();
}
List<SolrInputDocument> docs = new ArrayList<SolrInputDocument>();
for(int i=1; i<4; i++){
SolrInputDocument doc = new SolrInputDocument();
//对于Solr而言,默认的id是唯一的主键(如果我们没有自定义主键的话),当多次添加的时候,最后添加的相同id的域会覆盖已有的
doc.addField("id", Integer.toString(i));
//这里使用的是自定义的Field,所以就需要在schema.xml中为其添加<field>描述,并通过type属性指定其使用的分词器
doc.addField("my_title", "这是我的第" + i + "个SolrJ程序");
doc.addField("my_content", "my第" + i + "个SolrJ程序的运行情况");
docs.add(doc);
}
try {
//可以只添加一个Document,也可以一次添加多个Document,这里添加多个
server.add(docs);
//commit后方能生效,否则搜索到的还是add前的索引信息
server.commit();
} catch (Exception e) {
e.printStackTrace();
}
}
/**
* 搜索高亮
*/
public void testHighLighter(){
SolrQuery query = new SolrQuery("my_title:SolrJ");
query.setHighlight(true).setHighlightSimplePre("<span style='color:red'>").setHighlightSimplePost("</span>");
//设置高亮的区域,可设置多个区域(但有个前提-->待高亮的信息必须是被存储的,即schema.xml中配置<field store="true">)
query.setParam("hl.fl", "my_title, my_content");
QueryResponse resp = null;
try {
resp = server.query(query);
} catch (SolrServerException e) {
e.printStackTrace();
}
for(SolrDocument sd : resp.getResults()){
Map<String, List<String>> snippetMap = resp.getHighlighting().get(sd.getFieldValue("id"));
if(snippetMap!=null && !snippetMap.isEmpty()){
List<String> snippetList = snippetMap.get("my_title");
for(String snippet : snippetList){
System.out.print(snippet);
}
System.out.print(" ");
//注意,如果在配置<field my_content multiValued="true"/>时指定其支持多值域
//那么,此处获取到的就会是第一个值域的值
snippetList = snippetMap.get("my_content");
for(String snippet : snippetList){
System.out.println(snippet);
}
}
}
}
/**
* 测试一下SolrJ的高亮效果
* @see 测试前记得启动Web服务器
* @see 关于Solr与Tomcat的整合,详见http://blog.csdn.net/jadyer/article/details/10104077
*/
public static void main(String[] args) {
HelloSolrJHighLighter.INSTANCE.testHighLighter();
}
}
