Metacat

/**

 *  '$RCSfile$'

 *    Purpose: A Class that implements utility methods like:

 *             1/ Reding all doctypes from db connection

 *             2/ Reading Lore type Data Guide from db connection

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

 *             National Center for Ecological Analysis and Synthesis

 *    Authors: Matt Jones

 * 

 *   '$Author$'

 *     '$Date$'

 * '$Revision$'

 */

package edu.ucsb.nceas.metacat;

import java.sql.Connection;

import java.sql.SQLException;

import java.sql.PreparedStatement;

import java.sql.ResultSet;

import java.util.Hashtable;

import java.util.Enumeration;

import java.util.Vector;

/**

 * A suite of utility classes for quering DB

 */

public class DBUtil {

  private Connection	conn = null;

  /**

   * main routine used for testing.

   * <p>

   * Usage: java DBUtil <-dt|-dg>

   *

   * @param -dt for selecting all doctypes

   *        -dg for selecting DataGuide

   */

  static public void main(String[] args) {

     if (args.length < 1)

     {

        System.err.println("Wrong number of arguments!!!");

        System.err.println("USAGE: java DBUtil <-dt|-dg>");

        return;

     } else {

        try {

          // Open a connection to the database

          MetaCatUtil   util = new MetaCatUtil();

          Connection dbconn = util.openDBConnection();

          DBUtil dbutil = new DBUtil( dbconn );

          if ( args[0].equals("-dt") ) {

            String doctypes = dbutil.readDoctypes();

            System.out.println(doctypes);

          } else if ( args[0].equals("-dg") ) {

            String dataguide = dbutil.readDataGuide("");

            System.out.println(dataguide);

          } else {

            System.err.println("USAGE: java DBUtil <-dt|-dg>");

          }  

        } catch (Exception e) {

          System.err.println("EXCEPTION HANDLING REQUIRED");

          System.err.println(e.getMessage());

          e.printStackTrace(System.err);

        }

     }

  }

  /**

   * Construct an instance of the utility class

   */

  public DBUtil( Connection conn ) {

    this.conn = conn;

  }

  /**

   * read all doctypes from db connection in XML format

   * select all Public Id from xml_catalog table

   */

  public String readDoctypes()

        throws SQLException  {

    Vector doctypeList = new Vector();

    try {

      PreparedStatement pstmt =

        conn.prepareStatement("SELECT public_id FROM xml_catalog");

      pstmt.execute();

      ResultSet rs = pstmt.getResultSet();

      boolean tableHasRows = rs.next();

      while (tableHasRows) {

           doctypeList.addElement(rs.getString(1));

           tableHasRows = rs.next();

      }

      pstmt.close();

    } catch (SQLException e) {

      System.out.println("DBUtil.readDoctypes(): " + e.getMessage());

      throw e;

    }

    return formatToXML(doctypeList, "doctype");

  }

  /**

   * read Data Guide for a given doctype from db connection in XML format

   * select all distinct absolute paths from xml_index table

   */

  public String readDataGuide( String doctype )

        throws SQLException  {

    Vector dataguide = new Vector();

    String path;

    try {

      PreparedStatement pstmt =

        conn.prepareStatement("SELECT distinct path, nodeid FROM xml_index" +

                              " WHERE path LIKE '/' || ? || '%'" + 

                              " ORDER BY nodeid");

      pstmt.setString(1, doctype);

      pstmt.execute();

      ResultSet rs = pstmt.getResultSet();

      boolean tableHasRows = rs.next();

      while (tableHasRows) {

        path = rs.getString(1);

        if ( dataguide.indexOf(path) == -1 ) {

            dataguide.addElement(path);

        }    

        tableHasRows = rs.next();

      }

      pstmt.close();

    } catch (SQLException e) {

      System.out.println("DBUtil.readDoctypes(): " + e.getMessage());

      throw e;

    }

    return formatToXML(dataguide, "path");

  }

  /**

   * format the ResultSet to XML

   */

  private String formatToXML(Vector resultset, String tag) {

    String val = null;

    String doctype = null;

    StringBuffer result = new StringBuffer();

    Enumeration rs = resultset.elements(); 

    result.append("<?xml version=\"1.0\"?>\n");

    result.append("<resultset>\n"); 

    while (rs.hasMoreElements()) {

        val = (String)rs.nextElement();

        result.append("   <" + tag + ">" + val + "</" + tag + ">\n");

    }

    result.append("</resultset>\n"); 

    return result.toString();

  }

}