l2j_mobius/trunk/java/com/l2jmobius/gameserver/model/eventengine/EventScheduler.java

/*
 * This file is part of the L2J Mobius project.
 * 
 * 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 3 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, see <http://www.gnu.org/licenses/>.
 */
package com.l2jmobius.gameserver.model.eventengine;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.Timestamp;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.ScheduledFuture;
import java.util.concurrent.TimeUnit;
import java.util.logging.Level;
import java.util.logging.Logger;

import com.l2jmobius.commons.database.DatabaseFactory;
import com.l2jmobius.gameserver.ThreadPoolManager;
import com.l2jmobius.gameserver.model.StatsSet;

import it.sauronsoftware.cron4j.PastPredictor;
import it.sauronsoftware.cron4j.Predictor;

/**
 * @author UnAfraid
 */
public class EventScheduler
{
	private static final Logger LOGGER = Logger.getLogger(EventScheduler.class.getName());
	private final AbstractEventManager<?> _eventManager;
	private final String _name;
	private final String _pattern;
	private final boolean _repeat;
	private List<EventMethodNotification> _notifications;
	private ScheduledFuture<?> _task;
	
	public EventScheduler(AbstractEventManager<?> manager, StatsSet set)
	{
		_eventManager = manager;
		_name = set.getString("name", "");
		_pattern = set.getString("minute", "*") + " " + set.getString("hour", "*") + " " + set.getString("dayOfMonth", "*") + " " + set.getString("month", "*") + " " + set.getString("dayOfWeek", "*");
		_repeat = set.getBoolean("repeat", false);
	}
	
	public String getName()
	{
		return _name;
	}
	
	public long getNextSchedule()
	{
		final Predictor predictor = new Predictor(_pattern);
		return predictor.nextMatchingTime();
	}
	
	public long getNextSchedule(long fromTime)
	{
		final Predictor predictor = new Predictor(_pattern, fromTime);
		return predictor.nextMatchingTime();
	}
	
	public long getPrevSchedule()
	{
		final PastPredictor predictor = new PastPredictor(_pattern);
		return predictor.prevMatchingTime();
	}
	
	public long getPrevSchedule(long fromTime)
	{
		final PastPredictor predictor = new PastPredictor(_pattern, fromTime);
		return predictor.prevMatchingTime();
	}
	
	public boolean isRepeating()
	{
		return _repeat;
	}
	
	public void addEventNotification(EventMethodNotification notification)
	{
		if (_notifications == null)
		{
			_notifications = new ArrayList<>();
		}
		_notifications.add(notification);
	}
	
	public List<EventMethodNotification> getEventNotifications()
	{
		return _notifications;
	}
	
	public void startScheduler()
	{
		if (_notifications == null)
		{
			LOGGER.info("Scheduler without notificator manager: " + _eventManager.getClass().getSimpleName() + " pattern: " + _pattern);
			return;
		}
		
		final Predictor predictor = new Predictor(_pattern);
		final long nextSchedule = predictor.nextMatchingTime();
		final long timeSchedule = nextSchedule - System.currentTimeMillis();
		if (timeSchedule <= (30 * 1000))
		{
			LOGGER.warning("Wrong reschedule for " + _eventManager.getClass().getSimpleName() + " end up run in " + (timeSchedule / 1000) + " seconds!");
			ThreadPoolManager.getInstance().scheduleEvent(this::startScheduler, timeSchedule + 1000);
			return;
		}
		
		if (_task != null)
		{
			_task.cancel(false);
		}
		
		_task = ThreadPoolManager.getInstance().scheduleEvent(() ->
		{
			run();
			updateLastRun();
			
			if (isRepeating())
			{
				ThreadPoolManager.getInstance().scheduleEvent(this::startScheduler, 1000);
			}
		}, timeSchedule);
	}
	
	public boolean updateLastRun()
	{
		try (Connection con = DatabaseFactory.getInstance().getConnection();
			PreparedStatement ps = con.prepareStatement("INSERT INTO event_schedulers (eventName, schedulerName, lastRun) VALUES (?, ?, ?) ON DUPLICATE KEY UPDATE eventName = ?, schedulerName = ?, lastRun = ?"))
		{
			ps.setString(1, _eventManager.getName());
			ps.setString(2, _name);
			ps.setTimestamp(3, new Timestamp(System.currentTimeMillis()));
			ps.setString(4, _eventManager.getName());
			ps.setString(5, _name);
			ps.setTimestamp(6, new Timestamp(System.currentTimeMillis()));
			ps.execute();
			return true;
		}
		catch (Exception e)
		{
			LOGGER.log(Level.WARNING, "Failed to insert/update information for scheduled task manager: " + _eventManager.getClass().getSimpleName() + " scheduler: " + _name, e);
		}
		return false;
	}
	
	public void stopScheduler()
	{
		if (_task != null)
		{
			_task.cancel(false);
			_task = null;
		}
	}
	
	public long getRemainingTime(TimeUnit unit)
	{
		return (_task != null) && !_task.isDone() ? _task.getDelay(unit) : 0;
	}
	
	public void run()
	{
		for (EventMethodNotification notification : _notifications)
		{
			try
			{
				notification.execute();
			}
			catch (Exception e)
			{
				LOGGER.warning("Failed to notify to event manager: " + notification.getManager().getClass().getSimpleName() + " method: " + notification.getMethod().getName());
			}
		}
	}
}