Metacat

/**

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

 *             attributes that have free text matches a query string,

 *             or structured query matches to a path specified node in the

 *             XML hierarchy.  It returns a result set consisting of the

 *             document ID 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

 *     '$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

import java.net.MalformedURLException;

import java.sql.*;

import java.util.Stack;

import java.util.Hashtable;

import java.util.Enumeration;

import java.io.FileWriter;

import java.io.BufferedWriter;

/**

 * attributes that have free text matches a query string,

 * or structured query matches to a path specified node in the

 * XML hierarchy.  It returns a result set consisting of the

 * document ID for each document that satisfies the query

public class DBQuery {

  static final int WRITE = 2;

  static final int READ = 4;

   * the main routine used to test the DBQuery utility.

   * Usage: java DBQuery <xmlfile>

  static public void main(String[] args) {

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

     } else {

        try {

          boolean showRuntime = false;

          boolean useXMLIndex = false;

          if ( args[i].equals( "-t" ) ) {

            showRuntime = true;

            i++;

          }

          if ( args[i].equals( "-index" ) ) {

            useXMLIndex = true;

            i++;

          }

          String xmlfile  = args[i];

          // Time the request if asked for

          double startTime = System.currentTimeMillis();

          String document = null;

            Enumeration doclist = nodelist.keys();

            while (doclist.hasMoreElements()) {

              docid = (String)doclist.nextElement();

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

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

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

            }

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

          double stopTime = System.currentTimeMillis();

            System.out.print("  " + dbOpenTime);

            System.out.print("  " + readTime);

            System.out.print("  " + nodelist.size());

            System.out.println();

          //System.out.println(result);

          //write into a file "result.txt"

            File f = new File("./result.txt");

            FileWriter fw = new FileWriter(f);

            BufferedWriter out = new BufferedWriter(fw);

            out.write(result.toString());

            out.flush();

            out.close();

            fw.close();

          }

        }

        catch (Exception e) {

          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

                         SQLException,

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

   *

   * @param xmlquery the xml serialization of the query (@see pathquery.dtd)

   * @param user the username of the user

   * @param group the group of the user

   */

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

   *

   * @param group the group of the user

      String doctype = null;

      String updateDate = null;

      String fieldname = null;

      String fielddata = null;

      //Connection dbconn2 = null;

      DBConnection dbconn = null;

      int serialNumber = -1;

        dbconn=DBConnectionPool.getDBConnection("DBQuery.findDocuments");

        serialNumber=dbconn.getCheckOutSerialNumber();

        // if query is not the owner query, we need to check the permission

        // otherwise we don't need (owner has all permission by default)

        if (!query.equals(ownerQuery))

        {

          // set user name and group

          qspec.setUserName(user);

          qspec.setGroup(groups);

          // Get access query

          String accessQuery = qspec.getAccessQuery();

          query = query + accessQuery;

          MetaCatUtil.debugMessage(" final query: "+query, 30);

        }

        pstmt = dbconn.prepareStatement(query);

        ResultSet rs = pstmt.getResultSet();

        MetaCatUtil.debugMessage("Time for execute query: "+

                                            (queryExecuteTime -startTime), 30);

        {

          //boolean permit =hasPermission(user, groups, docid);

          //long checkTimeEnd = System.currentTimeMillis();

          //MetaCatUtil.debugMessage("check permission time: "+

                                  //(checkTimeEnd - checkTimeStart), 30);

          //if ( !permit ) {

            //continue;

          //}

          doctype = rs.getString(3);

          updateDate = rs.getString(5);

          rev = rs.getInt(6);

          // otherwise, just return the document that was hit

          Vector returndocVec = qspec.getReturnDocList();

              && !qspec.isPercentageSearch())

            {

              btBuf.append("'").append((String)returndocVec.get(i)).append("'");

                btBuf.append(", ");

            btBuf.append(") ");

            btBuf.append("and (subject like '");

                                       prepareStatement(btBuf.toString());

            dbconn.increaseUsageCount(1);

            ResultSet btrs = npstmt.getResultSet();

              DocumentImpl xmldoc = null;

              {

                // xmldoc = new DocumentImpl(dbconn, packageDocid);

                //  thus use the following to get the doc info only

                //  xmldoc = new DocumentImpl(dbconn);

              catch(Exception e)

              {

                                   "DBQuery.findDocuments: " + e.getMessage());

              if (docid_org == null) {

              docid   = docid_org.trim();

              doctype = xmldoc.getDoctype();

              createDate = xmldoc.getCreateDate();

              updateDate = xmldoc.getUpdateDate();

              document = new StringBuffer();

              String completeDocid = docid + util.getOption("accNumSeparator");

              completeDocid += rev;

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

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

              if (docname != null) {

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

              }

              if (doctype != null) {

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

              }

              if (createDate != null) {

                document.append("<createdate>" + createDate + "</createdate>");

              }

              if (updateDate != null) {

                document.append("<updatedate>" + updateDate + "</updatedate>");

              }

              // Store the document id and the root node id

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

              // Get the next package document linked to our hit

              hasBtRows = btrs.next();

          else if (returndocVec.size() != 0 && returndocVec.contains(doctype))

          {

            completeDocid += rev;

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

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

            }

            if (doctype != null) {

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

            }

            }

            }

            // Store the document id and the root node id

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

          // Advance to the next record in the cursor

          tableHasRows = rs.next();

        }

        MetaCatUtil.debugMessage("prepare docid list time: "

                                          +(docListTime-queryExecuteTime), 30);

          Vector extendedFields = new Vector(qspec.getReturnFieldList());

          Vector results = new Vector();

          StringBuffer doclist = new StringBuffer();

          Hashtable returnFieldValue = new Hashtable();

          {

            doclist.append("'");

            doclist.append((String)keylist.nextElement());

            doclist.append("',");

          }

          {

            Hashtable controlPairs = new Hashtable();

            double extendedQueryStart = System.currentTimeMillis()/1000;

            String accessControlSQL = qspec.

                        printAccessControlSQLForReturnField(doclist.toString());

            pstmt = dbconn.prepareStatement(accessControlSQL);

            //increase dbconnection usage count

            dbconn.increaseUsageCount(1);

            pstmt.execute();

            rs = pstmt.getResultSet();

            tableHasRows = rs.next();

            while(tableHasRows)

            {

              long startNodeId = rs.getLong(1);

              long endNodeId = rs.getLong(2);

              controlPairs.put(new Long(startNodeId), new Long(endNodeId));

            }

            double extendedAccessQueryEnd = System.currentTimeMillis()/1000;

            MetaCatUtil.debugMessage("Time for execute access extended query: "

                              +(extendedAccessQueryEnd-extendedQueryStart), 30);

                                                          controlPairs);

            dbconn.increaseUsageCount(1);

            rs = pstmt.getResultSet();

            MetaCatUtil.debugMessage("Time for execute extended query: "

                                    +(extendedQueryEnd-extendedQueryStart), 30);

              fieldname = rs.getString(2);

              fielddata = rs.getString(3);

              StringBuffer value = new StringBuffer();

              if (!parentidList.containsKey(parentId))

              {

                // don't need to merger nodedata

                value.append("<param name=\"");

                value.append(fieldname);

                value.append("\">");

                value.append(fielddata);

                value.append("</param>");

                //set returnvalue

                returnValue.setDocid(docid);

                returnValue.setFieldValue(fielddata);

                returnValue.setXMLFieldValue(value.toString());

                // Store it in hastable

                parentidList.put(parentId, returnValue);

              }

              else

              {

                // need to merge nodedata if they have same parent id ant

                // node type is text

                fielddata = (String)((ReturnFieldValue)

                       parentidList.get(parentId)).getFieldValue() +  fielddata;

                value.append("<param name=\"");

                value.append(fieldname);

                value.append("\">");

                value.append(fielddata);

                value.append("</param>");

                returnValue.setDocid(docid);

                returnValue.setFieldValue(fielddata);

                returnValue.setXMLFieldValue(value.toString());

                // remove the old return value from paretnidList

                parentidList.remove(parentId);

                // store the new return value in parentidlit

                parentidList.put(parentId, returnValue);

              }

               tableHasRows = rs.next();

            }//while

            rs.close();

            pstmt.close();

            Enumeration xmlFieldValue = parentidList.elements();

            while( xmlFieldValue.hasMoreElements() )

            {

              ReturnFieldValue object = (ReturnFieldValue)

                                         xmlFieldValue.nextElement();

              docid = object.getDocid();

              {

                  docListResult.put(docid, removedelement +

                                    object.getXMLFieldValue());

              else

              {

            MetaCatUtil.debugMessage("Time for prepare doclistresult after"+

                                      " execute extended query: "

                                    +(docListResultEnd-extendedQueryEnd), 30);

            docListResult = getAttributeValueForReturn

                                      (qspec,docListResult, doclist.toString());

          }//if doclist lenght is great than zero

        }//if has extended query

        //this code might be able to be added to the backtracking code above

        while(docidkeys.hasMoreElements())

          String connstring = "%docid=";

          rs = pstmt.getResultSet();

          tableHasRows = rs.next();

          while(tableHasRows)

          {

            String sub = rs.getString(1);

            String rel = rs.getString(2);

            String obj = rs.getString(3);

            String objDT = rs.getString(5);

            document.append("<triple>");

            document.append("<subject>").append(MetaCatUtil.normalize(sub));

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

            if ( subDT != null ) {

              document.append("<subjectdoctype>").append(subDT);

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

            }

                                          append(MetaCatUtil.normalize(rel));

            document.append("<object>").append(MetaCatUtil.normalize(obj));

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

            if ( objDT != null ) {

              document.append("<objectdoctype>").append(objDT);

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

            }

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

            String removedelement = (String)docListResult.remove(docidkey);

            docListResult.put(docidkey, removedelement +

                              document.toString());

          pstmt.close();

        MetaCatUtil.debugMessage("Time for adding relation to docListResult: "+

                                (endRelation-startRelation), 30);

                           e.getMessage());

      {

        {

        }//try

        catch (SQLException sqlE)

                                      +sqlE.getMessage(), 30);

        }//catch

        finally

        {

          DBConnectionPool.returnDBConnection(dbconn, serialNumber);

        }//finally

      }//finally

    //System.out.println(docListResult.toString());

  }

   * field have attribue

   */

  private Hashtable getAttributeValueForReturn(QuerySpecification squery,

                                               Hashtable docInformationList,

                                               String docList)

  {

    StringBuffer XML = null;

    String sql = null;

    DBConnection dbconn = null;

    PreparedStatement pstmt = null;

    ResultSet rs = null;

    int serialNumber = -1;

    boolean tableHasRows =false;

    //check the parameter

    if (squery == null || docList==null || docList.length() <0)

    {

      return docInformationList;

    }

    // if has attribute as return field

    if (squery.containAttributeReturnField())

    {

      sql = squery.printAttributeQuery(docList);

      try

      {

        dbconn=DBConnectionPool.getDBConnection("DBQuery.getAttributeValue");

        serialNumber=dbconn.getCheckOutSerialNumber();

        pstmt = dbconn.prepareStatement(sql);

        pstmt.execute();

        rs = pstmt.getResultSet();

        tableHasRows = rs.next();

        while(tableHasRows)

        {

          String docid = rs.getString(1).trim();

          String fieldname = rs.getString(2);

          String fielddata = rs.getString(3);

          String attirbuteName = rs.getString(4);

          XML = new StringBuffer();

          XML.append("<param name=\"");

          XML.append(fieldname);

          XML.append(QuerySpecification.ATTRIBUTESYMBOL);

          XML.append(attirbuteName);

          XML.append("\">");

          XML.append(fielddata);

          XML.append("</param>");

          tableHasRows = rs.next();

          if (docInformationList.containsKey(docid))