Metacat

/**

 *      Name: QuerySpecification.java

 *   Purpose: A Class that represents a structured query, and can be 

 *            constructed from an XML serialization

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

 *            National Center for Ecological Analysis and Synthesis

 *   Authors: Matt Jones

 *

 *   Version: '$Id$'

 */

package edu.ucsb.nceas.metacat;

import java.io.*;

import java.net.URL;

import java.net.MalformedURLException;

import java.util.Stack;

import java.util.Hashtable;

import java.util.Enumeration;

import org.xml.sax.*;

import oracle.xml.parser.v2.SAXParser;

/** 

 * A Class that represents a structuredd query,and can be constructed from an

 * XML serialization

 */

public class QuerySpecification extends HandlerBase {

  /**

   * construct an instance of the QuerySpecification class 

   *

   * @param queryspec the XML representation of the query as a Reader

   */

  public QuerySpecification( Reader queryspec ) throws IOException {

    super();

/*

    int bytesread = 0;

    while (bytesread != -1) {

      char[] cbuf = new char[1000];

      bytesread = queryspec.read(cbuf);

      System.out.print(cbuf);

    }

*/

    SAXParser parser = initializeParser();

    try {

      parser.parse(new InputSource(queryspec));

    } catch (SAXException e) {

      System.out.println("error parsing data");

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

    }

  }

  /**

   * construct an instance of the QuerySpecification class 

   *

   * @param queryspec the XML representation of the query as a String

   */

  public QuerySpecification( String queryspec ) throws IOException {

    this(new StringReader(queryspec));

  }

  /** Main routine for testing */

  static public void main(String[] args) {

     if (args.length < 1) {

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

       System.err.println("USAGE: java QuerySpecification <xmlfile>");

       return;

     } else {

       String xmlfile  = args[0];

       try {

         FileReader xml = new FileReader(new File(xmlfile));

         QuerySpecification qspec = new QuerySpecification(xml);

       } catch (IOException e) {

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

       }

     }

  }

  private SAXParser initializeParser() {

    SAXParser parser = null;

    //

    // Set up the SAX document handlers for parsing

    //

    try {

      // For all the other interface use the default provided by

      // Handler base

      HandlerBase defHandler = new HandlerBase();

      // Get an instance of the parser

      parser = new SAXParser();

      // Set Handlers in the parser

      // Set the DocumentHandler to XMLDocumentHandler

      parser.setDocumentHandler(this);

      // Set the other Handler to the defHandler

      parser.setErrorHandler(defHandler);

    } catch (Exception e) {

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

    }

    return parser;

  }

  public void startElement (String name, AttributeList atts) 

         throws SAXException {

    System.out.println(name);

  }

  public void endElement (String name) throws SAXException {

  }

}

/**

 *      Name: DBQuery.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$'

 */

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 structured 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 DBQuery {

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

  private Connection	conn = null;

  /**

   * the main routine used to test the DBQuery utility.

   *

   * Usage: java DBQuery <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 < 4)

     {

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

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

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

        return;

     } else {

        try {

          String query    = args[0];

          String doctype  = args[1];

          String user     = args[2];

          String password = args[3];

          String dbstring = null;

          if (args.length <= 4) {

            dbstring = defaultDB;

          } else {

            dbstring = args[4];

          }

          // Open a connection to the database

          Connection dbconn = MetaCatUtil.openDBConnection( 

                              "oracle.jdbc.driver.OracleDriver",

                              dbstring, user, password);

          // Execute the simple query

          DBQuery rd = new DBQuery(dbconn);

          Hashtable nodelist = null;

          nodelist = rd.findDocuments(query);

          // Print the reulting root nodes

          StringBuffer result = new StringBuffer();

          String document = null;

          Long docid = null;

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

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

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

          Enumeration doclist = nodelist.keys(); 

          while (doclist.hasMoreElements()) {

            docid = (Long)doclist.nextElement();

            document = (String)nodelist.get(docid);

            result.append("  <document>\n    " + document + 

                          "\n  </document>\n");

          }

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

          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 DBQuery 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 DBQuery( Connection conn ) 

                  throws IOException, 

                         SQLException, 

                         ClassNotFoundException

  {

    this.conn = conn;

  }

  /** 

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

   *

   * @param queryspec the xml serialization of the query

   * @param requestedDoctype the type of documents to return from the query

   */

  public Hashtable findDocuments(String queryspec) {

      Hashtable	 docListResult = new Hashtable();

      //QuerySpecification qspec = new QuerySpecification(queryspec);

/*

      PreparedStatement pstmt;

      // Now look up the document id

      long docid = 0;

      String docname = null;

      String doctype = null;

      String doctitle = null;

      StringBuffer document = null; 

      try {

        if (requestedDoctype == null || 

            requestedDoctype.equals("any") || 

            requestedDoctype.equals("ANY")) {

          pstmt = conn.prepareStatement(

                "SELECT docid,docname,doctype,doctitle " +

                "FROM xml_documents " +

                "WHERE docid IN " +

                "(SELECT docid " +

                "FROM xml_nodes WHERE nodedata LIKE ? )");

                // Bind the values to the query

                pstmt.setString(1, query);

        } else {

          pstmt = conn.prepareStatement(

                "SELECT docid,docname,doctype,doctitle " +

                "FROM xml_documents " +

                "WHERE docid IN " +

                "(SELECT docid " +

                "FROM xml_nodes WHERE nodedata LIKE ? ) " +

                "AND doctype = ?");

                // Bind the values to the query

                pstmt.setString(1, query);

                pstmt.setString(2, requestedDoctype);

        }

        pstmt.execute();

        ResultSet rs = pstmt.getResultSet();

        boolean tableHasRows = rs.next();

        while (tableHasRows) {

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

          }

          // Store the document id and the root node id

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

          // Advance to the next record in the cursor

          tableHasRows = rs.next();

        }

        pstmt.close();

      } catch (SQLException e) {

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

      }

*/

    return docListResult;

  }

}