Metacat

/**

 *  '$RCSfile$'

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

 *             National Center for Ecological Analysis and Synthesis

 *

 * Author: Matthew Perry 

 * '$Date: 2006-08-31 16:54:32 -0700 (Thu, 31 Aug 2006) $'

 * '$Revision: 3035 $'

 *

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

import java.sql.ResultSet;

import java.sql.PreparedStatement;

import java.util.Vector;

import java.io.IOException;

import org.apache.log4j.Logger;

import edu.ucsb.nceas.metacat.MetaCatUtil;

import edu.ucsb.nceas.metacat.DBConnection;

import org.geotools.feature.Feature;

/* 

 * Harvests spatial data from metacat and saves to persistent cache

 */

public class SpatialHarvester {

  private static Logger log = Logger.getLogger(SpatialHarvester.class.getName());

  private DBConnection dbconn;

  /** constructor to initialize db connection **/

  public SpatialHarvester() {

      try {

          dbconn = new DBConnection();

      } catch( Exception e ) {

          log.error("Error getting docids from queryAllDocids");

          e.printStackTrace();

      }

  }

  /*

   * Closes the database connection

   * MUST be called after you're done with

   * the SpatialHarvester

   */

  public void destroy() {

      try {

          dbconn.close();

      } catch( Exception e ) {

          log.error("Error closing out dbconn");

          e.printStackTrace();

      }

  }

  /**

   * Returns a Vector of all the docids in the xml_path_index tables

   */

  protected Vector queryAllDocids() {

    Vector _docs = new Vector();

    PreparedStatement pstmt = null;

    ResultSet rs = null;

    String query = "select distinct docid from xml_path_index";

    try {

      pstmt = dbconn.prepareStatement(query);

      pstmt.execute();

      rs = pstmt.getResultSet();

      while (rs.next()) {

        String docid = rs.getString(1);

        //log.fatal("adding docid: " + docid);

        _docs.add(docid);

    }

      rs.close();

      pstmt.close();

    }

    catch(Exception e) {

      log.error("Error getting docids from queryAllDocids");

      e.printStackTrace();

    }

    return _docs;

  }

  /*

   * Currently just a wrapper around the harvester

   * Eventually we can use this method as a

   * timed que like the indexing process

   */                   

  public void addToUpdateQue( String docid ) {

      harvestDocument(docid);

  }

  /*

   * Delete from spatial cache

   * Just a wrapper around delete for now

   */

  public void addToDeleteQue( String docid ) {

      deleteDocument(docid);

  }

  /*

   * Given a docid, will attempt to delete

   * from spatial cache

   */

  public void deleteDocument( String docid ) {

     // Read the existing spatial dataset cache

     SpatialDataset sds = new SpatialDataset();

     try {

         // Delete both the polygon(s) and point(s)

         sds.delete( "polygon" , docid ); 

         sds.delete( "point" , docid ); 

     } catch (IOException e) {

         log.error("IOException while deleting from spatial cache");

     }

     log.info(" --------- Spatial Harvester - Deleted from spatial cache : " + docid );

  }      

  /*

   * Given a docid, will update the spatial cache accordingly

   */

  public void harvestDocument( String docid ) {

     // Read the existing spatial dataset cache

     SpatialDataset sds = new SpatialDataset();

     // insert OR update the spatial cache

     // SpatialDataset.insertOrUpdate takes care of the difference 

     SpatialDocument sdoc = new SpatialDocument( docid, dbconn );

     try {

         Feature polygonFeature = sdoc.getPolygonFeature();

         sds.insertOrUpdate("polygon", polygonFeature, docid );

         Feature pointFeature = sdoc.getPointFeature();

         sds.insertOrUpdate("point", pointFeature, docid );

         log.info(" --------- Spatial Harvester - spatial cache updated for : " + docid );

     } catch (IOException e) {

         log.error("IOException while performing spatial harvest ");

     }

  }

  /*

   * Completely regenerates the spatial cache

   */

  public void regenerate() {

      // Create new Spatial Dataset

      SpatialDataset sds = new SpatialDataset();

      // Get list of all docids in the database

      Vector docids = queryAllDocids(); 

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

          SpatialDocument sdoc = new SpatialDocument( (String)docids.elementAt(i), dbconn );

          // Get the polygon representation of SpatialDocument

          // and add it to the spatial dataset

          Feature polygonFeature = sdoc.getPolygonFeature();

          sds.add("polygon", polygonFeature );

          // Get the point representation of SpatialDocument

          // and add it to the spatial dataset

          Feature pointFeature = sdoc.getPointFeature();

          sds.add("point", pointFeature );

          log.info(" ****** Spatial harvest of docid " + docids.elementAt(i) );

      }

      // save SpatialDataset

      sds.save();

  }

}