DailyTrigger.cs 2.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687
  1. using System;
  2. using System.Globalization;
  3. using System.Threading;
  4. using MediaBrowser.Model.Tasks;
  5. using Microsoft.Extensions.Logging;
  6. namespace Emby.Server.Implementations.ScheduledTasks
  7. {
  8. /// <summary>
  9. /// Represents a task trigger that fires everyday
  10. /// </summary>
  11. public class DailyTrigger : ITaskTrigger
  12. {
  13. /// <summary>
  14. /// Get the time of day to trigger the task to run
  15. /// </summary>
  16. /// <value>The time of day.</value>
  17. public TimeSpan TimeOfDay { get; set; }
  18. /// <summary>
  19. /// Gets or sets the options of this task.
  20. /// </summary>
  21. public TaskOptions TaskOptions { get; set; }
  22. /// <summary>
  23. /// Gets or sets the timer.
  24. /// </summary>
  25. /// <value>The timer.</value>
  26. private Timer Timer { get; set; }
  27. /// <summary>
  28. /// Stars waiting for the trigger action
  29. /// </summary>
  30. /// <param name="lastResult">The last result.</param>
  31. /// <param name="isApplicationStartup">if set to <c>true</c> [is application startup].</param>
  32. public void Start(TaskResult lastResult, ILogger logger, string taskName, bool isApplicationStartup)
  33. {
  34. DisposeTimer();
  35. var now = DateTime.Now;
  36. var triggerDate = now.TimeOfDay > TimeOfDay ? now.Date.AddDays(1) : now.Date;
  37. triggerDate = triggerDate.Add(TimeOfDay);
  38. var dueTime = triggerDate - now;
  39. logger.LogInformation("Daily trigger for {0} set to fire at {1}, which is {2} minutes from now.", taskName, triggerDate.ToString(), dueTime.TotalMinutes.ToString(CultureInfo.InvariantCulture));
  40. Timer = new Timer(state => OnTriggered(), null, dueTime, TimeSpan.FromMilliseconds(-1));
  41. }
  42. /// <summary>
  43. /// Stops waiting for the trigger action
  44. /// </summary>
  45. public void Stop()
  46. {
  47. DisposeTimer();
  48. }
  49. /// <summary>
  50. /// Disposes the timer.
  51. /// </summary>
  52. private void DisposeTimer()
  53. {
  54. if (Timer != null)
  55. {
  56. Timer.Dispose();
  57. }
  58. }
  59. /// <summary>
  60. /// Occurs when [triggered].
  61. /// </summary>
  62. public event EventHandler<EventArgs> Triggered;
  63. /// <summary>
  64. /// Called when [triggered].
  65. /// </summary>
  66. private void OnTriggered()
  67. {
  68. if (Triggered != null)
  69. {
  70. Triggered(this, EventArgs.Empty);
  71. }
  72. }
  73. }
  74. }