/*
* This is eMonocot, a global online biodiversity information resource.
*
* Copyright © 2011–2015 The Board of Trustees of the Royal Botanic Gardens, Kew and The University of Oxford
*
* eMonocot is free software: you can redistribute it and/or modify it under the terms of the
* GNU Affero General Public License as published by the Free Software Foundation, either version 3
* of the License, or (at your option) any later version.
*
* eMonocot is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even
* the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* The complete text of the GNU Affero General Public License is in the source repository as the file
* ‘COPYING’. It is also available from <http://www.gnu.org/licenses/>.
*/
package org.emonocot.harvest.common;
import java.util.HashMap;
import java.util.Map;
import org.apache.solr.client.solrj.SolrServerException;
import org.apache.solr.common.SolrDocument;
import org.emonocot.api.SearchableObjectService;
import org.springframework.batch.item.database.AbstractPagingItemReader;
import org.springframework.beans.factory.annotation.Autowired;
public class SolrDocumentItemReader extends AbstractPagingItemReader<SolrDocument> {
private SearchableObjectService searchableObjectService;
private String queryString = null;
private Map<String,String> selectedFacets = new HashMap<String,String>();
private String sort = null;
public void setSort(String sort) {
this.sort = sort;
}
public void setQueryString(String queryString) {
this.queryString = queryString;
}
public void setSelectedFacets(String[] selectedFacets) {
if(selectedFacets != null) {
for(String selectedFacet : selectedFacets) {
String[] f = selectedFacet.split("\\=");
this.selectedFacets.put(f[0],f[1]);
}
}
}
@Autowired
public void setSearchableObjectService(SearchableObjectService searchableObjectService) {
this.searchableObjectService = searchableObjectService;
}
@Override
protected void doReadPage() {
try {
results = searchableObjectService.searchForDocuments(queryString, getPageSize(), getPage(), selectedFacets, sort).getRecords();
} catch (SolrServerException sse) {
throw new RuntimeException("SolrServerException", sse);
}
logger.debug("Search for " + queryString + " (page number " + getPage() + " got a page of " + results.size()
+ " (results");
}
@Override
protected void doJumpToPage(int itemIndex) {
}
}