WeeklyTrigger.cs 3.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111
  1. using System;
  2. using System.Threading;
  3. using MediaBrowser.Model.Tasks;
  4. using Microsoft.Extensions.Logging;
  5. namespace Emby.Server.Implementations.ScheduledTasks
  6. {
  7. /// <summary>
  8. /// Represents a task trigger that fires on a weekly basis.
  9. /// </summary>
  10. public class WeeklyTrigger : ITaskTrigger
  11. {
  12. /// <summary>
  13. /// Occurs when [triggered].
  14. /// </summary>
  15. public event EventHandler<EventArgs> Triggered;
  16. /// <summary>
  17. /// Gets or sets the time of day to trigger the task to run.
  18. /// </summary>
  19. /// <value>The time of day.</value>
  20. public TimeSpan TimeOfDay { get; set; }
  21. /// <summary>
  22. /// Gets or sets the day of week.
  23. /// </summary>
  24. /// <value>The day of week.</value>
  25. public DayOfWeek DayOfWeek { get; set; }
  26. /// <summary>
  27. /// Gets or sets the options of this task.
  28. /// </summary>
  29. public TaskOptions TaskOptions { get; set; }
  30. /// <summary>
  31. /// Gets or sets the timer.
  32. /// </summary>
  33. /// <value>The timer.</value>
  34. private Timer Timer { get; set; }
  35. /// <summary>
  36. /// Stars waiting for the trigger action.
  37. /// </summary>
  38. /// <param name="lastResult">The last result.</param>
  39. /// <param name="logger">The logger.</param>
  40. /// <param name="taskName">The name of the task.</param>
  41. /// <param name="isApplicationStartup">if set to <c>true</c> [is application startup].</param>
  42. public void Start(TaskResult lastResult, ILogger logger, string taskName, bool isApplicationStartup)
  43. {
  44. DisposeTimer();
  45. var triggerDate = GetNextTriggerDateTime();
  46. Timer = new Timer(state => OnTriggered(), null, triggerDate - DateTime.Now, TimeSpan.FromMilliseconds(-1));
  47. }
  48. /// <summary>
  49. /// Gets the next trigger date time.
  50. /// </summary>
  51. /// <returns>DateTime.</returns>
  52. private DateTime GetNextTriggerDateTime()
  53. {
  54. var now = DateTime.Now;
  55. // If it's on the same day
  56. if (now.DayOfWeek == DayOfWeek)
  57. {
  58. // It's either later today, or a week from now
  59. return now.TimeOfDay < TimeOfDay ? now.Date.Add(TimeOfDay) : now.Date.AddDays(7).Add(TimeOfDay);
  60. }
  61. var triggerDate = now.Date;
  62. // Walk the date forward until we get to the trigger day
  63. while (triggerDate.DayOfWeek != DayOfWeek)
  64. {
  65. triggerDate = triggerDate.AddDays(1);
  66. }
  67. // Return the trigger date plus the time offset
  68. return triggerDate.Add(TimeOfDay);
  69. }
  70. /// <summary>
  71. /// Stops waiting for the trigger action.
  72. /// </summary>
  73. public void Stop()
  74. {
  75. DisposeTimer();
  76. }
  77. /// <summary>
  78. /// Disposes the timer.
  79. /// </summary>
  80. private void DisposeTimer()
  81. {
  82. if (Timer != null)
  83. {
  84. Timer.Dispose();
  85. }
  86. }
  87. /// <summary>
  88. /// Called when [triggered].
  89. /// </summary>
  90. private void OnTriggered()
  91. {
  92. Triggered?.Invoke(this, EventArgs.Empty);
  93. }
  94. }
  95. }