Metacat

/**

 *  '$RCSfile$'

 *    Purpose: A class to asyncronously do delta-T replication checking

 *  Copyright: 2000 Regents of the University of California and the

 *             National Center for Ecological Analysis and Synthesis

 *    Authors: Chad Berkley

 *    Release: @release@

 *

 *   '$Author$'

 *     '$Date$'

 * '$Revision$'

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

import java.sql.*;

import java.util.*;

import java.lang.Thread;

import java.io.*;

import java.net.*;

import org.xml.sax.ContentHandler;

import org.xml.sax.DTDHandler;

import org.xml.sax.EntityResolver;

import org.xml.sax.ErrorHandler;

import org.xml.sax.InputSource;

import org.xml.sax.XMLReader;

import org.xml.sax.SAXException;

import org.xml.sax.SAXParseException;

import org.xml.sax.helpers.XMLReaderFactory;

 * This class handles deltaT replication checking.  Whenever this TimerTask

 * is fired it checks each server in xml_replication for updates and updates

 * the local db as needed.

 */

public class ReplicationHandler extends TimerTask

{

  Hashtable serverList = new Hashtable();

  Connection conn;

  PrintWriter out;

  public ReplicationHandler(PrintWriter o)

  {

    this.out = o;

  }

  {

    this.out = o;

    this.serverCheckCode = serverCheckCode;

  }

   * Method that implements TimerTask.run().  It runs whenever the timer is

   * fired.

   */

  public void run()

