2019-01-13 14:54:44 -05:00
using System ;
2016-10-29 01:40:15 -04:00
using System.Threading ;
using MediaBrowser.Model.Tasks ;
2019-01-13 14:22:24 -05:00
using Microsoft.Extensions.Logging ;
2016-10-29 01:40:15 -04:00
2021-12-15 12:25:36 -05:00
namespace Emby.Server.Implementations.ScheduledTasks.Triggers
2016-10-29 01:40:15 -04:00
{
/// <summary>
2019-10-25 06:47:20 -04:00
/// Represents a task trigger that fires everyday.
2016-10-29 01:40:15 -04:00
/// </summary>
2022-02-15 12:59:46 -05:00
public sealed class DailyTrigger : ITaskTrigger , IDisposable
2016-10-29 01:40:15 -04:00
{
2021-05-28 08:33:54 -04:00
private readonly TimeSpan _timeOfDay ;
private Timer ? _timer ;
2022-02-15 12:59:46 -05:00
private bool _disposed = false ;
2020-08-31 16:20:19 -04:00
/// <summary>
2021-05-28 08:33:54 -04:00
/// Initializes a new instance of the <see cref="DailyTrigger"/> class.
2016-10-29 01:40:15 -04:00
/// </summary>
2021-05-28 08:33:54 -04:00
/// <param name="timeofDay">The time of day to trigger the task to run.</param>
/// <param name="taskOptions">The options of this task.</param>
public DailyTrigger ( TimeSpan timeofDay , TaskOptions taskOptions )
{
_timeOfDay = timeofDay ;
TaskOptions = taskOptions ;
}
2016-10-29 01:40:15 -04:00
/// <summary>
2021-05-28 08:33:54 -04:00
/// Occurs when [triggered].
2016-10-29 01:40:15 -04:00
/// </summary>
2021-05-28 08:33:54 -04:00
public event EventHandler < EventArgs > ? Triggered ;
2016-10-29 01:40:15 -04:00
/// <summary>
2021-05-28 08:33:54 -04:00
/// Gets the options of this task.
2016-10-29 01:40:15 -04:00
/// </summary>
2021-05-28 08:33:54 -04:00
public TaskOptions TaskOptions { get ; }
2016-10-29 01:40:15 -04:00
/// <summary>
2020-06-15 18:37:52 -04:00
/// Stars waiting for the trigger action.
2016-10-29 01:40:15 -04:00
/// </summary>
/// <param name="lastResult">The last result.</param>
2020-05-29 05:28:19 -04:00
/// <param name="logger">The logger.</param>
/// <param name="taskName">The name of the task.</param>
2016-10-29 01:40:15 -04:00
/// <param name="isApplicationStartup">if set to <c>true</c> [is application startup].</param>
2021-12-15 12:25:36 -05:00
public void Start ( TaskResult ? lastResult , ILogger logger , string taskName , bool isApplicationStartup )
2016-10-29 01:40:15 -04:00
{
DisposeTimer ( ) ;
var now = DateTime . Now ;
2021-05-28 08:33:54 -04:00
var triggerDate = now . TimeOfDay > _timeOfDay ? now . Date . AddDays ( 1 ) : now . Date ;
triggerDate = triggerDate . Add ( _timeOfDay ) ;
2016-10-29 01:40:15 -04:00
var dueTime = triggerDate - now ;
2021-01-03 13:13:21 -05:00
logger . LogInformation ( "Daily trigger for {Task} set to fire at {TriggerDate:yyyy-MM-dd HH:mm:ss.fff zzz}, which is {DueTime:c} from now." , taskName , triggerDate , dueTime ) ;
2016-10-29 01:40:15 -04:00
2021-12-15 12:25:36 -05:00
_timer = new Timer ( _ = > OnTriggered ( ) , null , dueTime , TimeSpan . FromMilliseconds ( - 1 ) ) ;
2016-10-29 01:40:15 -04:00
}
/// <summary>
2020-06-15 18:37:52 -04:00
/// Stops waiting for the trigger action.
2016-10-29 01:40:15 -04:00
/// </summary>
public void Stop ( )
{
DisposeTimer ( ) ;
}
/// <summary>
/// Disposes the timer.
/// </summary>
private void DisposeTimer ( )
{
2021-05-28 08:33:54 -04:00
_timer ? . Dispose ( ) ;
2022-02-15 12:59:46 -05:00
_timer = null ;
2016-10-29 01:40:15 -04:00
}
/// <summary>
/// Called when [triggered].
/// </summary>
private void OnTriggered ( )
{
2020-05-29 05:28:19 -04:00
Triggered ? . Invoke ( this , EventArgs . Empty ) ;
2016-10-29 01:40:15 -04:00
}
2022-02-15 12:59:46 -05:00
/// <inheritdoc />
public void Dispose ( )
{
if ( _disposed )
{
return ;
}
DisposeTimer ( ) ;
_disposed = true ;
}
2016-10-29 01:40:15 -04:00
}
}