IndexMessage.java

/*
   Copyright (C) 2005-2012, by the President and Fellows of Harvard College.

   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.0

   Unless 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.

   Dataverse Network - A web application to share, preserve and analyze research data.
   Developed at the Institute for Quantitative Social Science, Harvard University.
   Version 3.0.
*/
/*
 * IndexMessage.java
 *
 * Created on February 22, 2007, 1:11 PM
 *
 * To change this template, choose Tools | Template Manager
 * and open the template in the editor.
 */
package edu.harvard.iq.dvn.core.index;

import edu.harvard.iq.dvn.core.mail.MailServiceLocal;
import edu.harvard.iq.dvn.core.study.Study;
import edu.harvard.iq.dvn.core.study.StudyServiceLocal;
import edu.harvard.iq.dvn.core.vdc.VDCNetworkServiceLocal;
import java.io.IOException;
import java.net.InetAddress;
import java.net.UnknownHostException;
import java.util.Date;
import java.util.logging.Logger;
import javax.ejb.ActivationConfigProperty;
import javax.ejb.EJB;
import javax.ejb.MessageDriven;
import javax.jms.JMSException;
import javax.jms.Message;
import javax.jms.MessageListener;
import javax.jms.ObjectMessage;

/**
 * Entity class IndexMessage
 *
 * @author roberttreacy
 */
@MessageDriven(mappedName = "jms/IndexMessage", activationConfig =  {
    @ActivationConfigProperty(propertyName = "acknowledgeMode", propertyValue = "Auto-acknowledge"),
    @ActivationConfigProperty(propertyName = "destinationType", propertyValue = "javax.jms.Queue")
})
public class IndexMessage implements MessageListener, java.io.Serializable {
    @EJB StudyServiceLocal studyService;
    @EJB VDCNetworkServiceLocal vdcNetworkService;
    @EJB MailServiceLocal mailService;
    private static final Logger logger = Logger.getLogger("edu.harvard.iq.dvn.core.index.IndexMessage");

    
    /** Creates a new instance of IndexMessage */
    public IndexMessage() {
    }
    
    public void onMessage(Message message) {
        IndexEdit operation = null;
        
        
        try {
            ObjectMessage om = (ObjectMessage) message;
            operation = (IndexEdit) om.getObject();
            switch (operation.getOperation()){
                case ADD:
                    indexStudy(operation.getStudyId());
                    break;
                case UPDATE:
                    updateStudy(operation.getStudyId());
                    break;
                case DELETE:
                    deleteStudy(operation.getStudyId());
                    break;
                    
            }
        } catch (JMSException ex) {
            ex.printStackTrace(); // error in getting object from message; can't send e-mail
            
        } catch (Exception ex) { // we catch any exception that may have happened during the process
            ex.printStackTrace();
        }
    }
    
    private void indexStudy(long studyId){
        addDocument(studyId);
    }
    
    private void updateStudy(long studyId){
        Indexer indexer = Indexer.getInstance();
        boolean deleteSuccess = true; 
        try {
            indexer.deleteDocumentCarefully(studyId);
        } catch (Exception ex) {
            deleteSuccess = false;
        }
        if (deleteSuccess) {
            addDocument(studyId);
        }
    }
    
    private void deleteStudy(long studyId){
        Indexer indexer = Indexer.getInstance();
        indexer.deleteDocument(studyId);
    }
    
    private void addDocument(final long studyId) {
        Study study=null;
        Date indexTime = new Date();
        try {
            study = studyService.getStudy(studyId);
        } catch (IllegalArgumentException e) {
            // The study may not exist because it was deleted before the indexer 
            // could process the message.  Just ignore this.
           logger.warning("Could not find study for id "+studyId+", study probably deleted before asynchronous call to Indexer.");
            
        }
        Indexer indexer = Indexer.getInstance();
       
        try {
            indexer.addDocument(study);
            try {
                studyService.setIndexTime(studyId, indexTime);
            } catch (Exception e) {
                e.printStackTrace(); // print stacktrace, but continue processing
            }
        } catch (IOException ex) {
            ex.printStackTrace();
            try {
                mailService.sendDoNotReplyMail(vdcNetworkService.find().getSystemEmail(),"IO problem", "Check index write lock "+InetAddress.getLocalHost().getHostAddress() + " , studyId " + studyId);
            } catch (UnknownHostException u) {
                u.printStackTrace();
            }
        }
    }
    
}