TimerSchedulerMemento.java

/**
 * The contents of this file are subject to the OpenMRS Public License
 * Version 1.0 (the "License"); you may not use this file except in
 * compliance with the License. You may obtain a copy of the License at
 * http://license.openmrs.org
 *
 * Software distributed under the License is distributed on an "AS IS"
 * basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See the
 * License for the specific language governing rights and limitations
 * under the License.
 *
 * Copyright (C) OpenMRS, LLC.  All Rights Reserved.
 */
package org.openmrs.scheduler.timer;

import java.util.HashSet;
import java.util.Set;

import org.openmrs.scheduler.TaskDefinition;
import org.openmrs.util.OpenmrsMemento;

public class TimerSchedulerMemento extends OpenmrsMemento {
	
	private Set<TaskDefinition> startedTasks = new HashSet<TaskDefinition>();
	
	private static Set<TaskDefinition> errorTasks = new HashSet<TaskDefinition>();
	
	public TimerSchedulerMemento(Set<TaskDefinition> tasks) {
		this.startedTasks = tasks;
	}
	
	@Override
	public Object getState() {
		return startedTasks;
	}
	
	@Override
	@SuppressWarnings("unchecked")
	public void setState(Object state) {
		this.startedTasks = (Set<TaskDefinition>) state;
	}
	
	public Boolean addErrorTask(TaskDefinition task) {
		return errorTasks.add(task);
	}
	
	public Boolean removeErrorTask(TaskDefinition task) {
		return errorTasks.remove(task);
	}
	
	public static Set<TaskDefinition> getErrorTasks() {
		return errorTasks;
	}
	
	public void saveErrorTasks() {
		this.startedTasks.addAll(errorTasks);
	}
	
}