source: trunk/gsdl3/src/java/org/greenstone/gsdl3/gs3build/notifier/NotifierManager.java@ 8645

package org.greenstone.gsdl3.gs3build.notifier;



import java.io.*;
import java.net.*;
import java.sql.*;
import java.text.SimpleDateFormat;
import java.util.*;
import java.util.Date;
import java.util.TreeMap;

import org.apache.soap.Constants;
import org.apache.soap.SOAPException;
import org.apache.soap.rpc.*;
import org.apache.soap.rpc.Call;
import org.apache.soap.rpc.Parameter;
import org.greenstone.gsdl3.gs3build.CollectionManager;
import org.greenstone.gsdl3.gs3build.util.GS3SQLConnection;

public class NotifierManager {

    public NotifierManager() {
    }
    
    public void detectEvents(CollectionManager collManager) {
        
        String collectionName = collManager.getCollectionName();
        String host = "http://localhost:8080/soap/servlet/rpcrouter";
        
        String lastBuild = dateToSqlString(collManager.getBuildDate());
        Statement statement = collManager.getDatabase().createStatement();
        try {
            // detect all new documents. A document is new if and only if
            // AccessionDate >= CollectionLastRebuiltDate
            ResultSet results = statement.executeQuery("SELECT DocID FROM document WHERE DocType != 'GSMETADATA' AND AccessionDate >= " + lastBuild);
            notify(collectionName, host, results, collManager.getDatabase(), "new_document");
            
            // detect modified documents. A document is modified if and only if
            // AccessionDate < CollectionLastRebuiltDate (ie, it is not new) and 
            // ModifiedDate >= IndexedDate
            results = statement.executeQuery("SELECT DocID FROM document WHERE DocType != 'GSMETADATA' AND AccessionDate < " + lastBuild + " AND ModifiedDate >= IndexedDate");
            notify(collectionName, host, results, collManager.getDatabase(), "document_modified");
        } catch (SQLException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
    }

    /**
     * @param collName
     * @param hostURL
     * @param results
     * @param eventType TODO
     * @throws MalformedURLException
     * @throws IOException
     * @throws ProtocolException
     * @throws SQLException
     */
    private void notify(String collName, String hostURL, ResultSet results, GS3SQLConnection database, String eventType) {
        try {
            URL url = new URL(hostURL);
            System.out.println("trying to send (via SOAP) to " + url);
            
            while(results.next()) {
                System.out.println("new document: " + results.getString("DocID"));
                Map event = new TreeMap();
                
                String documentID = results.getString("DocID");
                
                event.put("documentID", documentID);
                event.put("type", eventType);
                event.put("collectionID", collName);
                event.put("host_url", hostURL);
                event.put("document_title", getDocumentTitle(documentID, database));
                
                Call call = new Call();
                call.setTargetObjectURI("alerting");
                call.setMethodName("receiveEvent");
                call.setEncodingStyleURI(Constants.NS_URI_SOAP_ENC);
                Vector params = new Vector();
                params.addElement(new Parameter("event", Map.class, event, null));
                call.setParams(params);
                try {
                    Response resp = call.invoke(url, "");
                    if (!resp.generatedFault()) {
                        System.out.println("successfully posted event");
                        return;
                    }
                    System.err.println("Posting event was unsuccessful:");
                    System.err.println(resp.getFault().getFaultString());
                } catch (SOAPException e1) {
                    System.err.println("Exception while posting event: ");
                    e1.printStackTrace();
                }
            }
        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        } catch (SQLException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
    }

    /**
     * @param documentID
     * @return
     * @throws SQLException
     */
    private String getDocumentTitle(String documentID, GS3SQLConnection database) throws SQLException {
        Statement statement = database.createStatement();
        ResultSet results = statement.executeQuery("select mdvalues.value " +
                "from mdvalues join namespaces on (mdvalues.namespaceref = namespaces.namespaceid) " +
                "join metadata on (metadata.metadataref = namespaces.metadataref) " +
                "where label like 'Title' and metadata.docID = '" + documentID +"';");
        return results.getString(0);
    }

    /**
     * @param date
     * @return
     */
    private String dateToSqlString(Date date) {
        return new SimpleDateFormat("yyyyMMddHHmmss").format(date);
    }
}