UserDataChangeNotifier.cs 4.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159
  1. #pragma warning disable CS1591
  2. #pragma warning disable SA1600
  3. using System;
  4. using System.Collections.Generic;
  5. using System.Globalization;
  6. using System.Linq;
  7. using System.Threading;
  8. using System.Threading.Tasks;
  9. using MediaBrowser.Controller.Entities;
  10. using MediaBrowser.Controller.Library;
  11. using MediaBrowser.Controller.Plugins;
  12. using MediaBrowser.Controller.Session;
  13. using MediaBrowser.Model.Entities;
  14. using MediaBrowser.Model.Session;
  15. namespace Emby.Server.Implementations.EntryPoints
  16. {
  17. public sealed class UserDataChangeNotifier : IServerEntryPoint
  18. {
  19. private const int UpdateDuration = 500;
  20. private readonly ISessionManager _sessionManager;
  21. private readonly IUserDataManager _userDataManager;
  22. private readonly IUserManager _userManager;
  23. private readonly Dictionary<Guid, List<BaseItem>> _changedItems = new Dictionary<Guid, List<BaseItem>>();
  24. private readonly object _syncLock = new object();
  25. private Timer _updateTimer;
  26. public UserDataChangeNotifier(IUserDataManager userDataManager, ISessionManager sessionManager, IUserManager userManager)
  27. {
  28. _userDataManager = userDataManager;
  29. _sessionManager = sessionManager;
  30. _userManager = userManager;
  31. }
  32. public Task RunAsync()
  33. {
  34. _userDataManager.UserDataSaved += OnUserDataManagerUserDataSaved;
  35. return Task.CompletedTask;
  36. }
  37. void OnUserDataManagerUserDataSaved(object sender, UserDataSaveEventArgs e)
  38. {
  39. if (e.SaveReason == UserDataSaveReason.PlaybackProgress)
  40. {
  41. return;
  42. }
  43. lock (_syncLock)
  44. {
  45. if (_updateTimer == null)
  46. {
  47. _updateTimer = new Timer(
  48. UpdateTimerCallback,
  49. null,
  50. UpdateDuration,
  51. Timeout.Infinite);
  52. }
  53. else
  54. {
  55. _updateTimer.Change(UpdateDuration, Timeout.Infinite);
  56. }
  57. if (!_changedItems.TryGetValue(e.UserId, out List<BaseItem> keys))
  58. {
  59. keys = new List<BaseItem>();
  60. _changedItems[e.UserId] = keys;
  61. }
  62. keys.Add(e.Item);
  63. var baseItem = e.Item;
  64. // Go up one level for indicators
  65. if (baseItem != null)
  66. {
  67. var parent = baseItem.GetOwner() ?? baseItem.GetParent();
  68. if (parent != null)
  69. {
  70. keys.Add(parent);
  71. }
  72. }
  73. }
  74. }
  75. private void UpdateTimerCallback(object state)
  76. {
  77. lock (_syncLock)
  78. {
  79. // Remove dupes in case some were saved multiple times
  80. var changes = _changedItems.ToList();
  81. _changedItems.Clear();
  82. var task = SendNotifications(changes, CancellationToken.None);
  83. if (_updateTimer != null)
  84. {
  85. _updateTimer.Dispose();
  86. _updateTimer = null;
  87. }
  88. }
  89. }
  90. private async Task SendNotifications(List<KeyValuePair<Guid, List<BaseItem>>> changes, CancellationToken cancellationToken)
  91. {
  92. foreach (var pair in changes)
  93. {
  94. await SendNotifications(pair.Key, pair.Value, cancellationToken).ConfigureAwait(false);
  95. }
  96. }
  97. private Task SendNotifications(Guid userId, List<BaseItem> changedItems, CancellationToken cancellationToken)
  98. {
  99. return _sessionManager.SendMessageToUserSessions(new List<Guid> { userId }, "UserDataChanged", () => GetUserDataChangeInfo(userId, changedItems), cancellationToken);
  100. }
  101. private UserDataChangeInfo GetUserDataChangeInfo(Guid userId, List<BaseItem> changedItems)
  102. {
  103. var user = _userManager.GetUserById(userId);
  104. var dtoList = changedItems
  105. .GroupBy(x => x.Id)
  106. .Select(x => x.First())
  107. .Select(i =>
  108. {
  109. var dto = _userDataManager.GetUserDataDto(i, user);
  110. dto.ItemId = i.Id.ToString("N", CultureInfo.InvariantCulture);
  111. return dto;
  112. })
  113. .ToArray();
  114. var userIdString = userId.ToString("N", CultureInfo.InvariantCulture);
  115. return new UserDataChangeInfo
  116. {
  117. UserId = userIdString,
  118. UserDataList = dtoList
  119. };
  120. }
  121. public void Dispose()
  122. {
  123. if (_updateTimer != null)
  124. {
  125. _updateTimer.Dispose();
  126. _updateTimer = null;
  127. }
  128. _userDataManager.UserDataSaved -= OnUserDataManagerUserDataSaved;
  129. }
  130. }
  131. }