Metacat

/**

 *  '$RCSfile$'

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

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

import java.io.File;

import java.io.FileNotFoundException;

import java.io.IOException;

import java.sql.PreparedStatement;

import java.sql.ResultSet;

import java.util.Enumeration;

import java.util.Hashtable;

import java.util.Iterator;

import edu.ucsb.nceas.metacat.DBConnection;

import edu.ucsb.nceas.metacat.DBConnectionPool;

import edu.ucsb.nceas.metacat.MetaCatUtil;

import edu.ucsb.nceas.metacat.Version;

import edu.ucsb.nceas.utilities.Options;

import junit.framework.Test;

import junit.framework.TestCase;

import junit.framework.TestSuite;

/**

 * @author jones

 *

 * Test class for getting idea how long it will take - to build 

 * records xml_queryresult table for a given return fields

 */

public class QueryResultBuilderTest extends TestCase

{

	/** DBConnection pool*/

	private DBConnectionPool connPool = null;

	/** Return field value in xml_returnfield table*/

	private String[] returnFieldString = {"dataset/title", "entityName", "individualName/surName", "keyword"};

	/** Return field id value for above return field string in xml_returnfield*/

	private int returnFieldId = 4;

	static

	  {

		  try

		  {

			  Options.initialize(new File("build/tests/metacat.properties"));

		  }

		  catch(Exception e)

		  {

			  System.err.println("Exception in initialize option in MetacatServletNetTest "+e.getMessage());

		  }

	  }

	/**

     * Create a suite of tests to be run together

     */

    public static Test suite()

    {

        TestSuite suite = new TestSuite();

        suite.addTest(new QueryResultBuilderTest("initialize"));

        //suite.addTest(new QueryResultBuilderTest("testBuildRecords"));

        return suite;

    }

    /**

     * Constructor to build the test

     * 

     * @param name the name of the test method

     */

    public QueryResultBuilderTest(String name)

    {

        super(name);

    }

	/*

     * @see TestCase#setUp()

     */

    protected void setUp() throws Exception

    {

        super.setUp();

        try {

        	connPool = DBConnectionPool.getInstance();      

        } catch (Exception e) {

            fail(e.getMessage());

        }

    }

    /**

     * Run an initial test that always passes to check that the test

     * harness is working.

     */

    public void initialize()

    {

        assertTrue(1 == 1);

    }

    /**

     * Test how long it will take to build records in xml_queryresult table

     */

    public void testBuildRecords() throws Exception

    {

        double start = System.currentTimeMillis()/1000;

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

        int serialNumber = dbconn.getCheckOutSerialNumber();

        String sql = "SELECT docid, nodedata from xml_path_index where path like ?";

        PreparedStatement pstmt = null;

        ResultSet result = null;

        //The key of the hash is docid, the element is the return field value (in xml_format)

        Hashtable table = new Hashtable();

        int size = returnFieldString.length;

        // Loop the return filed value in xml_path_index table and put the docid and value into hashtable

        for (int i=0; i< size; i++)

        {

        	String path = returnFieldString[i];

        	pstmt = dbconn.prepareStatement(sql);

            pstmt.setString(1,path);

            pstmt.execute();

            result = pstmt.getResultSet();

            boolean hasNext = result.next();

            // Get path value for one docid

            while (hasNext)

            {

            	String docid = result.getString(1);

            	String value = result.getString(2);

            	StringBuffer buffer  = new StringBuffer();

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

                buffer.append(path);

                buffer.append("\">");

                buffer.append(value);

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

                String xmlValue = buffer.toString();

                //If the hashtable already has key for this docid,

                //we should append obove value to the old value

            	if (table.containsKey(docid))

            	{

            		String oldValue = (String)table.get(docid);

            		String newValue = oldValue+xmlValue;

            		table.put(docid, newValue);

            	}

            	else

            	{

            		table.put(docid, xmlValue);

            	}

            	hasNext = result.next();

            }

            result.close();

            pstmt.close();

        }

        // Insert the hashtable value into xml_queryresult table

        Enumeration docids = table.keys();

        while (docids.hasMoreElements())

        {

        	String docId = (String)docids.nextElement();

        	String returnString = (String)table.get(docId);

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

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

            pstmt = dbconn.prepareStatement(query);

            pstmt.setInt(1,returnFieldId );

            pstmt.setString(2,docId);

            pstmt.setString(3, returnString);

           	pstmt.execute();     

            pstmt.close();

        }

        double end = System.currentTimeMillis()/1000;

        System.out.println("The time to build xml_queryresult is "+(end-start));

        DBConnectionPool.returnDBConnection(dbconn, serialNumber);

        assertTrue(1 == 1);

    }

}