Metacat

/**

 *  '$RCSfile$'

 *    Purpose: A Class that implements utility methods like:

 *             1/ Reding all doctypes from db connection

 *             3/ Reading Lore type Data Guide from db connection

 *             National Center for Ecological Analysis and Synthesis

 *   '$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.SQLException;

import java.sql.PreparedStatement;

import java.sql.ResultSet;

import java.io.BufferedInputStream;

import java.io.InputStream;

import java.io.IOException;

import java.net.URL;

import java.net.MalformedURLException;

import java.util.Vector;

import edu.ucsb.nceas.metacat.database.DBConnectionPool;

import edu.ucsb.nceas.utilities.PropertyNotFoundException;

public class DBUtil {

  public static final int NONEEXIST = -1;

  /**

   * 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!!!");

     } else {

        try {

          // Open a connection to the database

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

            String doctypes = dbutil.readDoctypes();

            System.out.println(doctypes);

            String doctype = null;

            if ( args.length == 2 ) { doctype = args[1]; }

            String dtdschema = dbutil.readDTDSchema(doctype);

            System.out.println(dtdschema);

            if ( args.length == 2 ) { scope = args[1]; }

            String docid = dbutil.getMaxDocid(scope);

            "USAGE: java DBUtil <-dt | -ds [doctype] | -dg [doctype]>");

        } catch (Exception e) {

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

        }

     }

  }

  /**

   * Construct an instance of the utility class

   */

    //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  {

    int serialNumber = -1;

    PreparedStatement pstmt = null;

                  getDBConnection("DBUtil.readDoctypes");

      serialNumber=dbConn.getCheckOutSerialNumber();

      pstmt =

        dbConn.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) {

    {

      try

      {

        pstmt.close();

      }//try

      finally

      {

        DBConnectionPool.returnDBConnection(dbConn, serialNumber);

      }//finally

    }//finally

    return formatToXML(doctypeList, "doctype");

  }

  /**

   */

  public String readDTDSchema(String doctype)

    String systemID = null;

    int serialNumber = -1;

    try {

                  getDBConnection("DBUtil.readDTDSchema");

      serialNumber=dbConn.getCheckOutSerialNumber();

      pstmt = dbConn.prepareStatement("SELECT system_id " +

                                    "WHERE entry_type in ('DTD','Schema') " +

                                    "AND public_id LIKE ?");

      pstmt.setString(1, doctype);

      pstmt.execute();

      ResultSet rs = pstmt.getResultSet();

      boolean hasRow = rs.next();

      if (hasRow) {

        systemID = rs.getString(1);

        if (!systemID.startsWith("http://")) {

      pstmt.close();

    } catch (SQLException e) {

    {

      try

      {

        pstmt.close();

      }//try

      finally

      {

        DBConnectionPool.returnDBConnection(dbConn, serialNumber);

      }//finally

    }//finally

    // read from URL stream as specified by the System ID.

    try {

      // open a connection to this URL and return an InputStream

      // for reading from that connection

      InputStream istream = new URL(systemID).openStream();

      // create a buffering character-input stream

      // that uses a default-sized input buffer

      BufferedInputStream in = new BufferedInputStream(istream);

      // read the input and write into the string buffer

	    int inputByte;

	    while ( (inputByte = in.read()) != -1 ) {

        cbuff.append((char)inputByte);

	    }

      // the input stream must be closed

	    in.close();

    } catch (MalformedURLException e) {

   return cbuff.toString();

  }

//   * format the DataGuide ResultSet to XML

//   */

//  private String formatToXML(Vector resultset) {

//

//    String currPath = null;

//    String currElement = null;

//    String prevElement = null;

//    StringBuffer result = new StringBuffer();

//    Enumeration<String> rs = resultset.elements();

//    Stack st = new Stack();

//    int i = 0;

//

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

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

//

//    while (rs.hasMoreElements()) {

//        currPath = (String)rs.nextElement();

//        while ( !In(prevElement, currPath) ) {

//            currElement = (String)st.pop();

//            result.append(pad(" ",i--) + "</" + currElement + ">\n");

//            if ( st.empty() )

//                prevElement = null;

//            else

//                prevElement = (String)st.peek();

//        }

//        currElement = getElementFromPath(currPath);

//        st.push(currElement);

//        result.append(pad(" ",++i) + "<" + currElement + ">\n");

//        prevElement = currElement;

//    }

//    while ( !st.empty() ) {

//        prevElement = (String)st.pop();

//        result.append(pad(" ",i--) + "</" + prevElement + ">\n");

//    }

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

//

//    return result.toString();

//  }

//   * check if element is in path like /elem1/elem2/elemn3

//   */

//  private boolean In(String element, String path) {

//

//    if ( element == null ) return true;

//    return ( path.indexOf(element) != -1 );

//  }

//

//  /**

//   * get last element from path like /elem1/elem2/elemn3

//   */

//  private String getElementFromPath(String path) {

//

//    return ( path.substring(path.lastIndexOf("/")+1) );

//  }

//

//  /**

//   * repeates the str n-times

//   */

//  private String pad(String str, int n) {

//

//    String result = "";

//    for ( int i = 0; i < n; i++ )

//        result = result.concat(str);

//

//    return result;

//  }

  /**

   */

    String val = null;

    StringBuffer result = new StringBuffer();

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

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

    while (rs.hasMoreElements()) {

    }

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

    return result.toString();

  }

    try {

    			"'accNumSeparator'.  setting to '.': " + pnfe.getMessage());

    DBConnection dbConn = null;

    int serialNumber = -1;

                  getDBConnection("DBUtil.getMaxDocid");

        serialNumber=dbConn.getCheckOutSerialNumber();

        pstmt =

        dbConn.prepareStatement(

                "WHERE docid LIKE ? " +

            "UNION " +

      pstmt.setString(1,scope + sep + "%");

      pstmt.setString(2,scope + sep + "%");

      ResultSet rs = pstmt.getResultSet();

      while(rs.next()){

    	  temp = rs.getString(1);

    	  if(temp != null){

    		  temp = temp.substring(temp.indexOf(scope) + scope.length() + 1);

    		  try {

    			  if (localid > max) {

    				  accnum = rs.getString(1) + sep + rs.getString(2);

    			  }

    		  } catch (NumberFormatException nfe){

    			  // ignore the exception as it is possible that the

    			  // localid in the identifier is not an integer

    		  }

    	  }

      pstmt.close();

    } catch (SQLException e) {

    {

      try

      {

        pstmt.close();

      }//try

      finally

      {

        DBConnectionPool.returnDBConnection(dbConn, serialNumber);

      }//finally

    }//finally

    return accnum;

  }

  /**

   * or xml_revisions table

   */

  public boolean idExists(String docid)

    throws SQLException

  {

    {

      String id = (String)v.elementAt(i);

      if(id.trim().equals(docid.trim()))

      {

        return true;

      }

    }

    return false;

  }

  /**

	  return getAllDocidsByType(doctype, includeRevs, -1);

  }

  /**

   * return all docids with a given doctype for a given server

   */

		String sep = ".";

	    try {

	    	PropertyService.getProperty("document.accNumSeparator");

	    } catch (PropertyNotFoundException pnfe) {

	    			"'accNumSeparator'.  setting to '.': " + pnfe.getMessage());

		PreparedStatement pstmt = null;

		DBConnection dbConn = null;

		int serialNumber = -1;

		try {

			dbConn = DBConnectionPool.getDBConnection("DBUtil.getAllDocidsByType");

			serialNumber = dbConn.getCheckOutSerialNumber();

			StringBuffer sb = new StringBuffer();

			sb.append("SELECT docid, rev FROM " + "( " + "SELECT docid, rev "

					+ "FROM xml_documents ");

				sb.append("AND server_location = ' " + serverLocation + "' ");

			}

				sb.append("UNION " + "SELECT docid, rev " + "FROM xml_revisions ");

					sb.append("AND server_location = ' " + serverLocation + "' ");

				}

			sb.append(") subquery GROUP BY docid, rev");

			pstmt = dbConn.prepareStatement(sb.toString());

			if (doctype != null) {

				pstmt.setString(1, doctype);

				if (includeRevs) {

					pstmt.setString(2, doctype);

				}

			}

			pstmt.execute();

			ResultSet rs = pstmt.getResultSet();

			String id = null;

			String rev = null;

			while (rs.next()) {

				id = rs.getString(1);

				rev = rs.getString(2);

				if (id != null) {

					resultVector.addElement(id + sep + rev);

				}

			}

			pstmt.close();

		} catch (SQLException e) {

			throw new SQLException("DBUtil.getAllDocidsByType(). " + e.getMessage());

		} finally {

			try {

				pstmt.close();

			}// try

			finally {

				DBConnectionPool.returnDBConnection(dbConn, serialNumber);

			}// finally

		}// finally

		return resultVector;

	}

  /**

    try {

    	logMetacat.error("could not get property 'accNumSeparator'.  setting to '.': "

    			+ pnfe.getMessage());

    }

    DBConnection dbConn = null;

    int serialNumber = -1;

    try

    {

      dbConn=DBConnectionPool.

                getDBConnection("DBUtil.getAllDocids");

      serialNumber=dbConn.getCheckOutSerialNumber();

      StringBuffer sb = new StringBuffer();

      sb.append("SELECT docid, rev FROM " +

                "( " +

                "SELECT docid, rev " +

                "FROM xml_documents ");

      if(scope != null)

      {

        sb.append("WHERE docid LIKE ? ");

      }

      sb.append("UNION " +

                "SELECT docid, rev " +

                "FROM xml_revisions ");

      if(scope != null)

      {

        sb.append("WHERE docid LIKE ?");

      }

      sb.append(") subquery GROUP BY docid, rev");

      pstmt = dbConn.prepareStatement(sb.toString());

      if(scope != null)

      {

        pstmt.setString(1,scope + sep + "%");

        pstmt.setString(2,scope + sep + "%");

      }

      pstmt.execute();

      ResultSet rs = pstmt.getResultSet();

      String rev = null;

      while(rs.next()){

    	  id = rs.getString(1);

        rev = rs.getString(2);

    	  if(id != null){

    		  //temp = temp.substring(id.indexOf(scope) + scope.length() + 1);

          resultVector.addElement(id + sep + rev);

        }

      }

      pstmt.close();

    } catch (SQLException e) {

    finally

    {

      try

      {

        pstmt.close();

      }//try

      finally

      {

        DBConnectionPool.returnDBConnection(dbConn, serialNumber);

      }//finally

    }//finally

    return resultVector;

  }

  /**

   * If couldn't find, null will be return

   * @param givenDocId, the docid which we want to find

   */

  public static String findDataSetDocIdForGivenDocument(String givenDocId)

  {

    // Prepared statement for sql

    PreparedStatement pStmt = null;

    // Result set

    ResultSet resultSet = null;

    // String to store the data set docid

    String dataSetDocId = null;

    // DBConnection will be checkout

    DBConnection dbConn = null;

    int serialNumber = -1;

    // String to store the sql command

    String sqlCommand = null;

    try

    {

      // Checkout DBConnection from pool

      dbConn=DBConnectionPool.

                  getDBConnection("DBUtil.findDataSetDocIdForGivenDocument");

      serialNumber=dbConn.getCheckOutSerialNumber();

                                                    + "subject like ?";

      // Prepared statement

      pStmt = dbConn.prepareStatement(sqlCommand);

      // Bind variable

      pStmt.setString(1, givenDocId);

      pStmt.setString(2, givenDocId);

      // Get result set

      resultSet = pStmt.getResultSet();

      // There has record

      if (resultSet.next())

      {

        // Put the docid into dataSetDocid

        dataSetDocId = resultSet.getString(1);

        return dataSetDocId;

      }//if

      else

      {

        // No record in xml_relation table for given doicd, null returned

        return dataSetDocId;

      }//else

    }//try

    catch ( SQLException e)

    {

      logMetacat.error("DBUtil.findDataSetDocIdForGivenDocument - " +

    		  "SQL error: " + e.getMessage());

      return dataSetDocId;

    }//catch

    finally

    {

      try

      {

        // Close result set

        resultSet.close();

        // Close preparedStatement

        pStmt.close();

      }//try

      catch ( SQLException e)

      {

    	  logMetacat.error("DBUtil.findDataSetDocIdForGivenDocument - " +

    			  "error closing db resources: "  + e.getMessage());

      }//catch

      finally

      {

        // Return DBConnection to the pool

        DBConnectionPool.returnDBConnection(dbConn, serialNumber);

      }//finally

    }//finally

  }//findDataSetDocIdForGivenDocument

  /**

   * Method to get current revision and doctype for a given docid

   * The output will look like "rev;doctype"

   * @param givenDocId, the docid which we want

   */

  public String getCurrentRevisionAndDocTypeForGivenDocument(String givenDocId)

                                                 throws SQLException

  {

    // DBConection for JDBC

    DBConnection dbConn = null;

    int serialNumber = -1;

    // Prepared Statement

    PreparedStatement pstmt = null;

    // String to store a docid without rev

    String docIdWithoutRevision = null;

    // String to store the revision

    String revision = null;

    // String to store the doctype

    String docType = null;

    // Get docid without rev

    try

    {

      // Check out the connection

      dbConn=DBConnectionPool.

         getDBConnection("DBUtil.getCurrentRevisionAndDocTypeForGivenDocument");

      serialNumber=dbConn.getCheckOutSerialNumber();

      // Prepare the sql command

      pstmt = dbConn.prepareStatement(sqlCommand);

      // Get result set

      rs = pstmt.getResultSet();

      // If there is some record

      if (rs.next())

      {

        revision = rs.getString(1);

        docType = rs.getString(2);

      }//if

      else

      {

        Vector<Integer> revisionList = getRevListFromRevisionTable(docIdWithoutRevision);

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

        {

          // No record, throw a exception

          throw new

                SQLException("DBUtil.getCurrentRevisionAndDocTypeForGivenDocument - " +

                    "There is no record for given docid: " + givenDocId);