org.mobicents.timers.timer.FaultTolerantTimerTimerTask Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of restcomm-cluster-timers Show documentation
The newest version!
/*
 * JBoss, Home of Professional Open Source
 * Copyright 2008, Red Hat Middleware LLC, and individual contributors
 * by the @authors tag. See the copyright.txt in the distribution for a
 * full listing of individual contributors.
 *
 * This is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation; either version 2.1 of
 * the License, or (at your option) any later version.
 *
 * This software 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this software; if not, write to the Free
 * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
 * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
 */
package org.mobicents.timers.timer;

import java.lang.reflect.Field;
import java.security.AccessController;
import java.security.PrivilegedAction;

import org.apache.log4j.Logger;
import org.mobicents.timers.FaultTolerantScheduler;
import org.mobicents.timers.TimerTask;

/**
 * A concrete {@link FaultTolerantScheduler} {@link TimerTask} to wrap a {@link java.util.TimerTask}.
 * 
 * @author martins
 *
 */
public class FaultTolerantTimerTimerTask extends TimerTask {

	private static final Logger logger = Logger.getLogger(FaultTolerantTimerTimerTask.class);
	
	/**
	 * 
	 */
	private final FaultTolerantScheduler scheduler;
	
	/**
	 * 
	 */
	private final FaultTolerantTimerTimerTaskData taskData;
	
	/**
	 * 
	 * @param taskData
	 */
	public FaultTolerantTimerTimerTask(String taskID, FaultTolerantTimerTimerTaskData taskData, FaultTolerantScheduler scheduler) {
		super(taskID,taskData);
		this.taskData = taskData;
		this.scheduler = scheduler;
		setPeriod(taskData.getJavaUtilTimerTask(),taskData.getPeriod());
	}
	
	/*
	 * (non-Javadoc)
	 * @see org.mobicents.timers.TimerTask#runTask()
	 */
	public void runTask() {
		if (isCanceled()) {
			scheduler.cancel(this.getTaskID());			
		}
		else {
			try{
				taskData.getJavaUtilTimerTask().run();
			} catch(Throwable e) {
				logger.error(e.getMessage(),e);
			}					
		}				
	}
	
	/**
	 * 
	 * @return
	 */
	public boolean isCanceled() {
		
		if(getTaskStatus(taskData.getJavaUtilTimerTask()) == 3 ) {
			return true;
		} 
		else {
			return false;
		}
	}
	
	private int getTaskStatus(final java.util.TimerTask timerTask) {
		if(System.getSecurityManager()!=null)
		{
			return AccessController.doPrivileged(new PrivilegedAction(){

				public Integer run() {
					return _getTaskStatus(timerTask);
				}

			});
		}else
		{
			return _getTaskStatus(timerTask);
		}
	}
	
	private Integer _getTaskStatus(java.util.TimerTask timerTask) {
		Class cc = java.util.TimerTask.class;
		try {
			Field stateField=cc.getDeclaredField("state");
			stateField.setAccessible(true);
			Integer taskStatus= (Integer) stateField.get(taskData.getJavaUtilTimerTask());
			stateField.setAccessible(false);
			return taskStatus;
		} catch (Exception e) {
			throw new RuntimeException("Fialed to get status",e);
		}
	}
	
	/**
	 * @param javaUtilTimerTask
	 * @param period
	 */
	private void setPeriod(final java.util.TimerTask javaUtilTimerTask,final long period) {
		if(System.getSecurityManager()!=null)
		{
			 AccessController.doPrivileged(new PrivilegedAction