Metacat

/**

 *  '$RCSfile$'

 *    Purpose: A Class for upgrading the database to version 1.5

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

 *             National Center for Ecological Analysis and Synthesis

 *    Authors: Saurabh Garg

 *    Release: @release@

 *

 *   '$Author: sgarg $'

 *     '$Date: 2005-01-18 13:26:01 -0800 (Tue, 18 Jan 2005) $'

 * '$Revision: 2379 $'

 *

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

import java.sql.Connection;

import java.sql.ResultSet;

import java.sql.PreparedStatement;

import java.sql.Statement;

import java.sql.SQLException;

import java.sql.DriverManager;

public class upgrade_db_to_1_5{

    static String database = "@database@";

    static String url = "@jdbc-connect@";

    static String user = "@user@";

    static String password = "@password@";

    public static void upgrade_xml_nodes(Connection sqlca) throws SQLException{

        Statement sqlStatement = null;

        PreparedStatement pstmt = null;

        ResultSet rset = null;

        // Delete old nodedatanumerical column from xml_nodes if one exsists

        try {

            System.out.println(

                "Deleting old nodedatanumerical column from xml_nodes...");

            pstmt = sqlca.prepareStatement(

                "ALTER TABLE xml_nodes DROP COLUMN nodedatanumerical");

            pstmt.execute();

            pstmt.close();

            System.out.println("Done.");

        } catch (Exception e) {

            System.out.println(" column not found.");

        }

        // Create nodedatanumerical column in xml_nodes

        System.out.println(

            "Creating new nodedatanumerical column in xml_nodes...");

        if (database.equals("oracle")) {

            pstmt = sqlca.prepareStatement(

                "ALTER TABLE xml_nodes ADD nodedatanumerical NUMBER");

        }

        else {

            pstmt = sqlca.prepareStatement(

                "ALTER TABLE xml_nodes ADD nodedatanumerical FLOAT8");

        }

        pstmt.execute();

        pstmt.close();

        System.out.println("Done.");

        // Copy numerical values from nodedata to

        // nodedatanumerical column in xml_nodes

        System.out.println(

            "Copy numerical values from nodedata to nodedatanumerical "

            + "in xml_nodes...");

        sqlStatement = sqlca.createStatement();

        rset = sqlStatement.executeQuery(

            "SELECT DISTINCT NODEID, NODEDATA "

            + "FROM xml_nodes WHERE "

            + "nodedata IS NOT NULL AND "

            + "UPPER(nodedata) = LOWER(nodedata)");

        int count = 0;

        while (rset.next()) {

            String nodeid = rset.getString(1);

            String nodedata = rset.getString(2);

            try {

                if (!nodedata.trim().equals("")) {

                    double s = Double.parseDouble(nodedata);

                    pstmt = sqlca.prepareStatement(

                        "update xml_nodes set nodedatanumerical = " + s +

                        " where nodeid=" + nodeid);

                    pstmt.execute();

                    pstmt.close();

                    count++;

                }

            } catch (NumberFormatException nfe) {

            } catch (Exception e) {

                System.out.println("Exception:" + e.getMessage());

            }

        }

        System.out.println("Done. " + count + " values copied.");

        rset.close();

        sqlStatement.close();

    }

    public static void upgrade_xml_index(Connection sqlca) throws SQLException{

        Statement sqlStatement, sqlStatement1 = null;

        PreparedStatement pstmt = null;

        ResultSet rset, rset1 = null;

        // Delete old nodedata column from xml_index if one exsists

        try {

            System.out.println(

                "Deleting old nodedata column from xml_index...");

            pstmt = sqlca.prepareStatement(

                "ALTER TABLE xml_index DROP COLUMN nodedata");

            pstmt.execute();

            pstmt.close();

            System.out.println("Done.");

        } catch (Exception e) {

            System.out.println(" column not found.");

        }

        // Create nodedata column in xml_index

        System.out.println(

            "Creating new nodedata column in xml_index...");

        if (database.equals("oracle")) {

            pstmt = sqlca.prepareStatement(

                "ALTER TABLE xml_index ADD nodedata VARCHAR2(4000)");

        } else {

            pstmt = sqlca.prepareStatement(

                "ALTER TABLE xml_index ADD nodedata VARCHAR(4000)");

        }

        pstmt.execute();

        pstmt.close();

        System.out.println("Done.");

        // Copy numerical values from nodedata to

        // nodedatanumerical column in xml_index

        System.out.println(

            "Copy nodedata from xml_nodes to nodedata in xml_index...");

        sqlStatement = sqlca.createStatement();

        rset = sqlStatement.executeQuery(

            "select docid, nodeid, path from xml_index");

        int j = 0;

        while (rset.next()) {

            String docid = rset.getString(1);

            String nodeid = rset.getString(2);

            String path = rset.getString(3);

            sqlStatement1 = sqlca.createStatement();

            rset1 = sqlStatement1.executeQuery(

                "select nodedata from xml_nodes where parentnodeid = " + nodeid

		+ " and docid = '" + docid

                + "' and (nodetype = 'TEXT' or nodetype ='ATTRIBUTE')");

            while (rset1.next()) {

                String nodedata = rset1.getString(1);

                if (nodedata != null && !nodedata.trim().equals("")) {

                    j++;

                    if (j % 100 == 0) {

                        System.out.println("Number of records modified:" + j);

                    }

                    pstmt = sqlca.prepareStatement(

                        "update xml_index set nodedata = ? where nodeid = " +

                        nodeid + " and path = ? and docid = ? and nodedata is null");

                    pstmt.setString(1, nodedata);

                    pstmt.setString(2, path);

                    pstmt.setString(3, docid);

                    pstmt.execute();

                    pstmt.close();

                    continue;

                }

            }

            rset1.close();

            sqlStatement1.close();

        }

        rset.close();

        sqlStatement.close();

        // Create index on xml_index.docid

        System.out.println("Creating index on docid column in xml_index...");

        pstmt = sqlca.prepareStatement(

            "CREATE INDEX xml_index_idx2 ON xml_index (docid)");

        pstmt.execute();

        pstmt.close();

    }

    public static void main(String [] ags){

        try {

            Connection sqlca = null;

            // Create a JDBC connection to the database

            DriverManager.registerDriver(new @dbDriver@());

            sqlca = DriverManager.getConnection(url, user, password);

            // Upgrade the xml_nodes table

            upgrade_xml_nodes(sqlca);

            // Upgrade the xml_index table

            upgrade_xml_index(sqlca);

            // Close the connection...

            sqlca.close();

        } catch (SQLException ex) {

            System.out.println("SQLException:" + ex.getMessage());

        }

    }

}