1 | package eu.clarin.cmdi.vlo.pages; |
---|
2 | |
---|
3 | import java.util.Iterator; |
---|
4 | import java.util.List; |
---|
5 | |
---|
6 | import org.apache.solr.client.solrj.SolrQuery; |
---|
7 | import org.apache.solr.client.solrj.response.FacetField; |
---|
8 | import org.apache.wicket.extensions.markup.html.repeater.util.SortableDataProvider; |
---|
9 | import org.apache.wicket.model.IModel; |
---|
10 | import org.apache.wicket.model.Model; |
---|
11 | |
---|
12 | import eu.clarin.cmdi.vlo.dao.DaoLocator; |
---|
13 | import eu.clarin.cmdi.vlo.dao.SearchResultsDao; |
---|
14 | |
---|
15 | public class SolrFacetDataProvider extends SortableDataProvider<FacetField, String> { |
---|
16 | |
---|
17 | private static final long serialVersionUID = 1L; |
---|
18 | private final SolrQuery query; |
---|
19 | private List<FacetField> facets; |
---|
20 | |
---|
21 | public SolrFacetDataProvider(SolrQuery query) { |
---|
22 | this.query = query; |
---|
23 | query.setFacet(true).setStart(0).setRows(0); //only get facets |
---|
24 | } |
---|
25 | |
---|
26 | private SearchResultsDao getSearchResultsDao() { |
---|
27 | return DaoLocator.getSearchResultsDao(); |
---|
28 | } |
---|
29 | |
---|
30 | @Override |
---|
31 | public Iterator<? extends FacetField> iterator(long first, long count) { |
---|
32 | return facets.iterator(); |
---|
33 | } |
---|
34 | |
---|
35 | private List<FacetField> getFacets() { |
---|
36 | if (facets == null) { |
---|
37 | SearchResultsDao searchResultsDao = getSearchResultsDao(); |
---|
38 | facets = searchResultsDao.getFacets(query); |
---|
39 | } |
---|
40 | return facets; |
---|
41 | } |
---|
42 | |
---|
43 | @Override |
---|
44 | public IModel<FacetField> model(FacetField facetField) { |
---|
45 | return new Model<FacetField>(facetField); |
---|
46 | } |
---|
47 | |
---|
48 | @Override |
---|
49 | public long size() { |
---|
50 | return (int) getFacets().size(); |
---|
51 | } |
---|
52 | |
---|
53 | } |
---|