Metacat

/**

 *

 * This program is free software; you can redistribute it and/or modify

 * it under the terms of the GNU General Public License as published by

 * the Free Software Foundation; either version 2 of the License, or

 * (at your option) any later version.

 *

 * This program 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 General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

 */

package edu.ucsb.nceas.metacat.index;

import java.io.InputStream;

import java.util.ArrayList;

import java.util.Iterator;

import java.util.Set;

import javax.xml.parsers.DocumentBuilder;

import javax.xml.parsers.DocumentBuilderFactory;

import org.apache.commons.io.output.ByteArrayOutputStream;

import org.apache.commons.logging.LogFactory;

import org.apache.solr.common.SolrDocumentList;

import org.dataone.cn.indexer.resourcemap.ResourceMap;

import org.dataone.cn.indexer.solrhttp.SolrElementField;

import org.dataone.service.exceptions.NotImplemented;

import org.xml.sax.SAXException;

 * A class does insert, update and remove indexes to a SOLR server

 * @author tao

 *

 */

public class SolrIndex {

    private static final String IDQUERY = ID+":*";

    private SolrServer solrServer = null;

    private XMLNamespaceConfig xmlNamespaceConfig = null;

    private static DocumentBuilderFactory documentBuilderFactory = null;

    private static DocumentBuilder builder = null;

    private static XPathFactory xpathFactory = null;

    private static XPath xpath = null;

        documentBuilderFactory = DocumentBuilderFactory.newInstance();

        documentBuilderFactory.setNamespaceAware(true);

        try {

            builder = documentBuilderFactory.newDocumentBuilder();

        } catch (ParserConfigurationException e) {

            e.printStackTrace();

        }

        xpathFactory = XPathFactory.newInstance();

