Metacat

import java.io.ByteArrayInputStream;

import java.io.ByteArrayOutputStream;

import java.io.InputStream;

import java.io.StringWriter;

import java.io.Writer;

import java.util.Collections;

import java.util.Date;

import javax.xml.parsers.DocumentBuilder;

import javax.xml.parsers.DocumentBuilderFactory;

import javax.xml.xpath.XPathConstants;

import javax.xml.xpath.XPathFactory;

import org.apache.solr.core.CoreContainer;

import org.apache.solr.core.SolrCore;

import org.apache.solr.request.LocalSolrQueryRequest;

import org.apache.solr.response.QueryResponseWriter;

import org.apache.solr.response.SolrQueryResponse;

import org.apache.solr.schema.DateField;

import org.apache.solr.schema.IndexSchema;

import org.apache.solr.schema.SchemaField;

import org.w3c.dom.Element;

import org.w3c.dom.NodeList;

import edu.ucsb.nceas.metacat.common.SolrQueryResponseTransformer;

import edu.ucsb.nceas.metacat.common.SolrQueryResponseWriterFactory;

    private static SolrCore solrCore = null;

    private static CoreContainer solrCoreContainer = null;

            CoreContainer solrCoreContainer = SolrServerFactory.getCoreContainer();

            String coreName = SolrServerFactory.getCollectionName();

            solrCore = solrCoreContainer.getCore(coreName);

        /*if(mergedDocuments != null) {

            for(SolrDoc doc : mergedDocuments) {

                ByteArrayOutputStream out = new ByteArrayOutputStream();

                doc.serialize(out, "UTF-8");

                String result = new String(out.toByteArray(), "UTF-8");

                System.out.println("after updated document===========================");

                System.out.println(result);

            }

        }*/

    private List<SolrDoc> getSolrDocs(List<String> ids) throws SolrServerException, IOException, ParserConfigurationException, SAXException, XPathExpressionException {

    private SolrDoc getSolrDoc(String id) throws SolrServerException, IOException, ParserConfigurationException, SAXException, XPathExpressionException {

           solrDoc = transformQueryResponseToSolrDoc(solrParams, response);

           /*if(solrDoc != null) {

               ByteArrayOutputStream out = new ByteArrayOutputStream();

               solrDoc.serialize(out, "UTF-8");

               String result = new String(out.toByteArray(), "UTF-8");

               System.out.println("need to be updated document ===========================");

               System.out.println(result);

           }*/

    /*

     * SolrDocuments. This method will transform the first SolrDocuments (in the Solr lib) to

     * the SolrDoc (in the d1_cn_index_processor lib).

    private SolrDoc transformQueryResponseToSolrDoc(SolrParams solrParams, QueryResponse response) throws SolrServerException, IOException, ParserConfigurationException, SAXException, XPathExpressionException {

        SolrDoc solrDoc = null;

        if(response != null) {

            SolrQueryResponseTransformer transformer = new SolrQueryResponseTransformer(solrCore);

            InputStream input = transformer.transformResults(solrParams, response, SolrQueryResponseWriterFactory.XML);

            DocumentBuilderFactory dbFactory = DocumentBuilderFactory.newInstance();

            DocumentBuilder dBuilder = dbFactory.newDocumentBuilder();

            Document doc = dBuilder.parse(input);

            solrDoc = parseResults(doc);

    /*

     * Parse the query result document. This method only choose the first one from a list.

     */

    private SolrDoc parseResults(Document document) throws XPathExpressionException {

        SolrDoc solrDoc = null;

        NodeList nodeList = (NodeList) XPathFactory.newInstance().newXPath()

                .evaluate("/response/result/doc", document, XPathConstants.NODESET);

        if(nodeList != null && nodeList.getLength() >0) {

            Element docElement = (Element) nodeList.item(0);

            solrDoc = parseDoc(docElement);

        }

        return solrDoc;

    }

    /*

     * Parse an element

     */

    private SolrDoc parseDoc(Element docElement) {

        List<String> validSolrFieldNames = getValidSchemaField();

        SolrDoc doc = new SolrDoc();

        doc.LoadFromElement(docElement, validSolrFieldNames);

        return doc;

    }

    /**

     * Get the valid schema fields from the solr server.

     * @return

     */

    private List<String> getValidSchemaField() {

        List<String> validSolrFieldNames = new ArrayList<String>();

        IndexSchema schema = solrCore.getSchema();

        Map<String, SchemaField> fieldMap = schema.getFields();

        Set<String> fieldNames = fieldMap.keySet();

        for(String fieldName : fieldNames) {

            SchemaField field = fieldMap.get(fieldName);

            //remove the field which is the target field of a CopyField.

            if(field != null && !schema.isCopyFieldTarget(field)) {

                 validSolrFieldNames.add(fieldName);

            }

        }

        //System.out.println("the valid file name is\n"+validSolrFieldNames);

        return validSolrFieldNames;

    }