org.ambraproject.search.service.DummySOLRMessageSender.java Source code

Introduction

Here is the source code for org.ambraproject.search.service.DummySOLRMessageSender.java

Source

/*
 * $HeadURL$
 * $Id$
 * Copyright (c) 2006-2012 by Public Library of Science http://plos.org http://ambraproject.org
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.ambraproject.search.service;

import org.ambraproject.service.search.SolrServerFactory;
import org.apache.solr.client.solrj.SolrServer;
import org.apache.solr.common.SolrInputDocument;
import org.springframework.beans.factory.annotation.Required;
import org.ambraproject.queue.MessageSender;
import org.w3c.dom.Document;
import org.w3c.dom.Node;
import javax.xml.xpath.*;
import java.util.Map;

/**
 * DummyMessageSender for sending messages to SOLR.  In the normal world this class is actually a proxy to the
 * queue.  But being we only want to make sure articles are being indexed here.  This just acts as a thin
 * wrapper to the SOLR server
 */

public class DummySOLRMessageSender implements MessageSender {
    private SolrServer solrServer;

    private static final String indexQueue = "activemq:plos.solr.article.index?transacted=false";
    private static final String deleteQueue = "seda:search.delete";
    private static final String indexPublishQueue = "seda:search.index";
    private static final String indexAll = "seda:search.indexall";

    @Required
    public void setSolrServerFactory(SolrServerFactory solrServerFactory) {
        solrServer = solrServerFactory.getServer();
    }

    /**
     * Send a text message.
     * @param destination URL of the destination "activemq:plos.pmc" fro example.
     * @param body Message body,
     */
    public void sendMessage(String destination, String body) {
        //Don't worry about index all.  (Thats a queue task)  But we can check the delete here
        if (deleteQueue.equals(destination)) {
            try {
                String solrID = body.replaceAll("info:doi/", "");
                solrServer.deleteById(solrID);
                solrServer.commit();
            } catch (Exception ex) {
                throw new RuntimeException(ex.getMessage(), ex);
            }
        }

    }

    /**
     * Send a DOM Document message.
     * @param destination URL of the destination "activemq:plos.pmc" for example.
     * @param body Message body,
     */
    public void sendMessage(String destination, Document body) {
        if (indexQueue.equals(destination) || indexPublishQueue.equals(destination)) {
            try {
                //We just want to confirm that SOLR gets something, so lets just index article ID and eIssn

                SolrInputDocument sid = new SolrInputDocument();

                Node res = XPathSingleNodeQuery(body,
                        "//article/front/article-meta/article-id[@pub-id-type='doi']/text()");
                sid.addField("id", res.getTextContent());

                res = XPathSingleNodeQuery(body, "//article/front/journal-meta/issn[@pub-type='epub']/text()");
                sid.addField("eissn", res.getTextContent());

                solrServer.add(sid);
                solrServer.commit();

            } catch (Exception ex) {
                throw new RuntimeException(ex.getMessage(), ex);
            }
        }
    }

    /**
     * Provided as a stub so unit tests compile.  This method only passes through requests and
     * methods are unit tested other places.
     */
    public void sendMessage(String destination, Object body, Map<String, Object> headers) {
        throw new RuntimeException("Not implemented, if you are calling this method, implement a unit test");
    }

    private Node XPathSingleNodeQuery(Document dom, String statement) throws XPathExpressionException {
        XPathFactory factory = XPathFactory.newInstance();
        XPath xpath = factory.newXPath();
        XPathExpression expr = xpath.compile(statement);

        return (Node) expr.evaluate(dom, XPathConstants.NODE);
    }
}