source: vlo/branches/vlo-3.3-oeaw/vlo-web-app/src/main/java/eu/clarin/cmdi/vlo/service/solr/impl/SearchResultsDaoImpl.java @ 6456

package eu.clarin.cmdi.vlo.service.solr.impl;

import eu.clarin.cmdi.vlo.service.solr.SearchResultsDao;
import eu.clarin.cmdi.vlo.config.VloConfig;
import java.util.List;
import org.apache.solr.client.solrj.SolrQuery;
import org.apache.solr.client.solrj.SolrServer;
import org.apache.solr.client.solrj.response.FacetField;
import org.apache.solr.client.solrj.response.FacetField.Count;
import org.apache.solr.client.solrj.response.QueryResponse;
import org.apache.solr.common.SolrDocumentList;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class SearchResultsDaoImpl extends SolrDaoImpl implements SearchResultsDao {

    private final static Logger logger = LoggerFactory.getLogger(SearchResultsDaoImpl.class);

    public SearchResultsDaoImpl(SolrServer solrServer, VloConfig config) {
        super(solrServer, config);
    }

    @Override
    public List<FacetField> getFacets(SolrQuery query) {
        final QueryResponse response = fireQuery(sanitise(query));
        final List<FacetField> facetFields = response.getFacetFields();
        if (logger.isDebugEnabled()) {
            if (facetFields.size() == 1) {
                final FacetField field = facetFields.get(0);
                logger.debug("Found facet field '{}' with {} values", field.getName(), field.getValueCount());
            } else {
                logger.debug("Found {} facet fields", facetFields.size());
            }
        }
        return facetFields;
    }
    
    
    @Override
    public java.util.List<FacetField> getFacets(SolrQuery query, SolrQuery completeFacetValList) {
        final QueryResponse response = fireQuery(sanitise(query));
        
        final List<FacetField> facetFields = getFacets(query);
        
        
        final QueryResponse completeList = fireQuery(sanitise(completeFacetValList));
        final FacetField excludedFacet = completeList.getFacetFields().get(0);
        FacetField newFacet = null;
        Integer ind = null;
        
        for(FacetField facet: facetFields){
                if(facet.getName().equals(excludedFacet.getName())){
                        logger.debug("Missing values for excluded Facet {} will be added", facet.getName());
                        newFacet = copyFacetField(facet);
                        ind = facetFields.indexOf(facet);
nextVal:                for(FacetField.Count exVal: excludedFacet.getValues()){
                                        boolean exists = false;
                                for(FacetField.Count val: facet.getValues()){
                                        if(exVal.getName().equals(val.getName())){
                                                exists = true;
                                                break;
                                        }
                                }
                                //import missing values into the facet from the list and set count to zero
                                if(!exists){
                                        logger.debug("Missing value {} is added with count 0", exVal.getName());
                                        newFacet.add(exVal.getName(), 0);
                                }
                        }
                }
        }
        
        if(ind != null){
                facetFields.remove(ind);
                facetFields.add(ind, newFacet);
        }
        
        return facetFields;
    }
    
    private FacetField copyFacetField(FacetField from){
        FacetField newFacet = new FacetField(from.getName(), from.getGap(), from.getEnd());
        for(FacetField.Count val: from.getValues())
                newFacet.add(val.getName(), val.getCount());
        
        return newFacet;
    }
    

    @Override
    public SolrDocumentList getDocuments(SolrQuery query) {
        QueryResponse queryResponse = fireQuery(query);
        final SolrDocumentList documents = queryResponse.getResults();
        logger.debug("Found {} documents", documents.getNumFound());
        return documents;
    }
    
}