forked from apache/ranger
-
Notifications
You must be signed in to change notification settings - Fork 0
/
SolrClient.java
30 lines (23 loc) · 1.05 KB
/
SolrClient.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
package org.apache.ranger.index.solr.client.solrj;
import org.apache.ranger.index.solr.client.solrj.response.UpdateResponse;
import org.apache.ranger.index.solr.common.SolrInputDocument;
import org.apache.solr.client.solrj.SolrServerException;
import java.io.IOException;
import java.util.Collection;
import java.util.stream.Collectors;
public class SolrClient extends org.apache.ranger.index.solr.SolrClient {
public final org.apache.solr.client.solrj.SolrClient inner;
public SolrClient(org.apache.solr.client.solrj.SolrClient inner) {
super(inner);
this.inner = inner;
}
public UpdateResponse add(Collection<SolrInputDocument> docs) throws IOException, SolrServerException {
return new UpdateResponse(inner.add(docs.stream().map(x->x.inner).collect(Collectors.toList())));
}
public void close() throws IOException {
inner.close();
}
public UpdateResponse add(SolrInputDocument document) throws IOException, SolrServerException {
return new UpdateResponse(inner.add(document.inner));
}
}