Metacat

/**

 *  '$RCSfile$'

 *    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

 *

 *   '$Author: tao $'

 *     '$Date: 2007-08-02 16:58:44 -0700 (Thu, 02 Aug 2007) $'

 * '$Revision: 3342 $'

 *

 * 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.io.*;

import java.util.zip.*;

import java.sql.PreparedStatement;

import java.sql.ResultSet;

import java.sql.SQLException;

import java.util.*;

import javax.servlet.ServletOutputStream;

import javax.servlet.http.HttpServletResponse;

import javax.servlet.http.HttpSession;

import org.apache.log4j.Logger;

import org.w3c.dom.*;

import javax.xml.parsers.DocumentBuilderFactory;

import org.xml.sax.InputSource;

import org.w3c.dom.ls.*;

import edu.ucsb.nceas.morpho.datapackage.Triple;

import edu.ucsb.nceas.morpho.datapackage.TripleCollection;

/**

 * 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

 */

public class DBQuery

{

    static final int ALL = 1;

    static final int WRITE = 2;

    static final int READ = 4;

    //private Connection conn = null;

    private String parserName = null;

    private MetaCatUtil util = new MetaCatUtil();

    private Logger logMetacat = Logger.getLogger(DBQuery.class);

    /** true if the metacat spatial option is installed **/

    private final boolean METACAT_SPATIAL = true;

    /** useful if you just want to grab a list of docids **/

    Vector docidOverride = new Vector();

    // a hash table serves as query reuslt cache. Key of hashtable

    // is a query string and value is result xml string

    private static Hashtable queryResultCache = new Hashtable();

    // Capacity of the query result cache

    private static final int QUERYRESULTCACHESIZE = Integer.parseInt(MetaCatUtil.getOption("queryresult_cache_size"));

    /**

     * the main routine used to test the DBQuery utility.

     * <p>

     * Usage: java DBQuery <xmlfile>

     *

     * @param xmlfile the filename of the xml file containing the query

     */

    static public void main(String[] args)

