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: 2012-01-05 14:49:29 -0800 (Thu, 05 Jan 2012) $'

 * '$Revision: 6856 $'

 *

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

import java.io.FileFilter;

import java.sql.Connection;

import java.sql.Driver;

import java.sql.DriverManager;

import java.sql.PreparedStatement;

import java.sql.SQLException;

import java.util.ArrayList;

import java.util.List;

import org.apache.commons.io.filefilter.EmptyFileFilter;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

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

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

import edu.ucsb.nceas.metacat.util.DocumentUtil;

import edu.ucsb.nceas.utilities.PropertyNotFoundException;

import edu.ucsb.nceas.utilities.SortedProperties;

public class UpgradeEmptyReplicatedDataFile implements UpgradeUtilityInterface {

	protected static Log log = LogFactory.getLog(UpgradeEmptyReplicatedDataFile.class);

    private String driver = null;

    private String url = null;

    private String user = null;

    private String password = null;

    public boolean upgrade() throws AdminException {

    	boolean success = true;    		

        Connection sqlca = null;

        PreparedStatement pstmt = null;

        // look up the empty files from the file system

        File dataDir = null;

		try {

			dataDir = new File(PropertyService.getProperty("application.datafilepath"));

		} catch (PropertyNotFoundException e) {

			e.printStackTrace();

			return false;

		}

		File[] emptyFiles = dataDir.listFiles((FileFilter) EmptyFileFilter.EMPTY);

		if (emptyFiles.length == 0) {

			log.info("No empty data files found");

			return true;

		}

		// track the ones we end up deleting from the DB

		List<File> processedFiles = new ArrayList<File>();

        try {

        	// get the properties

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

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

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

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

	        // Create a JDBC connection to the database    

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

	        DriverManager.registerDriver(d);

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

	        sqlca.setAutoCommit(false);        

			for (File emptyFile: emptyFiles) {

				int count = 0;

				// delete each empty file from the tables 

				String emptyDocid = emptyFile.getName();

				String docid = DocumentUtil.getDocIdFromString(emptyDocid);

				String rev = DocumentUtil.getRevisionStringFromString(emptyDocid);

		        // xml_documents

				pstmt = sqlca.prepareStatement("DELETE FROM xml_documents WHERE docid = ? and server_location != '1'");

				pstmt.setString(1, docid);

				count = pstmt.executeUpdate();

				// xml_revisions

				pstmt = sqlca.prepareStatement("DELETE FROM xml_revisions WHERE docid = ? and server_location != '1'");

				pstmt.setString(1, docid);

				count = count + pstmt.executeUpdate();

				// if we did remove it from the the DB

				if (count > 0) {

					processedFiles.add(emptyFile);

				}

			}

			// all or nothing

			sqlca.commit();

        } catch (Exception e) {

        	try {

        		// rollback if there was even a single error

				sqlca.rollback();

			} catch (SQLException e1) {

				// TODO Auto-generated catch block

				e1.printStackTrace();

			}

        	success = false;

		} finally {

			// clean up

			if (sqlca != null) {

				try {

					sqlca.close();

				} catch (SQLException e) {

					// TODO Auto-generated catch block

					e.printStackTrace();

				}

			}

		}

		// now delete the actual files that were removed from the db

		if (success) {

			for (File emptyFile: processedFiles) {

				log.info("Deleting empty replicated data file from filesystem: " + emptyFile.getAbsolutePath());

				emptyFile.delete();

			}

		}

    	return success;

    }

    public static void main(String [] ags){

        try {

        	// set up the properties based on the test/deployed configuration of the workspace

        	SortedProperties testProperties = 

				new SortedProperties("test/test.properties");

			testProperties.load();

			String metacatContextDir = testProperties.getProperty("metacat.contextDir");

			PropertyService.getInstance(metacatContextDir + "/WEB-INF");

			// now run it

            UpgradeEmptyReplicatedDataFile upgrader = new UpgradeEmptyReplicatedDataFile();

	        upgrader.upgrade();

        } catch (Exception ex) {

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

            ex.printStackTrace();

        }

    }

}