solr开发,提交索引数据的几种方式
今天抽空学习了一下solr,有新东西学习就是哈皮! 期待能有机会与实战.实例仅为个人理解学习实例.
提交到Solr服务器上的数据必须是 SolrInputDocument 类型.
方案一:利用反射,自定义转换机制
JavaBean对象转换成SolrInputDocument的方式我们可以自定义.如果我们的Model类符合JavaBean的规范,那么我们可以使用反射获取其内部信息.
/**
* @author dennisit@163.com
*
* 实体类与SolrInputDocument转换 [测试通过]
*
* @param obj
* 实体对象
* @return SolrInputDocument
* SolrInputDocument对象
*/
public static SolrInputDocument entity2SolrInputDocument(Object obj) {
if (obj != null) {
Class<?> cls = obj.getClass();
Field[] filedArrays = cls.getDeclaredFields(); //获取类中所有属性
Method m = null;
SolrInputDocument sid = new SolrInputDocument();
for (Field f : filedArrays) {
//因为如果对象序列化之后,会增加该属性,不用对该属性进行反射
if(!f.getName().equals("serialVersionUID")){
try {
//跟进属性xx构造对应的getXx()方法
String dynamicGetMethod = dynamicMethodName(f.getName(), "get");
//调用构造的getXx()方法
m = cls.getMethod(dynamicGetMethod);
//属性名,与对应的属性值 get方法获取到的值
LOG.info(f.getName() + ":" + m.invoke(obj));
sid.addField(""+ f.getName(), m.invoke(obj));
} catch (IllegalArgumentException e) {
e.printStackTrace();
} catch (IllegalAccessException e) {
e.printStackTrace();
} catch (SecurityException e) {
e.printStackTrace();
} catch (InvocationTargetException e) {
e.printStackTrace();
} catch (NoSuchMethodException e) {
e.printStackTrace();
}
}
}
return sid;
}
LOG.warn("Object to convert is null.");
return null;
}
此时我们可以使用server中提供的添加solrInputDocument对象的方法进行数据建立索引.
/**
* 将对象集合添加至索引[测试通过]
* @author pudongping
* @param server
* @param lists 自定义转换机制
*/
public static <T> void addConvertBeans(SolrServer server, List<T> lists){
try {
server.add(EntityConvert.entityList2SolrInputDocument(lists));
server.commit(false, false);
LOG.info("Add convert object list to index finished. ");
} catch (Exception e) {
LOG.error("Add convert object list to index error, " + e.getMessage(), e);
}
}
/**
* 将对象集合添加至索引[测试通过]
* @author pudongping
* @param server
* @param lists 自定义转换机制
*/
public static <T> void addConvertBeans(SolrServer server, List<T> lists){
try {
server.add(EntityConvert.entityList2SolrInputDocument(lists));
server.commit(false, false);
LOG.info("Add convert object list to index finished. ");
} catch (Exception e) {
LOG.error("Add convert object list to index error, " + e.getMessage(), e);
}
}
测试自定义转换对象建立索引.
@Test
public void addConvertBean(){
Article article1 = new Article();
article1.setId(UUID.randomUUID().toString());
List<String> title = new ArrayList<String>();
title.add("我在门后假装你人还没走");
article1.setTitle(title);
article1.setAuthor("周杰伦");
SolrEngineHandler.addConvertBean(server, article1);
}
@Test
public void addConvertBeans(){
List<Article> lists = new ArrayList<Article>();
Article art1 = new Article();
art1.setId(UUID.randomUUID().toString());
List<String> atlts1 = new ArrayList<String>();
atlts1.add("谁言别后终无悔，寒月清宵绮梦回");
atlts1.add("深知身在情长在，前尘不共彩云飞");
art1.setTitle(atlts1);
art1.setAuthor("柳梦璃");
lists.add(art1);
Article art2 = new Article();
art2.setId(UUID.randomUUID().toString());
List<String> atlts2 = new ArrayList<String>();
atlts2.add("萧萧衷曲无处诉；为伊故，乐所苦。");
atlts2.add("锦色芳华，岂堪人虚度？欲寄相思情万缕，捎不到，君心处。");
art2.setTitle(atlts2);
art2.setAuthor("柳梦璃");
lists.add(art2);
SolrEngineHandler.addConvertBeans(server, lists);
}
方法二: 使用solr提供的注解机制
说明: @Field无参数时,匹配当前字段,也可以自定义,字段必须在schema.xml 中的Filed中存在.
package com.test.model;
import java.io.Serializable;
import java.util.List;
import org.apache.commons.lang.builder.ToStringBuilder;
import org.apache.solr.client.solrj.beans.Field;
public class Article implements Serializable{
/**
*
*/
private static final long serialVersionUID = 4017316764889231758L;
@Field("id")
private String id;
@Field("title")
private List<String> title;
@Field
private String author; //@Field无参数时,匹配当前字段
public String getId() {
return id;
}
public void setId(String id) {
this.id = id;
}
public String getAuthor() {
return author;
}
public void setAuthor(String author) {
this.author = author;
}
public List<String> getTitle() {
return title;
}
public void setTitle(List<String> title) {
this.title = title;
}
@Override
public String toString() {
return ToStringBuilder.reflectionToString(this);
}
}
package com.test.model;
import java.io.Serializable;
import java.util.List;
import org.apache.commons.lang.builder.ToStringBuilder;
import org.apache.solr.client.solrj.beans.Field;
public class Article implements Serializable{
/**
*
*/
private static final long serialVersionUID = 4017316764889231758L;
@Field("id")
private String id;
@Field("title")
private List<String> title;
@Field
private String author; //@Field无参数时,匹配当前字段
public String getId() {
return id;
}
public void setId(String id) {
this.id = id;
}
public String getAuthor() {
return author;
}
public void setAuthor(String author) {
this.author = author;
}
public List<String> getTitle() {
return title;
}
public void setTitle(List<String> title) {
this.title = title;
}
@Override
public String toString() {
return ToStringBuilder.reflectionToString(this);
}
}
使用solrJ提供的addBean(..)与addBeans(…)相关的方法添加对象
/**
* 将单个对象添加至索引
* @author pudongping
* @param server
* @param object solr机制进行对象转换
*/
public static void addBean(SolrServer server, Object object){
try {
server.addBean(object);
server.commit(false,false);
LOG.info("Add object to index finished.");
} catch (Exception e) {
LOG.error("Add object to index error, " + e.getMessage(), e);
e.printStackTrace();
}
}
/**
* 添加集合对象至索引 [测试通过]
* @author pudongping
* @param <T>
* @param server
* @param lists 使用solr内部转换机制
*/
public static <T> void addBeans(SolrServer server, List<T> lists){
try {
server.addBeans(lists);
server.commit(false, false);
LOG.info("Add object list to index finished. ");
} catch (Exception e) {
LOG.error("Add object list to index finished. ");
}
}
编写junit测试
@Test
public void addBean(){
Article article1 = new Article();
article1.setId(UUID.randomUUID().toString());
List<String> title = new ArrayList<String>();
title.add("地图青岛,末世孤岛.");
article1.setTitle(title);
article1.setAuthor("苏若年");
SolrEngineHandler.addBean(server, article1);
}
@Test
public void addBeans(){
List<Article> lists = new ArrayList<Article>();
Article art1 = new Article();
art1.setId(UUID.randomUUID().toString());
List<String> atlts1 = new ArrayList<String>();
atlts1.add("谁言别后终无悔，寒月清宵绮梦回");
atlts1.add("深知身在情长在，前尘不共彩云飞");
art1.setTitle(atlts1);
art1.setAuthor("苏若年");
lists.add(art1);
Article art2 = new Article();
art2.setId(UUID.randomUUID().toString());
List<String> atlts2 = new ArrayList<String>();
atlts2.add("萧萧衷曲无处诉；为伊故，乐所苦。");
atlts2.add("锦色芳华，岂堪人虚度？欲寄相思情万缕，捎不到，君心处。");
art2.setTitle(atlts2);
art2.setAuthor("柳梦璃");
lists.add(art2);
SolrEngineHandler.addBeans(server, lists);
}
热爱生活,热爱Coding,敢于挑战,用于探索 ...