    {

        if (args.length < 1) {

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

            System.err.println("USAGE: java DBQuery [-t] [-index] <xmlfile>");

            return;

        } else {

            try {

                int i = 0;

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

                // Open a connection to the database

                MetaCatUtil util = new MetaCatUtil();

                //Connection dbconn = util.openDBConnection();

                double connTime = System.currentTimeMillis();

                // Execute the query

                DBQuery queryobj = new DBQuery();

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

                Hashtable nodelist = null;

                //nodelist = queryobj.findDocuments(xml, null, null, useXMLIndex);

                // Print the reulting document listing

                StringBuffer result = new StringBuffer();

                String document = null;

                String docid = null;

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

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

                if (!showRuntime) {

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

                }

                // Time the request if asked for

                double stopTime = System.currentTimeMillis();

                double dbOpenTime = (connTime - startTime) / 1000;

                double readTime = (stopTime - connTime) / 1000;

                double executionTime = (stopTime - startTime) / 1000;

                if (showRuntime) {

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

                    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"

                if (!showRuntime) {

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

                System.err.println("Error in DBQuery.main");

                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 parserName the fully qualified name of a Java class implementing

     *            the org.xml.sax.XMLReader interface

     */

    public DBQuery()

    {

        String parserName = MetaCatUtil.getOption("saxparser");

        this.parserName = parserName;

    }

    /**

     * 

     * Construct an instance of DBQuery Class

     * BUT accept a docid Vector that will supersede

     * the query.printSQL() method

     *

     * If a docid Vector is passed in,

     * the docids will be used to create a simple IN query 

     * without the multiple subselects of the printSQL() method

     *

     * Using this constructor, we just check for 

     * a docidOverride Vector in the findResultDoclist() method

     *

     * @param docids List of docids to display in the resultset

     */

    public DBQuery(Vector docids)

    {

        this.docidOverride = docids;

        String parserName = MetaCatUtil.getOption("saxparser");

        this.parserName = parserName;

    }

  /**

   * Method put the search result set into out printerwriter

   * @param resoponse the return response

   * @param out the output printer

   * @param params the paratermer hashtable

   * @param user the user name (it maybe different to the one in param)

   * @param groups the group array

   * @param sessionid  the sessionid

   */

  public void findDocuments(HttpServletResponse response,

                                       PrintWriter out, Hashtable params,

                                       String user, String[] groups,

                                       String sessionid)

  {

    boolean useXMLIndex = (new Boolean(MetaCatUtil.getOption("usexmlindex")))

               .booleanValue();

    findDocuments(response, out, params, user, groups, sessionid, useXMLIndex);

  }

    /**

     * Method put the search result set into out printerwriter

     * @param resoponse the return response

     * @param out the output printer

     * @param params the paratermer hashtable

     * @param user the user name (it maybe different to the one in param)

     * @param groups the group array

     * @param sessionid  the sessionid

     */

    public void findDocuments(HttpServletResponse response,

                                         PrintWriter out, Hashtable params,

                                         String user, String[] groups,

                                         String sessionid, boolean useXMLIndex)

    {

      int pagesize = 0;

      int pagestart = 0;

      if(params.containsKey("pagesize") && params.containsKey("pagestart"))

      {

        String pagesizeStr = ((String[])params.get("pagesize"))[0];

        String pagestartStr = ((String[])params.get("pagestart"))[0];

        if(pagesizeStr != null && pagestartStr != null)

        {

          pagesize = (new Integer(pagesizeStr)).intValue();

          pagestart = (new Integer(pagestartStr)).intValue();

        }

      }

      // get query and qformat

      String xmlquery = ((String[])params.get("query"))[0];

      logMetacat.info("SESSIONID: " + sessionid);

      logMetacat.info("xmlquery: " + xmlquery);

      String qformat = ((String[])params.get("qformat"))[0];

      logMetacat.info("qformat: " + qformat);

      // Get the XML query and covert it into a SQL statment

      QuerySpecification qspec = null;

      if ( xmlquery != null)

      {

         xmlquery = transformQuery(xmlquery);

         try

         {

           qspec = new QuerySpecification(xmlquery,

                                          parserName,

                                          MetaCatUtil.getOption("accNumSeparator"));

         }

         catch (Exception ee)

         {

           logMetacat.error("error generating QuerySpecification object"

                                    +" in DBQuery.findDocuments"

                                    + ee.getMessage());

         }

      }

      if (qformat != null && qformat.equals(MetaCatServlet.XMLFORMAT))

      {

        //xml format

        response.setContentType("text/xml");

        createResultDocument(xmlquery, qspec, out, user, groups, useXMLIndex, 

          pagesize, pagestart, sessionid);

      }//if

      else

      {

        //knb format, in this case we will get whole result and sent it out

        response.setContentType("text/html");

        PrintWriter nonout = null;

        StringBuffer xml = createResultDocument(xmlquery, qspec, nonout, user,

                                                groups, useXMLIndex, pagesize, 

                                                pagestart, sessionid);

        //transfer the xml to html

        try

        {

         double startHTMLTransform = System.currentTimeMillis()/1000;

         DBTransform trans = new DBTransform();

         response.setContentType("text/html");

         // if the user is a moderator, then pass a param to the 

         // xsl specifying the fact

         if(MetaCatUtil.isModerator(user, groups)){

        	 params.put("isModerator", new String[] {"true"});

         }

         trans.transformXMLDocument(xml.toString(), "-//NCEAS//resultset//EN",

                                 "-//W3C//HTML//EN", qformat, out, params,

                                 sessionid);

         double endHTMLTransform = System.currentTimeMillis()/1000;

          logMetacat.warn("The time to transfrom resultset from xml to html format is "

                  		                             +(endHTMLTransform -startHTMLTransform));

          MetaCatUtil.writeDebugToFile("---------------------------------------------------------------------------------------------------------------Transfrom xml to html  "

                             +(endHTMLTransform -startHTMLTransform));

          MetaCatUtil.writeDebugToDelimiteredFile(" "+(endHTMLTransform -startHTMLTransform), false);

        }

        catch(Exception e)

        {

         logMetacat.error("Error in MetaCatServlet.transformResultset:"

                                +e.getMessage());

         }

      }//else

  }

  /**

   * Transforms a hashtable of documents to an xml or html result and sent

   * the content to outputstream. Keep going untill hastable is empty. stop it.

   * add the QuerySpecification as parameter is for ecogrid. But it is duplicate

   * to xmlquery String

   * @param xmlquery

   * @param qspec

   * @param out

   * @param user

   * @param groups

   * @param useXMLIndex

   * @param sessionid

   * @return

   */

    public StringBuffer createResultDocument(String xmlquery,

                                              QuerySpecification qspec,

                                              PrintWriter out,

                                              String user, String[] groups,

                                              boolean useXMLIndex)

    {

    	return createResultDocument(xmlquery,qspec,out, user,groups, useXMLIndex, 0, 0,"");

    }

  /*

   * Transforms a hashtable of documents to an xml or html result and sent

   * the content to outputstream. Keep going untill hastable is empty. stop it.

   * add the QuerySpecification as parameter is for ecogrid. But it is duplicate

   * to xmlquery String

   */

  public StringBuffer createResultDocument(String xmlquery,

                                            QuerySpecification qspec,

                                            PrintWriter out,

                                            String user, String[] groups,

                                            boolean useXMLIndex, int pagesize,

                                            int pagestart, String sessionid)

  {

    DBConnection dbconn = null;

    int serialNumber = -1;

    StringBuffer resultset = new StringBuffer();

    //try to get the cached version first    

    Hashtable sessionHash = MetaCatServlet.getSessionHash();

    HttpSession sess = (HttpSession)sessionHash.get(sessionid);

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

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

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

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

    resultset.append("  <nextpage>" + (pagestart + 1) + "</nextpage>\n");

    resultset.append("  <previouspage>" + (pagestart - 1) + "</previouspage>\n");

    resultset.append("  <query>" + xmlquery + "</query>");

    //send out a new query

    if (out != null)

    {

      out.println(resultset.toString());

    }

    if (qspec != null)

    {

      try

      {

        //checkout the dbconnection

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

        serialNumber = dbconn.getCheckOutSerialNumber();

        //print out the search result

        // search the doc list

        StringBuffer resultContent = findResultDoclist(qspec, out, user, groups,

                                      dbconn, useXMLIndex, pagesize, pagestart, 

                                      sessionid);

        resultset.append(resultContent);

      } //try

      catch (IOException ioe)

      {

        logMetacat.error("IO error in DBQuery.findDocuments:");

        logMetacat.error(ioe.getMessage());

      }

      catch (SQLException e)

      {

        logMetacat.error("SQL Error in DBQuery.findDocuments: "

                                 + e.getMessage());

      }

      catch (Exception ee)

      {

        logMetacat.error("Exception in DBQuery.findDocuments: "

                                 + ee.getMessage());

        ee.printStackTrace();

      }

      finally

      {

        DBConnectionPool.returnDBConnection(dbconn, serialNumber);

      } //finally

    }//if

    String closeRestultset = "</resultset>";

    resultset.append(closeRestultset);

    if (out != null)

    {

      out.println(closeRestultset);

    }

    //default to returning the whole resultset

    return resultset;

  }//createResultDocuments

    /*

     * Find the doc list which match the query

     */

    private StringBuffer findResultDoclist(QuerySpecification qspec,

                                      PrintWriter out,

                                      String user, String[]groups,

                                      DBConnection dbconn, boolean useXMLIndex,

                                      int pagesize, int pagestart, String sessionid)

                                      throws Exception

    {

      StringBuffer resultsetBuffer = new StringBuffer();

      String query = null;

      int count = 0;

      int index = 0;

      ResultDocumentSet docListResult = new ResultDocumentSet();

      PreparedStatement pstmt = null;

      String docid = null;

      String docname = null;

      String doctype = null;

      String createDate = null;

      String updateDate = null;

      StringBuffer document = null;

      boolean lastpage = false;

      int rev = 0;

      double startTime = 0;

      //int offset = 1;

      double startSelectionTime = System.currentTimeMillis()/1000;

      ResultSet rs = null;

      //offset = 1;

      // this is a hack for offset

      /*if (out == null)

      {

        // for html page, we put everything into one page

        offset =

            (new Integer(MetaCatUtil.getOption("web_resultsetsize"))).intValue();

      }

      else

      {

          offset =

              (new Integer(MetaCatUtil.getOption("app_resultsetsize"))).intValue();

      }*/

      /*

       * Check the docidOverride Vector

       * if defined, we bypass the qspec.printSQL() method

       * and contruct a simpler query based on a 

       * list of docids rather than a bunch of subselects

       */

      if ( this.docidOverride.size() == 0 ) {

          query = qspec.printSQL(useXMLIndex);

      } else {

          logMetacat.info("*** docid override " + this.docidOverride.size());

          StringBuffer queryBuffer = new StringBuffer( "SELECT docid,docname,doctype,date_created, date_updated, rev " );

          queryBuffer.append( " FROM xml_documents WHERE docid IN (" );

          for (int i = 0; i < docidOverride.size(); i++) {  

              queryBuffer.append("'");

              queryBuffer.append( (String)docidOverride.elementAt(i) );

              queryBuffer.append("',");

          }

          // empty string hack 

          queryBuffer.append( "'') " );

          query = queryBuffer.toString();

      } 

      String ownerQuery = getOwnerQuery(user);

      logMetacat.info("\n\n\n query: " + query);

      logMetacat.info("\n\n\n owner query: "+ownerQuery);

      // 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();

        if(!query.endsWith("WHERE")){

            query = query + accessQuery;

        } else {

            query = query + accessQuery.substring(4, accessQuery.length());

        }

      }

      logMetacat.warn("============ final selection query: " + query);

      // we only get cache for public

      if (user != null && user.equalsIgnoreCase("public") 

     		 && pagesize == 0 && MetaCatUtil.getOption("query_cache_on").equals("true"))

      {

   	      String cachedResult = getResultXMLFromCache(query);

   	      //System.out.println("==========the string from cache is "+cachedResult);

   	      if (cachedResult != null)

   	      {

   	    	 if (out != null)

   	         {

   	             out.println(cachedResult);

   	         }

   	    	 resultsetBuffer.append(cachedResult);

   	    	 return resultsetBuffer;

   	      }

      }

      startTime = System.currentTimeMillis() / 1000;

      pstmt = dbconn.prepareStatement(query);

      rs = pstmt.executeQuery();

      double queryExecuteTime = System.currentTimeMillis() / 1000;

      logMetacat.warn("Time to execute select docid query is "

                    + (queryExecuteTime - startTime));

      MetaCatUtil.writeDebugToFile("\n\n\n\n\n\nExecute selection query  "

              + (queryExecuteTime - startTime));

      MetaCatUtil.writeDebugToDelimiteredFile(""+(queryExecuteTime - startTime), false);

      boolean tableHasRows = rs.next();

      if(pagesize == 0)

      { //this makes sure we get all results if there is no paging

        pagesize = 9999999;

        pagestart = 9999999;

      } 

      int currentIndex = 0;

      while (tableHasRows)

      {

        logMetacat.info("############getting result: " + currentIndex);

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

        logMetacat.info("############processing: " + docid);

        docname = rs.getString(2);

        doctype = rs.getString(3);

        logMetacat.info("############processing: " + doctype);

        createDate = rs.getString(4);

        updateDate = rs.getString(5);

        rev = rs.getInt(6);

         Vector returndocVec = qspec.getReturnDocList();

       if (returndocVec.size() == 0 || returndocVec.contains(doctype))

        {

          logMetacat.info("NOT Back tracing now...");

           document = new StringBuffer();

           String completeDocid = docid

                            + MetaCatUtil.getOption("accNumSeparator");

           completeDocid += rev;

           document.append("<docid>").append(completeDocid).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.addResultDocument(

             new ResultDocument(docid, (String) document.toString()));

           logMetacat.info("$$$$$$$real result: " + docid);

           currentIndex++;

           count++;

        }//else

        // when doclist reached the offset number, send out doc list and empty

        // the hash table

        /*if (count == offset && pagesize == 0)

        { //if pagesize is not 0, do this later.

          //reset count

          //logMetacat.warn("############doing subset cache");

          count = 0;

          handleSubsetResult(qspec, resultsetBuffer, out, docListResult,

                              user, groups,dbconn, useXMLIndex);

          //reset docListResult

          docListResult = new ResultDocumentSet();

        }*/

       logMetacat.info("currentIndex: " + currentIndex);

       logMetacat.info("page comparator: " + (pagesize * pagestart) + pagesize);

       if(currentIndex >= ((pagesize * pagestart) + pagesize))

       {

         ResultDocumentSet pagedResultsHash = new ResultDocumentSet();

         for(int i=pagesize*pagestart; i<docListResult.size(); i++)

         {

           pagedResultsHash.put(docListResult.get(i));

         }

         docListResult = pagedResultsHash;

         break;

       }

       // Advance to the next record in the cursor

       tableHasRows = rs.next();

       if(!tableHasRows)

       {

         ResultDocumentSet pagedResultsHash = new ResultDocumentSet();

         //get the last page of information then break

         if(pagesize != 9999999)

         {

           for(int i=pagesize*pagestart; i<docListResult.size(); i++)

           {

             pagedResultsHash.put(docListResult.get(i));

           }

           docListResult = pagedResultsHash;

         }

         lastpage = true;

         break;

       }

     }//while

     rs.close();

     pstmt.close();

     double docListTime = System.currentTimeMillis() / 1000;

     logMetacat.warn("======Total time to get docid list is: "

                          + (docListTime - startSelectionTime ));

     MetaCatUtil.writeDebugToFile("---------------------------------------------------------------------------------------------------------------Total selection: "

             + (docListTime - startSelectionTime ));

     MetaCatUtil.writeDebugToDelimiteredFile(" "+ (docListTime - startSelectionTime ), false);

     //if docListResult is not empty, it need to be sent.

     if (docListResult.size() != 0)

     {

       handleSubsetResult(qspec,resultsetBuffer, out, docListResult,

                              user, groups,dbconn, useXMLIndex);

     }

     resultsetBuffer.append("\n<lastpage>" + lastpage + "</lastpage>\n");

     if (out != null)

     {

         out.println("\n<lastpage>" + lastpage + "</lastpage>\n");

     }

     // now we only cached none-paged query and user is public

     if (user != null && user.equalsIgnoreCase("public") 

    		 && pagesize == 9999999 && MetaCatUtil.getOption("query_cache_on").equals("true"))

     {

       //System.out.println("the string stored into cache is "+ resultsetBuffer.toString());

  	   storeQueryResultIntoCache(query, resultsetBuffer.toString());

     }

     return resultsetBuffer;

    }//findReturnDoclist

    /*

     * Send completed search hashtable(part of reulst)to output stream

     * and buffer into a buffer stream

     */

    private StringBuffer handleSubsetResult(QuerySpecification qspec,

                                           StringBuffer resultset,

                                           PrintWriter out, ResultDocumentSet partOfDoclist,

                                           String user, String[]groups,

                                       DBConnection dbconn, boolean useXMLIndex)

                                       throws Exception

   {

     double startReturnField = System.currentTimeMillis()/1000;

     // check if there is a record in xml_returnfield

     // and get the returnfield_id and usage count

     int usage_count = getXmlReturnfieldsTableId(qspec, dbconn);

     boolean enterRecords = false;

     // get value of xml_returnfield_count

     int count = (new Integer(MetaCatUtil

                            .getOption("xml_returnfield_count")))

                            .intValue();

     // set enterRecords to true if usage_count is more than the offset

     // specified in metacat.properties

     if(usage_count > count){

         enterRecords = true;

     }

     if(returnfield_id < 0){

         logMetacat.warn("Error in getting returnfield id from"

                                  + "xml_returnfield table");

         enterRecords = false;

     }

     // get the hashtable containing the docids that already in the

     // xml_queryresult table

     logMetacat.info("size of partOfDoclist before"

                             + " docidsInQueryresultTable(): "

                             + partOfDoclist.size());

     double startGetReturnValueFromQueryresultable = System.currentTimeMillis()/1000;

     Hashtable queryresultDocList = docidsInQueryresultTable(returnfield_id,

                                                        partOfDoclist, dbconn);

     // remove the keys in queryresultDocList from partOfDoclist

     Enumeration _keys = queryresultDocList.keys();

     while (_keys.hasMoreElements()){

         partOfDoclist.remove((String)_keys.nextElement());

     }

     double endGetReturnValueFromQueryresultable = System.currentTimeMillis()/1000;

     logMetacat.warn("Time to get return fields from xml_queryresult table is (Part1 in return fields) " +

          		               (endGetReturnValueFromQueryresultable-startGetReturnValueFromQueryresultable));

     MetaCatUtil.writeDebugToFile("-----------------------------------------Get fields from xml_queryresult(Part1 in return fields) " +

               (endGetReturnValueFromQueryresultable-startGetReturnValueFromQueryresultable));

     MetaCatUtil.writeDebugToDelimiteredFile(" " +

             (endGetReturnValueFromQueryresultable-startGetReturnValueFromQueryresultable),false);

     // backup the keys-elements in partOfDoclist to check later

     // if the doc entry is indexed yet

     Hashtable partOfDoclistBackup = new Hashtable();

     Iterator itt = partOfDoclist.getDocids();

     while (itt.hasNext()){

       Object key = itt.next();

         partOfDoclistBackup.put(key, partOfDoclist.get(key));

     }

     logMetacat.info("size of partOfDoclist after"

                             + " docidsInQueryresultTable(): "

                             + partOfDoclist.size());

     //add return fields for the documents in partOfDoclist

     partOfDoclist = addReturnfield(partOfDoclist, qspec, user, groups,

                                        dbconn, useXMLIndex);

     double endExtendedQuery = System.currentTimeMillis()/1000;

     logMetacat.warn("Get fields from index and node table (Part2 in return fields) "

        		                                          + (endExtendedQuery - endGetReturnValueFromQueryresultable));

     MetaCatUtil.writeDebugToFile("-----------------------------------------Get fields from extened query(Part2 in return fields) "

             + (endExtendedQuery - endGetReturnValueFromQueryresultable));

     MetaCatUtil.writeDebugToDelimiteredFile(" "

             + (endExtendedQuery - endGetReturnValueFromQueryresultable), false);

     //add relationship part part docid list for the documents in partOfDocList

     partOfDoclist = addRelationship(partOfDoclist, qspec, dbconn, useXMLIndex);

     double startStoreReturnField = System.currentTimeMillis()/1000;

     Iterator keys = partOfDoclist.getDocids();

     String key = null;

     String element = null;

     String query = null;

     int offset = (new Integer(MetaCatUtil

                               .getOption("queryresult_string_length")))

                               .intValue();

     while (keys.hasNext())

     {

         key = (String) keys.next();

         element = (String)partOfDoclist.get(key);

	 // check if the enterRecords is true, elements is not null, element's

         // length is less than the limit of table column and if the document

         // has been indexed already

         if(enterRecords && element != null

		&& element.length() < offset

		&& element.compareTo((String) partOfDoclistBackup.get(key)) != 0){

             query = "INSERT INTO xml_queryresult (returnfield_id, docid, "

                 + "queryresult_string) VALUES (?, ?, ?)";

             PreparedStatement pstmt = null;

             pstmt = dbconn.prepareStatement(query);

             pstmt.setInt(1, returnfield_id);

             pstmt.setString(2, key);

             pstmt.setString(3, element);

             dbconn.increaseUsageCount(1);

             pstmt.execute();

             pstmt.close();

         }

         // A string with element

         String xmlElement = "  <document>" + element + "</document>";

         //send single element to output

         if (out != null)

         {

             out.println(xmlElement);

         }

         resultset.append(xmlElement);

     }//while

     double endStoreReturnField = System.currentTimeMillis()/1000;

     logMetacat.warn("Time to store new return fields into xml_queryresult table (Part4 in return fields) "

                   + (endStoreReturnField -startStoreReturnField));

     MetaCatUtil.writeDebugToFile("-----------------------------------------Insert new record to xml_queryresult(Part4 in return fields) "

             + (endStoreReturnField -startStoreReturnField));

     MetaCatUtil.writeDebugToDelimiteredFile(" "

             + (endStoreReturnField -startStoreReturnField), false);

     Enumeration keysE = queryresultDocList.keys();

     while (keysE.hasMoreElements())

     {

         key = (String) keysE.nextElement();

         element = (String)queryresultDocList.get(key);

         // A string with element

         String xmlElement = "  <document>" + element + "</document>";

         //send single element to output

         if (out != null)

         {

             out.println(xmlElement);

         }

         resultset.append(xmlElement);

     }//while

     double returnFieldTime = System.currentTimeMillis() / 1000;

     logMetacat.warn("======Total time to get return fields is: "

                           + (returnFieldTime - startReturnField));

     MetaCatUtil.writeDebugToFile("---------------------------------------------------------------------------------------------------------------"+

    		 "Total to get return fields  "

                                   + (returnFieldTime - startReturnField));

     MetaCatUtil.writeDebugToDelimiteredFile(" "+ (returnFieldTime - startReturnField), false);

     return resultset;

 }

   /**

    * Get the docids already in xml_queryresult table and corresponding

    * queryresultstring as a hashtable

    */

   private Hashtable docidsInQueryresultTable(int returnfield_id,

                                              ResultDocumentSet partOfDoclist,

                                              DBConnection dbconn){

         Hashtable returnValue = new Hashtable();

         PreparedStatement pstmt = null;

         ResultSet rs = null;

         // get partOfDoclist as string for the query

         Iterator keylist = partOfDoclist.getDocids();

         StringBuffer doclist = new StringBuffer();

         while (keylist.hasNext())

         {

             doclist.append("'");

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

             doclist.append("',");

         }//while

         if (doclist.length() > 0)

         {

             doclist.deleteCharAt(doclist.length() - 1); //remove the last comma