        xpath = xpathFactory.newXPath();

    }

     * Constructor

     * @throws IOException

         this.xmlNamespaceConfig = xmlNamespaceConfig;

         this.sysmetaSolrFields = sysmetaSolrFields;

         init();

        xpath.setNamespaceContext(xmlNamespaceConfig);

        initExpressions();

    }

    private void initExpressions() throws XPathExpressionException {

        for (SolrField field : sysmetaSolrFields) {

            field.initExpression(xpath);

        }

    }

     * Get the list of the Subprocessors in this index.

     * @return the list of the Subprocessors.

     */

    public List<IDocumentSubprocessor> getSubprocessors() {

        return subprocessors;

    }

    /**

     * Set the list of Subprocessors.

     * @param subprocessorList  the list will be set.

     */

    public void setSubprocessors(List<IDocumentSubprocessor> subprocessorList) {

    }

    /**

     * Generate the index for the given information

     * @param id

     * @return

     * @throws IOException

     * @throws SAXException

     * @throws ParserConfigurationException

     * @throws XPathExpressionException

     * @throws NotFound

     * @throws NotImplemented

        // Load the System Metadata document

        TypeMarshaller.marshalTypeToOutputStream(systemMetadata, systemMetadataOutputStream);

        ByteArrayInputStream systemMetadataStream = new ByteArrayInputStream(systemMetadataOutputStream.toByteArray());

        Document sysMetaDoc = generateXmlDocument(systemMetadataStream);

            log.error("Could not load System metadata for ID: " + id);

            return null;

        }

        // Extract the field values from the System Metadata

        List<SolrElementField> sysSolrFields = processSysmetaFields(sysMetaDoc, id);

        SolrDoc indexDocument = new SolrDoc(sysSolrFields);

        Map<String, SolrDoc> docs = new HashMap<String, SolrDoc>();

        docs.put(id, indexDocument);

        // Determine if subprocessors are available for this ID

        if (subprocessors != null) {

                    // for each subprocessor loaded from the spring config

                    for (IDocumentSubprocessor subprocessor : subprocessors) {

                        // Does this subprocessor apply?

                        if (subprocessor.canProcess(sysMetaDoc)) {

                            // if so, then extract the additional information from the

                            // document.

                            try {

                                // docObject = the resource map document or science

                                // metadata document.

                                // note that resource map processing touches all objects

                                // referenced by the resource map.

                                if (docObject == null) {

                                    docs = subprocessor.processDocument(id, docs, docObject);

                                }

                            } catch (Exception e) {

                        }

                    }

       }

       // TODO: in the XPathDocumentParser class in d1_cn_index_process module,

       // merge is only for resource map. We need more work here.

       for (SolrDoc mergeDoc : docs.values()) {

           if (!mergeDoc.isMerged()) {

       }

       //SolrElementAdd addCommand = getAddCommand(new ArrayList<SolrDoc>(docs.values()));

       return docs;

    }

     * Merge updates with existing solr documents

     *

     * This method appears to re-set the data package field data into the

     * document about to be updated in the solr index. Since packaging

     * information is derived from the package document (resource map), this

     * information is not present when processing a document contained in a data

     * package. This method replaces those values from the existing solr index

     * record for the document being processed. -- sroseboo, 1-18-12

     *

     * @param indexDocument

     * @return

     * @throws IOException

     * @throws EncoderException

     * @throws XPathExpressionException

     * @throws SAXException

     * @throws ParserConfigurationException

     * @throws SolrServerException

     * @throws NotFound

     * @throws NotImplemented

    // TODO:combine merge function with resourcemap merge function

    private SolrDoc mergeWithIndexedDocument(SolrDoc indexDocument) throws IOException,

        ids.add(indexDocument.getIdentifier());

        List<SolrDoc> indexedDocuments = ResourceMapSubprocessor.getSolrDocs(ids);

        SolrDoc indexedDocument = indexedDocuments == null || indexedDocuments.size() <= 0 ? null

                : indexedDocuments.get(0);

        IndexSchema indexSchema = SolrQueryServiceController.getInstance().getSchema();

            return indexDocument;

        } else {

            for (SolrElementField field : indexedDocument.getFieldList()) {

                if ((field.getName().equals(SolrElementField.FIELD_ISDOCUMENTEDBY)

                        || field.getName().equals(SolrElementField.FIELD_DOCUMENTS) || field

                        .getName().equals(SolrElementField.FIELD_RESOURCEMAP))

                        && !indexDocument.hasFieldWithValue(field.getName(), field.getValue())) {

                    indexDocument.addField(field);

                    indexDocument.addField(field);

            }

            indexDocument.setMerged(true);

            return indexDocument;

        }

    }

     * Generate a Document from the InputStream

     */

    private Document generateXmlDocument(InputStream smdStream) throws SAXException {

        Document doc = null;

        try {

            doc = builder.parse(smdStream);

        } catch (IOException e) {

            log.error(e.getMessage(), e);

        }

        return doc;

    }

    /*

     * Index the fields of the system metadata

     */

    private List<SolrElementField> processSysmetaFields(Document doc, String identifier) {

        List<SolrElementField> fieldList = new ArrayList<SolrElementField>();

        // solrFields is the list of fields defined in the application context

        for (SolrField field : sysmetaSolrFields) {

            try {

                // the field.getFields method can return a single value or

                // multiple values for multi-valued fields

                // or can return multiple SOLR document fields.

                fieldList.addAll(field.getFields(doc, identifier));

            } catch (Exception e) {

                e.printStackTrace();

            }

        }

        return fieldList;

    }

    /**

     * @param pid

     * @param systemMetadata

     * @param data

     * @throws SolrServerException

     */

        }

        if(systemMetadata == null) {

    }

    /**

     * @param systemMetadata  the system metadata associated with the data object

     * @throws NotFound

     * @throws NotImplemented

        Map<String, SolrDoc> docs = process(pid.getValue(), systemMetadata, objectPath);

        //transform the Map to the SolrInputDocument which can be used by the solr server

        if(docs != null) {

            Set<String> ids = docs.keySet();

            for(String id : ids) {

                if(id != null) {

                    SolrDoc doc = docs.get(id);

                }

            }

        }

    }

     * Adds the given fields to the solr index for the given pid, preserving the index values

     * that previously existed

     * @param pid

     * @param fields

     */

    public void insertFields(Identifier pid, Map<String, List<Object>> fields) {

    	try {

			// copy the original values already indexed for this document

	    	SolrQuery query = new SolrQuery("id:\"" + pid.getValue() + "\"");

	    	QueryResponse res = solrServer.query(query);

	    	SolrDoc doc = new SolrDoc();

	    	// include existing values if they exist

		        SolrDocument orig = res.getResults().get(0);

		    	for (String fieldName: orig.getFieldNames()) {

		        	//  don't transfer the copyTo fields, otherwise there are errors

		        	if (indexSchema.isCopyFieldTarget(indexSchema.getField(fieldName))) {

		        		continue;

		        	}

		        	for (Object value: orig.getFieldValues(fieldName)) {

		        		String stringValue = value.toString();

		        		// special handling for dates in ISO 8601

		        		if (value instanceof Date) {

		        			stringValue = DateTimeMarshaller.serializeDateToUTC((Date)value);

		        			SolrDateConverter converter = new SolrDateConverter();

		        			stringValue = converter.convert(stringValue);

		        		}

						SolrElementField field = new SolrElementField(fieldName, stringValue);

						log.debug("Adding field: " + fieldName);

						doc.addField(field);

		        	}

		        }

	        // add the additional fields we are trying to include in the index

	        for (String fieldName: fields.keySet()) {

	    		List<Object> values = fields.get(fieldName);

	    		for (Object value: values) {

	    					doc.addField(new SolrElementField(fieldName, value.toString()));

	    				} else {

	    	    	    	doc.updateOrAddField(fieldName, value.toString());

	    				}

	    	}

	        if (!doc.hasField(ID)) {

	        	doc.updateOrAddField(ID, pid.getValue());

	        }

	        insertToIndex(doc);

    	} catch (Exception e) {

    		String error = "SolrIndex.insetFields - could not update the solr index: " + e.getMessage();

            writeEventLog(null, pid, error);

            log.error(error, e);

    	}

    }

     * Insert a SolrDoc to the solr server.

     */

    private synchronized void insertToIndex(SolrDoc doc) throws SolrServerException, IOException {

        if(doc != null ) {

            SolrInputDocument solrDoc = new SolrInputDocument();

            List<SolrElementField> list = doc.getFieldList();

            if(list != null) {

                //solrDoc.addField(METACATPIDFIELD, pid);

                Iterator<SolrElementField> iterator = list.iterator();

                while (iterator.hasNext()) {

                    SolrElementField field = iterator.next();

                    if(field != null) {

                        String value = field.getValue();

                        String name = field.getName();

                        //System.out.println("add name/value pair - "+name+"/"+value);

                        solrDoc.addField(name, value);

                }

            }

                Identifier pid = new Identifier();

                pid.setValue(doc.getIdentifier());

                    solrServer.commit();

                    try {

                        EventlogFactory.createIndexEventLog().write(event);

                    } catch (Exception e) {

                    try {

                        EventlogFactory.createIndexEventLog().write(event);

                    } catch (Exception ee) {

                } catch (IOException e) {

                    try {

                        EventlogFactory.createIndexEventLog().write(event);

                    } catch (Exception ee) {

            }

    }

    /**

        String objectPath = null;

        try {

            objectPath = DistributedMapsFactory.getObjectPathMap().get(pid);

        } catch (Exception e) {

            String error = "SolrIndex.update - could not update the solr index since " + e.getMessage();

            writeEventLog(systemMetadata, pid, error);

            log.error(error, e);

        }

     * Update the solr index. This method handles the three scenarios:

     * 1. Remove an existing doc - if the the system metadata shows the value of the archive is true,

     *    remove the index for the previous version(s) and generate new index for the doc.

     * 2. Add a new doc - if the system metadata shows the value of the archive is false, generate the

     *    index for the doc.

     * @param pid

     * @param systemMetadata

     * @param data

     * @throws SolrServerException

     * @throws ServiceFailure

     * @throws XPathExpressionException

     * @throws NotImplemented

     * @throws NotFound

     * @throws UnsupportedType

     * @throws IOException

     * @throws SAXException

     * @throws ParserConfigurationException

     * @throws OREParserException

     * @throws JiBXException

     * @throws EncoderException

     */

                                IOException, SAXException, ParserConfigurationException, OREParserException, JiBXException, EncoderException {

            //delete the index for the archived objects

            remove(pid.getValue(), systemMetadata);

            log.info("SolrIndex.update============================= archive the idex for the identifier "+pid);

        } else {

            //generate index for either add or update.

        }

    }

    /*

     * Is the pid a resource map

     */

        return isDataPackage;

    }

    private boolean isPartOfDataPackage(String pid) throws XPathExpressionException, NotImplemented, NotFound, UnsupportedType, SolrServerException, IOException, ParserConfigurationException, SAXException {

        SolrDoc dataPackageIndexDoc = ResourceMapSubprocessor.getSolrDoc(pid);

        if (dataPackageIndexDoc != null) {

            String resourceMapId = dataPackageIndexDoc

                    .getFirstFieldValue(SolrElementField.FIELD_RESOURCEMAP);

            return StringUtils.isNotEmpty(resourceMapId);

        } else {

            return false;

        }

    }

     * Remove the indexed associated with specified pid.

     * @param pid  the pid which the indexes are associated with

     * @throws IOException

     * @throws SolrServerException

     * @throws ParserConfigurationException

     * @throws SAXException

     * @throws UnsupportedType

     * @throws NotFound

     * @throws NotImplemented

     * @throws XPathExpressionException

     * @throws ServiceFailure

     * @throws OREParserException

     */

    private void remove(String pid, SystemMetadata sysmeta) throws IOException, SolrServerException, ServiceFailure, XPathExpressionException, NotImplemented, NotFound, UnsupportedType, SAXException, ParserConfigurationException, OREParserException {

        if (isDataPackage(pid, sysmeta)) {

            removeDataPackage(pid);

        } else if (isPartOfDataPackage(pid)) {

            removeFromDataPackage(pid);

        } else {

            removeFromIndex(pid);

        }

    }

    /*

     * Remove a resource map pid

     */

    private void removeDataPackage(String pid) throws ServiceFailure, SAXException, XPathExpressionException, NotImplemented, NotFound, UnsupportedType, SolrServerException, IOException, ParserConfigurationException, OREParserException  {

        Document resourceMapDoc = generateXmlDocument(DistributedMapsFactory.getDataObject(pid));

        //ResourceMap resourceMap = new ResourceMap(resourceMapDoc);

        ResourceMap resourceMap = ResourceMapFactory.buildResourceMap(resourceMapDoc);

        List<String> documentIds = resourceMap.getAllDocumentIDs();

        List<SolrDoc> indexDocuments =ResourceMapSubprocessor.getSolrDocs(documentIds);

        removeFromIndex(pid);

        //List<SolrDoc> docsToUpdate = new ArrayList<SolrDoc>();

        // for each document in data package:

        for (SolrDoc indexDoc : indexDocuments) {

                continue; // skipping the resource map, no need update

                          // it.

                          // will

                          // be removed.

            }

            // Remove resourceMap reference

            indexDoc.removeFieldsWithValue(SolrElementField.FIELD_RESOURCEMAP,

                    resourceMap.getIdentifier());

            // // Remove documents/documentedby values for this resource

            // map

            for (ResourceEntry entry : resourceMap.getMappedReferences()) {

                if (indexDoc.getIdentifier().equals(entry.getIdentifier())) {

                    for (String documentedBy : entry.getDocumentedBy()) {

                        // Using removeOneFieldWithValue in-case same

                        // documents

                        // are in more than one data package. just

                        // remove

                        // one

                        // instance of data package info.

                        indexDoc.removeOneFieldWithValue(SolrElementField.FIELD_ISDOCUMENTEDBY,

                                documentedBy);

                    }

                    for (String documents : entry.getDocuments()) {

                        indexDoc.removeOneFieldWithValue(SolrElementField.FIELD_DOCUMENTS,

                                documents);

                    }

                    break;

                }

            }

            removeFromIndex(indexDoc.getIdentifier());

            insertToIndex(indexDoc);

            //docsToUpdate.add(indexDoc);

        }

        //SolrElementAdd addCommand = new SolrElementAdd(docsToUpdate);

        //httpService.sendUpdate(solrIndexUri, addCommand);

    }

     * Remove a pid which is part of resource map.

     */

        SolrDoc indexedDoc = ResourceMapSubprocessor.getSolrDoc(pid);

        removeFromIndex(pid);

        List<SolrDoc> docsToUpdate = new ArrayList<SolrDoc>();

        List<String> documents = indexedDoc.getAllFieldValues(SolrElementField.FIELD_DOCUMENTS);

        for (String documentsValue : documents) {

            SolrDoc solrDoc = ResourceMapSubprocessor.getSolrDoc(documentsValue);

            solrDoc.removeFieldsWithValue(SolrElementField.FIELD_ISDOCUMENTEDBY, pid);

            removeFromIndex(documentsValue);

            insertToIndex(solrDoc);

        }

        List<String> documentedBy = indexedDoc

                .getAllFieldValues(SolrElementField.FIELD_ISDOCUMENTEDBY);

        for (String documentedByValue : documentedBy) {

            SolrDoc solrDoc = ResourceMapSubprocessor.getSolrDoc(documentedByValue);

            removeFromIndex(documentedByValue);

            insertToIndex(solrDoc);

        }

        //SolrElementAdd addCommand = new SolrElementAdd(docsToUpdate);

        //httpService.sendUpdate(solrIndexUri, addCommand);

    }

    /*

     * Remove a pid from the solr index

     */

            /*IndexEvent event = new IndexEvent();

            event.setDate(Calendar.getInstance().getTime());

            Identifier identifier = new Identifier();

            identifier.setValue(pid);

            event.setIdentifier(identifier);*/

            try {

                solrServer.deleteById(pid);

                solrServer.commit();

                /*event.setType(IndexEvent.SUCCESSDELETE);

                event.setDescription("Successfully remove the solr index for the id "+identifier.getValue());