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.resourcemap;

import java.io.IOException;

import java.util.List;

import java.util.Map;

import javax.xml.parsers.ParserConfigurationException;

import javax.xml.xpath.XPathExpressionException;

import org.apache.commons.codec.EncoderException;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.apache.solr.client.solrj.SolrServer;

import org.apache.solr.client.solrj.SolrServerException;

import org.apache.solr.client.solrj.response.QueryResponse;

import org.apache.solr.common.SolrDocument;

import org.apache.solr.common.params.SolrParams;

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

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

import org.dataone.service.exceptions.NotImplemented;

import org.dataone.service.exceptions.UnsupportedType;

import org.xml.sax.SAXException;

 * A solr index parser for the ResourceMap file.

 * The solr doc of the ResourceMap self only has the system metadata information.

 * The solr docs of the science metadata doc and data file have the resource map package information.

 */

    private static final String QUERY ="q=id:";

    private static SolrServer solrServer =  null;

    static {

        try {

            solrServer = SolrServerFactory.createSolrServer();

        } catch (Exception e) {

            log.error("ResourceMapSubprocessor - can't generate the SolrServer since - "+e.getMessage());

        }

    }

    @Override

    public Map<String, SolrDoc> processDocument(String identifier, Map<String, SolrDoc> docs,

        for (SolrDoc processedDoc : processedDocs) {

            processedDocsMap.put(processedDoc.getIdentifier(), processedDoc);

        }

        return processedDocsMap;

    }

    private List<SolrDoc> processResourceMap(SolrDoc indexDocument, Document resourceMapDocument)

            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);

            }

        }*/

        return mergedDocuments;

    }

        if(ids != null) {

            for(String id : ids) {

                    list.add(doc);

                } else if ( !id.equals(resourceMapId)) {

            }

        }

        return list;

    }

     * Get the SolrDoc list for the list of the ids.

     */

        if(ids != null) {

            for(String id : ids) {

                    list.add(doc);

                }

            }

        }

        return list;

    }

	 * Get the SolrDoc for the specified id

	 */

	public static SolrDoc getSolrDoc(String id) throws SolrServerException,

			IOException, ParserConfigurationException, SAXException,

			XPathExpressionException, NotImplemented, NotFound, UnsupportedType {

	    List<SolrDoc> list = getDocumentsByQuery(query);

	    if(list != null && !list.isEmpty()) {

	        doc = list.get(targetIndex);

	    }

	}

	/**

	 * Get a list of solr documents which's resourcemap field matches the given value.

	 * @param resourceMapId - the target resource map id

	 * @return the list of solr document

	 * @throws MalformedURLException

	 * @throws UnsupportedType

	 * @throws NotFound

	 * @throws SolrServerException

	 * @throws ParserConfigurationException

	 * @throws IOException

	 * @throws SAXException

	 */

	public static List<SolrDoc> getDocumentsByResourceMap(String resourceMapId) throws MalformedURLException,

	            UnsupportedType, NotFound, SolrServerException, ParserConfigurationException, IOException, SAXException {

	    String query = QUERY2 + "\"" + resourceMapId + "\"";

	    return getDocumentsByQuery(query);

	}

	/**

	 * Get a list of slor docs which match the query.

	 * @param query - a string of a query

	 * @return

	 * @throws SolrServerException

	 * @throws MalformedURLException

	 * @throws UnsupportedType

	 * @throws NotFound

	 * @throws ParserConfigurationException

	 * @throws IOException

	 * @throws SAXException

	 */

	public static List<SolrDoc> getDocumentsByQuery(String query) throws SolrServerException, MalformedURLException, UnsupportedType,

	                                                                NotFound, ParserConfigurationException, IOException, SAXException {

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

	    if (solrServer != null && query != null && !query.trim().equals("")) {

            SolrParams solrParams = SolrRequestParsers.parseQueryString(query);

            QueryResponse qr = solrServer.query(solrParams);

            if (qr != null && qr.getResults() != null) {

                for(int i=0; i<qr.getResults().size(); i++) {

                    SolrDocument orig = qr.getResults().get(i);

                    SolrDoc doc = new SolrDoc();

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

                    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);

                        }

                    }

                    docs.add(doc);

                }

            }

	    }

	    return docs;

	}