Metacat

package edu.ucsb.nceas.metacat.admin.upgrade;

/**

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

 *

 *   '$Author: leinfelder $'

 *     '$Date: 2011-03-16 22:56:31 -0700 (Wed, 16 Mar 2011) $'

 * '$Revision: 6012 $'

 *

 * 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.sql.Connection;

import java.sql.Driver;

import java.sql.ResultSet;

import java.sql.PreparedStatement;

import java.sql.Statement;

import java.sql.DriverManager;

import edu.ucsb.nceas.metacat.admin.AdminException;

import edu.ucsb.nceas.metacat.properties.PropertyService;

public class Upgrade1_5_0 implements UpgradeUtilityInterface {

    private String driver = null;

    private String database = null;

    private String url = null;

    private String user = null;

    private String password = null;

    public boolean upgrade() throws AdminException {

    	try {

	    	// get the properties

			driver = PropertyService.getProperty("database.driver");

			database = PropertyService.getProperty("database.type");

		    url = PropertyService.getProperty("database.connectionURI");

		    user = PropertyService.getProperty("database.user");

		    password = PropertyService.getProperty("database.password");

		    // only do oracle

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

		    	return true;

		    }

	        // Create a JDBC connection to the database    	

	    	Connection sqlca = null;

	        Driver d = (Driver) Class.forName(driver).newInstance();

	        DriverManager.registerDriver(d);

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

	        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(

	            "Please be patient as the next upgrade step can be extremely time consuming.");

	        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++;

	                    if (count%5 == 0) {

	                        System.out.println(count + "...");

	                    }

	                }

	            } catch (NumberFormatException nfe) {

	            } catch (Exception e) {

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

	            }

	        }

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

	        rset.close();

	        sqlStatement.close();

	        sqlca.close();

    	} catch (Exception e) {

            throw new AdminException("Error upgrading system to 1.5: " + e.getMessage());

        }

    	return true;

    }

    public static void main(String [] ags){

        try {

            Upgrade1_5_0 upgrader = new Upgrade1_5_0();

            upgrader.upgrade();

        } catch (Exception ex) {

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

        }

    }

}