Metacat

/**

 *        Name: DBSimpleQuery.java

 *     Purpose: A Class that searches a relational DB for elements and 

 *              attributes that have free text matches to the query string.  

 *              It returns a result set consisting of the root nodeid for 

 *              each document that satisfies the query

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

 *              National Center for Ecological Analysis and Synthesis

 *     Authors: Matt Jones

 *

 *     Version: '$Id: DBSimpleQuery.java 98 2000-05-13 01:29:51Z jones $'

 */

package edu.ucsb.nceas.metacat;

import java.io.*;

import java.net.URL;

import java.net.MalformedURLException;

import java.sql.*;

import java.util.Stack;

import java.util.Hashtable;

import java.util.Enumeration;

/** 

 * A Class that searches a relational DB for elements and attributes that

 * have free text matches to the query string.  It returns a result set 

 * consisting of the root nodeid for each document that satisfies the query

 */

public class DBSimpleQuery {

  static  String 	defaultDB = "jdbc:oracle:thin:@localhost:1521:test";

  private Connection	conn = null;

  /**

   * the main routine used to test the DBSimpleQuery utility.

   *

   * Usage: java DBSimpleQuery <query> <user> <password> [dbstring]

   *

   * @param query the text to search for in the element and attribute content

   * @param user the username to use for the database connection

   * @param password the password to use for the database connection

   * @param dbstring the connection info to use for the database connection

   */

  static public void main(String[] args) {

     if (args.length < 3)

     {

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

        System.err.println("USAGE: java DBSimpleQuery " +

                           "<query> <user> <password> [dbstring]");

        return;

     } else {

        try {

          String query    = args[0];

          String user     = args[1];

          String password = args[2];

          String dbstring = null;

          if (args.length <= 3) {

            dbstring = defaultDB;

          } else {

            dbstring = args[3];

          }

          // Open a connection to the database

          Connection dbconn = MetaCatUtil.openDBConnection( 

                              "oracle.jdbc.driver.OracleDriver",

                              dbstring, user, password);

          // Execute the simple query

          DBSimpleQuery rd = new DBSimpleQuery(dbconn);

          Hashtable nodelist = rd.findDocuments(query);

          // Print the reulting root nodes

          StringBuffer result = new StringBuffer();

          long docid = 0;

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

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

          Enumeration doclist = nodelist.keys(); 

          while (doclist.hasMoreElements()) {

            docid = ((Long)doclist.nextElement()).longValue();

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

          }

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

          System.out.println(result);

        } catch (Exception e) {

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

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

          e.printStackTrace(System.err);

        }

     }

  }

  /**

   * construct an instance of the DBSimpleQuery class 

   *

   * <p>Generally, one would call the findDocuments() routine after creating 

   * an instance to specify the search query</p>

   *

   * @param conn the JDBC connection that we use for the query

   */

  public DBSimpleQuery( Connection conn ) 

                  throws IOException, 

                         SQLException, 

                         ClassNotFoundException

  {

    this.conn = conn;

  }

  /** 

   * routine to search the elements and attributes looking to match query

   *

   * @param query the text to search for

   */

  public Hashtable findDocuments(String query) {

      Hashtable	nodeListResult = new Hashtable();

      Hashtable	rootListResult = new Hashtable();

      Hashtable	 docListResult = new Hashtable();

      PreparedStatement pstmt;

      long nodeid;

      try {

        pstmt =

          conn.prepareStatement("SELECT nodeid " +

                  "FROM xml_nodes WHERE nodedata LIKE ?");

        // Bind the values to the query

        pstmt.setString(1, query);

        pstmt.execute();

        try {

          ResultSet rs = pstmt.getResultSet();

          try {

            boolean tableHasRows = rs.next();

            while (tableHasRows) {

              try {

                nodeid = rs.getLong(1);

                nodeListResult.put(new Long(nodeid),new Long(nodeid));

              } catch (SQLException e) {

                System.out.println("Error with getInt: " + e.getMessage());

              }

              // Advance to the next record in the cursor

              tableHasRows = rs.next();

            }

          } catch (SQLException e) {

            System.out.println("Error with next: " + e.getMessage());

          }

        } catch (SQLException e) {

          System.out.println("Error with getrset: " + e.getMessage());

        }

        pstmt.close();

      } catch (SQLException e) {

        System.out.println("Error getting id: " + e.getMessage());

      }

      Enumeration nodelist = nodeListResult.keys(); 

      while (nodelist.hasMoreElements()) {

        nodeid = ((Long)nodelist.nextElement()).longValue();

        try {

          pstmt =

            conn.prepareStatement("SELECT nodeid " +

                    "FROM xml_nodes START WITH nodeid = ? " +

                    "CONNECT BY nodeid = PRIOR parentnodeid");

          // Bind the values to the query

          pstmt.setLong(1, nodeid);

          pstmt.execute();

          try {

            ResultSet rs = pstmt.getResultSet();

            try {

              boolean tableHasRows = rs.next();

              while (tableHasRows) {

                try {

                  nodeid = rs.getLong(1);

                } catch (SQLException e) {

                  System.out.println("Error with getInt: " + e.getMessage());

                }

                // Advance to the next record in the cursor

                tableHasRows = rs.next();

              }

            } catch (SQLException e) {

              System.out.println("Error with next: " + e.getMessage());

            }

          } catch (SQLException e) {

            System.out.println("Error with getrset: " + e.getMessage());

          }

          pstmt.close();

        } catch (SQLException e) {

          System.out.println("Error getting id: " + e.getMessage());

        }

        // Record that the last record should be the root

        rootListResult.put(new Long(nodeid),new Long(nodeid));

      } 

      // Now look up the document id

      long docid = 0;

      String docname = null;

      String doctype = null;

      String doctitle = null;

      StringBuffer document = null; 

      Enumeration rootlist = rootListResult.keys();

      while (rootlist.hasMoreElements()) {

        nodeid = ((Long)rootlist.nextElement()).longValue();

        try {

          pstmt =

            conn.prepareStatement("SELECT docid,docname,doctype,doctitle " +

                    "FROM xml_documents " +

                    "WHERE rootnodeid = ?");

          // Bind the values to the query

          pstmt.setLong(1, nodeid);

          pstmt.execute();

          try {

            ResultSet rs = pstmt.getResultSet();

            try {

              boolean tableHasRows = rs.next();

              while (tableHasRows) {

                try {

                  docid = rs.getLong(1);

                  docname = rs.getString(2);

                  doctype = rs.getString(3);

                  doctitle = rs.getString(4);

                  document = new StringBuffer();

                  document.append("<docid>").append(docid).append("</docid>");

                  if (docname != null) {

                    document.append("<docname>" + docname + "</docname>");

                  }

                  if (doctype != null) {

                    document.append("<doctype>" + doctype + "</doctype>");

                  }

                  if (doctitle != null) {

                    document.append("<doctitle>" + doctitle + "</doctitle>");

                  }

                } catch (SQLException e) {

                  System.out.println("Error with getLong: " + e.getMessage());

                }

                // Advance to the next record in the cursor

                tableHasRows = rs.next();

              }

            } catch (SQLException e) {

              System.out.println("Error with next: " + e.getMessage());

            }

          } catch (SQLException e) {

            System.out.println("Error with getrset: " + e.getMessage());

          }

          pstmt.close();

        } catch (SQLException e) {

          System.out.println("Error getting id: " + e.getMessage());

        }

        // Store the document id and the root node id

        docListResult.put(new Long(docid),(String)document.toString());

      }

      return docListResult;

  }

}