  {

    //find out the last_checked time of each server in the server list and

    //send a query to each server to see if there are any documents in

    //xml_documents with an update_date > last_checked

    try

    {

      { //this connection is prone to error for some reason so we

        //try to connect it three times before quiting.

        conn = util.openDBConnection();

      }

      catch(SQLException sqle)

      {

        try

        {

          conn = util.openDBConnection();

        }

        catch(SQLException sqlee)

        {

          try

          {

            conn = util.openDBConnection();

          }

          catch(SQLException sqleee)

          {

            System.out.println("error getting db connection in " +

                               "ReplicationHandler.run: " +

                               sqleee.getMessage());

          }

        }

      }

      if (serverList==null||serverList.isEmpty())

      {

        return;

      }

    catch (Exception e)

    {

  }

  /**

   */

    /*

     Pseudo-algorithm

     - request a doc list from each server in xml_replication

     - check the rev number of each of those documents agains the

       documents in the local database

     - pull any documents that have a lesser rev number on the local server

       from the remote server

     - delete any documents that still exist in the local xml_documents but

       are in the deletedDocuments tag of the remote host response.

     - update last_checked to keep track of the last time it was checked.

       (this info is theoretically not needed using this system but probably

       should be kept anyway)

    */

    Connection conn = null;

    try

    {

    String server;

    String update;

    ReplMessageHandler message = new ReplMessageHandler();

    Vector responses = new Vector();

    boolean tablehasrows;

    boolean flag=false;

    XMLReader parser;

    URL u;

    try

    {

      keys = serverList.keys();

      while(keys.hasMoreElements())

      {

        server = (String)(keys.nextElement());

        responses.add(result);

      }

      //If response is null, mecat don't need do anything

       if (responses==null||responses.isEmpty())

      {

        return;

      }

      { //check each server for updated files

        //System.out.println("parsing responses");

        parser.parse(new InputSource(

                     new StringReader(

                     (String)(responses.elementAt(i)))));

        //System.out.println("v: " + v.toString());

        //in the local DB.

        for(int j=0; j<v.size(); j++)

        {

          dataFile=false;

          //w is information for one document, information contain

          //docid, rev, server or datafile.

          //If it has, this document is data file

          if (w.contains((String)util.getOption("datafileflag")))

          {

            dataFile=true;

          }

          String docid = (String)w.elementAt(0);

          //System.out.println("docid: " + docid);

          int rev = Integer.parseInt((String)w.elementAt(1));

          //System.out.println("rev: " + rev);

          String docServer = (String)w.elementAt(2);

          //System.out.println("docServer: " + docServer);

          pstmt = conn.prepareStatement("select rev from xml_documents where "+

                                        "docid like '" + docid + "'");

          pstmt.execute();

          rs = pstmt.getResultSet();

          tablehasrows = rs.next();

          if(tablehasrows)

          { //check the revs for an update because this document is in the

            //local DB, it just might be out of date.

            int localrev = rs.getInt(1);

            if(localrev == rev)

            {

              flag = false;

            }

            else if(localrev < rev)

            {//this document needs to be updated so send an read request

              action = "UPDATE";

              flag = true;

            }

          }

          else

          { //insert this document as new because it is not in the local DB

            action = "INSERT";

            flag = true;

          }

          if(flag && !dataFile)

            DocInfoHandler dih = new DocInfoHandler();

            XMLReader docinfoParser = initParser(dih);

            //System.out.println("Sending message: " + docinfoUrl.toString());

            docinfoParser.parse(new InputSource(new StringReader(docInfoStr)));

            Hashtable docinfoHash = dih.getDocInfo();

            int serverCode = MetacatReplication.getServerCode(docServer);

            //docid should include rev number too

            String accnum=docid+util.getOption("accNumSeparator")+

                                              (String)docinfoHash.get("rev");

            //System.out.println("accnum: "+accnum);

            {

              String newDocid = DocumentImpl.write(conn,

                              null,  /* the dtd text */

                              false /* validate */);

                                         docServer);

                                 docServer);*/

            catch(Exception e)

            {

                                 "ReplicationHandler.update: " + docid +

                                 ": " + e.getMessage());

           // this is for data file

          if(flag && dataFile)

          {

            //if the document needs to be updated or inserted, this is executed

            //String newxmldoc = MetacatReplication.getURLContent(u);

            DocInfoHandler dih = new DocInfoHandler();

            XMLReader docinfoParser = initParser(dih);

            URL docinfoUrl = new URL("https://" + docServer +

                  "?server="+util.getLocalReplicationServerName()+

                  "&action=getdocumentinfo&docid="+docid);

            //System.out.println("Sending message: " + docinfoUrl.toString());

            String docInfoStr = MetacatReplication.getURLContent(docinfoUrl);

            docinfoParser.parse(new InputSource(new StringReader(docInfoStr)));

            Hashtable docinfoHash = dih.getDocInfo();

            String user = (String)docinfoHash.get("user_owner");

            String docName = (String)docinfoHash.get("docname");

            String docType = (String)docinfoHash.get("doctype");

            int serverCode = MetacatReplication.getServerCode(docServer);

            //System.out.println("updating doc: " + docid +" action: "+ action);

            //docid should include rev number too

            String accnum=docid+util.getOption("accNumSeparator")+

                                              (String)docinfoHash.get("rev");

            String datafilePath = util.getOption("datafilepath");

            u = new URL("https://" + docServer + "?server="+

                                        util.getLocalReplicationServerName()+

                                            "&action=readdata&docid="+accnum);

            //System.out.println("accnum: "+accnum);

            try

            {

              //register data file into xml_documents table and wite data file

              //into file system

              DocumentImpl.writeDataFile(u.openStream(), datafilePath,

                                  docName,docType, accnum, user,serverCode);

              MetacatReplication.replLog("wrote doc " + docid + " from " +

                                         docServer);

              /*System.out.println("wrote doc " + docid + " from " +

                                 docServer);*/

            }

            catch(Exception e)

            {

              System.out.println("error writing document in " +

                                 "ReplicationHandler.update: " + docid +

                                 ": " + e.getMessage());

            }

          }//for datafile

        }//for update docs

        { //delete the deleted documents;

          Vector w = new Vector((Vector)d.elementAt(k));

          String docid = (String)w.elementAt(0);

          {

            //because delete method docid should have rev number

            //so we just add one for it. This rev number is no sence.

            String accnum=docid+util.getOption("accNumSeparator")+"1";

            //System.out.println("accnum: "+accnum);

            DocumentImpl.delete(conn, accnum, null, null);

        while(keys.hasMoreElements())

        {

          server = (String)(keys.nextElement());

          String datestr = datexml.substring(11, datexml.indexOf('<', 11));

          StringBuffer sql = new StringBuffer();

          sql.append("update xml_replication set last_checked = to_date('");

          sql.append(datestr).append("', 'YY-MM-DD HH24:MI:SS') where ");

          sql.append("server like '").append(server).append("'");

          //System.out.println("sql: " + sql.toString());

          pstmt.executeUpdate();

          //conn.commit();

      }//for

    }//try

    {

                          e.getMessage());*/

    }

    {

      try

      {

        pstmt.close();

      catch(Exception ee) {}

  }//update

  /**

   */

    {

      { //this connection is prone to error for some reason so we

        //try to connect it three times before quiting.

        conn = util.openDBConnection();

      }

      catch(SQLException sqle)

      {

        try

        {

          conn = util.openDBConnection();

        }

        catch(SQLException sqlee)

        {

          try

          {

            conn = util.openDBConnection();

          }

          catch(SQLException sqleee)

          {

            System.out.println("error getting db connection in " +

                               "ReplicationHandler.update: " +

                               sqleee.getMessage());

          }

        }

      }

      Enumeration keys = serverList.keys();

      while(keys.hasMoreElements())

      { //go through each server

        server = (String)(keys.nextElement());

        //System.out.println("sending message " + u.toString());

        XMLReader catparser = initParser(cmh);

        catparser.parse(new InputSource(new StringReader(catxml)));

        //parse the returned catalog xml and put it into a vector

        Vector remoteCatalog = cmh.getCatalogVect();

        String localcatxml = MetacatReplication.getCatalogXML();

        catparser = initParser(cmh);

        catparser.parse(new InputSource(new StringReader(localcatxml)));

        Vector localCatalog = cmh.getCatalogVect();

        //now we have the catalog from the remote server and this local server

        //we now need to compare the two and merge the differences.

        //the comparison is base on the public_id fields which is the 4th

        //entry in each row vector.

        Vector publicId = new Vector();

        for(int i=0; i<localCatalog.size(); i++)

        {

          Vector v = new Vector((Vector)localCatalog.elementAt(i));

        for(int i=0; i<remoteCatalog.size(); i++)

        {

          Vector v = (Vector)remoteCatalog.elementAt(i);

          //System.out.println("remoteCatalog.size(): " + remoteCatalog.size());

          //System.out.println("v.elementAt(3): " + (String)v.elementAt(3));

          { //so we don't have this public id in our local table so we need to

            //add it.

            sql.append("insert into xml_catalog (entry_type, source_doctype, ");

            sql.append("target_doctype, public_id, system_id) values (?,?,?,");

            sql.append("?,?)");

            pstmt.setString(1, (String)v.elementAt(0));

            pstmt.setString(2, (String)v.elementAt(1));

            pstmt.setString(3, (String)v.elementAt(2));

            pstmt.setString(4, (String)v.elementAt(3));

            pstmt.setString(5, (String)v.elementAt(4));

            pstmt.execute();

        }

      }

    catch(Exception e)

    {

                          e.getMessage());

      {

        conn.close();

      }

      catch (Exception ee)

      {

      }

  }

  /**

   * ------------------------------------

   *   X             X             FALSE

   *   X             _             FALSE

   *   _             X             TRUE

   *   _             _             TRUE

  private static boolean alreadyDeleted(String docid, Connection conn)

  {

    try

    {

      boolean xml_revs = false;

      PreparedStatement pstmt = conn.prepareStatement(sb.toString());

      pstmt.execute();

      ResultSet rs = pstmt.getResultSet();

      boolean tablehasrows = rs.next();

      if(tablehasrows)

      {

      }

      sb = new StringBuffer();

      sb.append("select docid from xml_documents where docid like '");

      sb.append(docid).append("'");

      pstmt.execute();

      rs = pstmt.getResultSet();

      tablehasrows = rs.next();

      {

        xml_docs = true;

      }

      if(xml_docs && xml_revs)

      {

        return false;

      }

      else if(xml_docs && !xml_revs)

      {

        return false;

      }

      else if(!xml_docs && xml_revs)

      {

      }

      {

        return true;

      }

    catch(Exception e)

    {

                          e.getMessage());

    }

    {

    }

  }

  /**

   * "UPDATE" if it is, "INSERT" if it isn't

   */

  private static String getAction(String docid)

  {

    try

    {

      MetaCatUtil util = new MetaCatUtil();

      StringBuffer sql = new StringBuffer();

      sql.append("select docid from xml_documents where docid like '");

      sql.append(docid).append("'");

      Connection conn = util.openDBConnection();

      PreparedStatement pstmt = conn.prepareStatement(sql.toString());

      pstmt.execute();

      ResultSet rs = pstmt.getResultSet();

      if(rs.next())

      {

        return "UPDATE";

      }

      else

      {

        return "INSERT";

      }

    }

    catch(Exception e)

    {

      System.out.println("error in replicationHandler.getAction: " +

                          e.getMessage());

    }

    return "";

  }

  /**

   * and it's last update time.

   * @param conn a connection to the database

   */

    Hashtable sl = new Hashtable();

    PreparedStatement pstmt;

    try

    {

                                    "from xml_replication");

      ResultSet rs = pstmt.getResultSet();

      boolean tableHasRows = rs.next();

      while(tableHasRows)

      {

        {//only put the server in the list if the replicate flag is true

          String server = rs.getString(1);

          String last_checked = rs.getString(2);

          if(!server.equals("localhost"))

          {

            sl.put(server, last_checked);

          }

    catch(Exception e)

    {

      System.out.println("error in replicationHandler.buildServerList(): " +

                         e.getMessage());

    }

    return sl;

  }

   * Method to get a host server information for given docid

   * @param conn a connection to the database

   */

  public static Hashtable getHomeServer(String docId)

  {

    Hashtable sl = new Hashtable();

    Connection conn = null;

    MetaCatUtil ut=new MetaCatUtil();

    int serverLocation;

    try

    {

      conn=ut.getConnection();

      //get a server location from xml_document table

                                            +"where docid = ?");

      pstmt.execute();

      ResultSet serverName = pstmt.getResultSet();

      //get a server location

      if(serverName.next())

      {

        serverLocation=serverName.getInt(1);

        pstmt.close();

      }

      else

      {

        pstmt.close();

        ut.returnConnection(conn);

        return null;

      }

      pstmt = conn.prepareStatement("select server, last_checked, replicate " +

                        "from xml_replication where serverid = ?");

      pstmt.setInt(1, serverLocation);

      pstmt.execute();

      ResultSet rs = pstmt.getResultSet();

      boolean tableHasRows = rs.next();

      if (tableHasRows)

      {

          String server = rs.getString(1);

          String last_checked = rs.getString(2);

          if(!server.equals("localhost"))

          {