Metacat

/**

 *  '$RCSfile$'

 *    Purpose: A Class that manages database access of scheduled task 

 *             information.

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

 *             National Center for Ecological Analysis and Synthesis

 *    Authors: Michael Daigle

 * 

 *   '$Author: daigle $'

 *     '$Date: 2009-03-23 13:56:56 -0800 (Mon, 23 Mar 2009) $'

 * '$Revision: 4854 $'

 *

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

import java.sql.PreparedStatement;

import java.sql.ResultSet;

import java.sql.SQLException;

import java.sql.Timestamp;

import java.util.Calendar;

import java.util.Hashtable;

import org.apache.log4j.Logger;

import org.quartz.Job;

import edu.ucsb.nceas.shared.AccessException;

import edu.ucsb.nceas.shared.BaseAccess;

public class ScheduledJobAccess extends BaseAccess {

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

	public ScheduledJobAccess() throws AccessException {

		super("ScheduledJobAccess");

	}

	public ScheduledJobDAO getJobByName(String jobName) throws AccessException {

		ScheduledJobDAO jobDAO = null;

		try {

			String sql = "SELECT * FROM scheduled_job WHERE name = ? "; 

			PreparedStatement pstmt = conn.prepareStatement(sql);

			pstmt.setString(1, jobName);

			logMetacat.info("SQL getJobByName - " + sql);

			logMetacat.info("SQL params: [" + jobName + "]");

			pstmt.execute();

			ResultSet resultSet = pstmt.getResultSet();

			if(resultSet.next()) {

				jobDAO = new ScheduledJobDAO();

				jobDAO.setId(resultSet.getLong("id"));

				jobDAO.setCreateTime(resultSet.getTimestamp("date_created"));

				jobDAO.setModTime(resultSet.getTimestamp("date_updated"));

				jobDAO.setStatus(resultSet.getString("status"));

				jobDAO.setName(resultSet.getString("name"));

				jobDAO.setTriggerName(resultSet.getString("trigger_name"));

				jobDAO.setTriggerName(resultSet.getString("group_name"));

				jobDAO.setClassName(resultSet.getString("class_name"));

				jobDAO.setStartTime(resultSet.getTimestamp("start_time"));

				jobDAO.setInterval(resultSet.getInt("interval"));

			}

		} catch (SQLException sqle) {

			throw new AccessException("ScheduledJobAccess.getJobByName - SQL error when getting scheduled job by name: " 

					+ jobName  + " : "  + sqle.getMessage());

		}	

		return jobDAO;		

	}

	public void createJob(String name, Class<Job> jobClass, Calendar startTime, int interval, Hashtable<String, String> jobParams) throws AccessException {

		ScheduledJobDAO jobDAO = new ScheduledJobDAO();

		jobDAO.setStatus(ScheduledJobInterface.STATE_SCHEDULED);

		jobDAO.setName(name);

		jobDAO.setClassName(jobClass.getName());

		jobDAO.setStartTime(new Timestamp(startTime.getTimeInMillis()));

		jobDAO.setInterval(interval);

		createJob(jobDAO, jobParams);

	}

	public void createJob(ScheduledJobDAO jobDAO, Hashtable<String, String> jobParams) throws AccessException {			

		try {

			String sql = 

				"INSERT INTO scheduled_job (date_created, date_updated, status, name, class_name, start_time, interval) " 

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

			PreparedStatement pstmt = conn.prepareStatement(sql);

			pstmt.setString(1, jobDAO.getStatus());

			pstmt.setString(2, jobDAO.getName());

			pstmt.setString(3, jobDAO.getClassName());

			pstmt.setTimestamp(4, jobDAO.getStartTime());

			pstmt.setInt(5, jobDAO.getInterval());

			logMetacat.info("SQL createJob - " + sql);

			logMetacat.info("SQL params:  [" + jobDAO.getStatus() + ","

					+ jobDAO.getName() + ","

					+ jobDAO.getClassName() + ",(Timestamp)"

					+ jobDAO.getStartTime().toString() + ","

					+ jobDAO.getInterval() + "]");

			pstmt.execute();

		} catch (SQLException sqle) {

			throw new AccessException("ScheduledJobAccess.createJob - SQL error when creating scheduled job " 

					+ jobDAO.getName()  + " : "  + sqle.getMessage());

		}	

		if (jobParams.size() > 0) {

			ScheduledJobParamsAccess scheduledJobParamsAccess = null;

			ScheduledJobDAO updatedJobDAO = null;

			try {

				updatedJobDAO = getJobByName(jobDAO.getName());

				scheduledJobParamsAccess = new ScheduledJobParamsAccess();

				scheduledJobParamsAccess.createJobParams(updatedJobDAO.getId(), jobParams);

			} catch (AccessException ae) {

				if (updatedJobDAO != null) {

					deleteJob(updatedJobDAO);

					scheduledJobParamsAccess.deleteJobParams(updatedJobDAO.getId());

				} else {

					logMetacat.warn("ScheduledJobAccess.createJob - Tried to delete non-existant scheduled job: " 

							+ jobDAO.getName());

				}

			}

		}

	}

	public void deleteJob(ScheduledJobDAO jobDAO) throws AccessException {	

		ScheduledJobParamsAccess scheduledJobParamsAccess = new ScheduledJobParamsAccess();

		if (jobDAO == null) {

			throw new AccessException("ScheduledJobAccess.deleteJob - job DAO cannot be null.");

		}

		scheduledJobParamsAccess.deleteJobParams(jobDAO.getId());

		try {

			String sql = "UPDATE scheduled_job SET status = ? WHERE id = ?";	

			pstmt.setString(1, ScheduledJobInterface.STATE_DELETED);

			pstmt.setLong(2, jobDAO.getId());

			logMetacat.info("ScheduledJobAccess.deleteJob - " + sql);

			logMetacat.info("SQL params:  [" + ScheduledJobInterface.STATE_DELETED + ","

					+ jobDAO.getId() + "]");

			pstmt.execute();

		} catch (SQLException sqle) {

			throw new AccessException("ScheduledJobAccess.deleteJob - SQL error when " 

					+ "deleting scheduled job " + jobDAO.getName()  + " : "  + sqle.getMessage());

		}			

	}

	public void updateJobStatus(ScheduledJobDAO jobDAO) throws AccessException {	

		if (jobDAO == null) {

			throw new AccessException("ScheduledJobAccess.updateJobStatus - job DAO cannot be null.");

		}

		try {

			String sql = "UPDATE scheduled_job SET status = ? WHERE id = ?";	

			pstmt.setString(1, jobDAO.getStatus());

			pstmt.setLong(2, jobDAO.getId());

			logMetacat.info("ScheduledJobAccess.deleteJob - " + sql);

			logMetacat.info("SQL params:  [" + jobDAO.getStatus() + ","

					+ jobDAO.getId() + "]");

			pstmt.execute();

		} catch (SQLException sqle) {

			throw new AccessException("ScheduledJobAccess.deleteJob - SQL error when " 

					+ "deleting scheduled job " + jobDAO.getName()  + " : "  + sqle.getMessage());

		}	

	}

}