【Lucene3.6.2入门系列】第15节_SolrJ高亮

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();
	}
}