显示目录

Java 访问 Solr

可以使用 SolrJ 来访问 Solr,其提供的 SolrClient 能够在连接断开后自动重连。

Gradle 依赖

1
compile 'org.apache.solr:solr-solrj:6.0.0'

连接 Solr

1
2
// ebag 是 Core 或者 Collection
SolrClient solr = new HttpSolrClient("http://localhost:8983/solr/ebag");

插入

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
/**
* 创建或者更新 document:
* 如果有存在 document 的 id 和输入的 id 相同, 则删除 document, 然后创建一个新的
* 如果没有 document 的 id 和输入的 id 相同, 则创建
* @throws IOException
* @throws SolrServerException
*/
public void insert() throws IOException, SolrServerException {
SolrInputDocument document = new SolrInputDocument();
document.setField("id", 10000); // 如果没有指定 id 属性, 则 Solr 会自动创建一个
document.setField("name", "Alice");
document.setField("age", 23);
solr.add(document);
solr.commit();
}

删除

1
2
3
4
public void delete() throws IOException, SolrServerException {
solr.deleteById("10000");
solr.commit();
}

查询

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
public void select() throws IOException, SolrServerException {
SolrQuery query = new SolrQuery();
query.set("wt", "json"); // 设置查询参数
query.set("q", "name:Alice");
QueryResponse response = solr.query(query);
SolrDocumentList documents = response.getResults();
System.out.println("Num Found: " + documents.getNumFound());
for (SolrDocument document : documents) {
System.out.println("---------------------------------------");
Collection<String> fieldNames = document.getFieldNames();
for (String fieldName : fieldNames) {
System.out.println(fieldName + " : " + document.get(fieldName));
}
}
}

插入 Bean

前面使用 document.set(field, value) 来创建 document,如果属性太多时,就容易出错,可以使用 Bean 的方式来进行插入。先定义一个 Java Bean,需要插入到 Solr 的属性给其加上 @Field 注解,然后使用这个 Bean 创建对象 bean,调用 solr.addBean(bean) 插入到 Solr 里。

1
2
3
4
5
6
7
8
9
10
public void insertBean() throws IOException, SolrServerException {
Person p1 = new Person("person-10001", "Bob", 34);
Person p2 = new Person("person-10002", "John", 88);
Person p3 = new Person("person-10003", "Alice", 88);
solr.addBean(p1);
solr.addBean(p2);
solr.addBean(p3);
solr.commit();
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
import org.apache.solr.client.solrj.beans.Field;
public class Person {
@Field
private String id;
@Field
private String name;
@Field
private int age;
public Person() {}
public Person(String id, String name, int age) {
this.id = id;
this.name = name;
this.age = age;
}
// Getters and setters
}