PersistentTaskRunnable.java

/**
 * <a href="http://www.openolat.org">
 * OpenOLAT - Online Learning and Training</a><br>
 * <p>
 * Licensed under the Apache License, Version 2.0 (the "License"); <br>
 * you may not use this file except in compliance with the License.<br>
 * You may obtain a copy of the License at the
 * <a href="http://www.apache.org/licenses/LICENSE-2.0">Apache homepage</a>
 * <p>
 * Unless required by applicable law or agreed to in writing,<br>
 * software distributed under the License is distributed on an "AS IS" BASIS, <br>
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. <br>
 * See the License for the specific language governing permissions and <br>
 * limitations under the License.
 * <p>
 * Initial code contributed and copyrighted by<br>
 * frentix GmbH, http://www.frentix.com
 * <p>
 */
package org.olat.core.commons.services.taskexecutor.model;

import org.olat.core.CoreSpringFactory;
import org.olat.core.commons.persistence.DBFactory;
import org.olat.core.commons.services.taskexecutor.TaskAwareRunnable;
import org.olat.core.commons.services.taskexecutor.manager.PersistentTaskDAO;
import org.olat.core.logging.OLog;
import org.olat.core.logging.Tracing;

/**
 * 
 * @author srosse, stephane.rosse@frentix.com, http://www.frentix.com
 */
public class PersistentTaskRunnable implements Runnable {
	
	private static final OLog log = Tracing.createLoggerFor(PersistentTaskRunnable.class);
	private final Long taskKey;
	
	public PersistentTaskRunnable(Long taskKey) {
		this.taskKey = taskKey;
	}

	@Override
	public void run() {
		PersistentTask task = null;
		PersistentTaskDAO taskDao = CoreSpringFactory.getImpl(PersistentTaskDAO.class);
		try {
			task = taskDao.pickTaskForRun(taskKey);
			if(task != null) {
				Runnable runnable = taskDao.deserializeTask(task);
				if(runnable instanceof TaskAwareRunnable) {
					((TaskAwareRunnable)runnable).setTask(task);
				}
				runnable.run();
				taskDao.taskDone(task);
			}
			DBFactory.getInstance().commitAndCloseSession();
		} catch (Throwable e) {
			DBFactory.getInstance().rollbackAndCloseSession();
			markAsFailed(task);
			log.error("Error while running task in a separate thread: " + (task == null ? "NULL" : task.getKey()), e);
		}
	}
	
	private void markAsFailed(PersistentTask task) {
		if(task == null) return;
		try {
			PersistentTaskDAO taskDao = CoreSpringFactory.getImpl(PersistentTaskDAO.class);
			taskDao.taskFailed(task);
			DBFactory.getInstance().commitAndCloseSession();
		} catch (Exception e1) {
			DBFactory.getInstance().rollbackAndCloseSession();
		}
	}
}