xc.mst.manager.record.DefaultBrowseRecordService.java Source code

Introduction

Here is the source code for xc.mst.manager.record.DefaultBrowseRecordService.java

Source

/**
 * Copyright (c) 2009 eXtensible Catalog Organization
 *
 * This program is free software; you can redistribute it and/or modify it under the terms of the MIT/X11 license. The text of the
 * license can be found at http://www.opensource.org/licenses/mit-license.php and copy of the license can be found on the project
 * website http://www.extensiblecatalog.org/.
 *
 */

package xc.mst.manager.record;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

import org.apache.log4j.Logger;
import org.apache.solr.client.solrj.SolrQuery;
import org.apache.solr.client.solrj.SolrServer;
import org.apache.solr.client.solrj.SolrServerException;
import org.apache.solr.client.solrj.response.FacetField;
import org.apache.solr.client.solrj.response.QueryResponse;
import org.apache.solr.common.SolrDocument;
import org.apache.solr.common.SolrDocumentList;

import xc.mst.bo.record.Record;
import xc.mst.bo.record.SolrBrowseResult;
import xc.mst.bo.service.ErrorCode;
import xc.mst.bo.service.Service;
import xc.mst.constants.Constants;
import xc.mst.dao.DatabaseConfigException;
import xc.mst.manager.BaseService;
import xc.mst.manager.IndexException;

/**
 * Browse for results using solr
 * 
 * @author Sharmila Ranganathan
 * 
 */
public class DefaultBrowseRecordService extends BaseService implements BrowseRecordService {

    /**
     * A reference to the logger for this class
     */
    static Logger log = Logger.getLogger(Constants.LOGGER_GENERAL);

    /**
     * Method to search the index for records
     * 
     * @param query
     *            Query to perform the search
     * @return Search results
     * @throws DatabaseConfigException
     */
    public SolrBrowseResult search(SolrQuery query) throws IndexException, DatabaseConfigException {

        SolrServer server = getMSTSolrService().getServer();
        SolrBrowseResult result = null;

        // Discard deleted records
        // query.addFilterQuery("deleted:false");

        if (log.isDebugEnabled()) {
            log.debug("Querying Solr server with query:" + query);
        }

        if (server == null) {
            log.error("Solr server is null. Check the path to solr folder.");
            throw new IndexException("Solr server is null. Check the path to solr folder.");
        }

        QueryResponse rsp = null;
        try {
            rsp = server.query(query);
        } catch (SolrServerException sse) {
            log.error("Exception occured while executing the query. Check the path to solr folder.", sse);
            throw new IndexException(sse.getMessage());
        }

        // Load the records in the SolrBrowseResilt object
        SolrDocumentList docs = rsp.getResults();

        RecordService recordService = (RecordService) config.getBean("RecordService");
        Iterator<SolrDocument> iteration = docs.iterator();
        List<Record> records = new ArrayList<Record>();

        while (iteration.hasNext()) {
            records.add(recordService.getRecordFieldsForBrowseFromDocument(iteration.next()));
        }

        // Load the facets in the SolrBrowseResult object
        List<FacetField> facets = rsp.getFacetFields();
        result = new SolrBrowseResult(records, facets);
        result.setTotalNumberOfResults(docs.getNumFound());

        return result;

    }

    /**
     * Get error description for this code and service
     * 
     * @param errorCode
     *            Error code
     * @param service
     *            Service which generated the error
     * @return Error if found
     * @throws DatabaseConfigException
     */
    public ErrorCode getError(String errorCode, Service service) throws DatabaseConfigException {

        return errorCodeDAO.getByErrorCodeAndService(errorCode, service);

    }
}