jellyfin/Emby.Server.Implementations/IO/FileRefresher.cs

223 lines
6.3 KiB
C#
Raw Normal View History

2019-11-01 17:38:54 +00:00
#pragma warning disable CS1591
2019-12-10 23:13:57 +00:00
#pragma warning disable SA1600
2019-11-01 17:38:54 +00:00
using System;
2016-05-26 04:11:27 +00:00
using System.Collections.Generic;
using System.IO;
using System.Linq;
2019-02-05 08:49:46 +00:00
using System.Threading;
2016-05-26 04:11:27 +00:00
using MediaBrowser.Controller.Configuration;
using MediaBrowser.Controller.Entities;
using MediaBrowser.Controller.Library;
using Microsoft.Extensions.Logging;
2016-05-26 04:11:27 +00:00
2016-11-04 18:56:47 +00:00
namespace Emby.Server.Implementations.IO
2016-05-26 04:11:27 +00:00
{
public class FileRefresher : IDisposable
{
private ILogger Logger { get; set; }
private ILibraryManager LibraryManager { get; set; }
private IServerConfigurationManager ConfigurationManager { get; set; }
private readonly List<string> _affectedPaths = new List<string>();
2019-02-05 08:49:46 +00:00
private Timer _timer;
2016-05-26 04:11:27 +00:00
private readonly object _timerLock = new object();
public string Path { get; private set; }
public event EventHandler<EventArgs> Completed;
2016-05-26 04:11:27 +00:00
2019-02-06 19:38:42 +00:00
public FileRefresher(string path, IServerConfigurationManager configurationManager, ILibraryManager libraryManager, ILogger logger)
2016-05-26 04:11:27 +00:00
{
logger.LogDebug("New file refresher created for {0}", path);
Path = path;
2016-05-26 04:11:27 +00:00
ConfigurationManager = configurationManager;
LibraryManager = libraryManager;
Logger = logger;
2016-07-03 02:47:39 +00:00
AddPath(path);
2016-05-26 04:11:27 +00:00
}
private void AddAffectedPath(string path)
{
2018-09-12 17:26:21 +00:00
if (string.IsNullOrEmpty(path))
{
throw new ArgumentNullException(nameof(path));
}
if (!_affectedPaths.Contains(path, StringComparer.Ordinal))
{
_affectedPaths.Add(path);
}
}
public void AddPath(string path)
{
2018-09-12 17:26:21 +00:00
if (string.IsNullOrEmpty(path))
{
throw new ArgumentNullException(nameof(path));
}
lock (_timerLock)
{
AddAffectedPath(path);
}
RestartTimer();
}
public void RestartTimer()
2016-05-26 04:11:27 +00:00
{
2016-08-11 03:56:01 +00:00
if (_disposed)
{
return;
}
2016-05-26 04:11:27 +00:00
lock (_timerLock)
{
2016-09-17 06:08:38 +00:00
if (_disposed)
{
return;
}
2016-05-26 04:11:27 +00:00
if (_timer == null)
{
2019-02-05 08:49:46 +00:00
_timer = new Timer(OnTimerCallback, null, TimeSpan.FromSeconds(ConfigurationManager.Configuration.LibraryMonitorDelay), TimeSpan.FromMilliseconds(-1));
2016-05-26 04:11:27 +00:00
}
else
{
_timer.Change(TimeSpan.FromSeconds(ConfigurationManager.Configuration.LibraryMonitorDelay), TimeSpan.FromMilliseconds(-1));
}
}
}
public void ResetPath(string path, string affectedFile)
{
lock (_timerLock)
{
Logger.LogDebug("Resetting file refresher from {0} to {1}", Path, path);
Path = path;
AddAffectedPath(path);
2018-09-12 17:26:21 +00:00
if (!string.IsNullOrEmpty(affectedFile))
{
AddAffectedPath(affectedFile);
}
}
RestartTimer();
}
2017-08-16 17:30:16 +00:00
private void OnTimerCallback(object state)
2016-05-26 04:11:27 +00:00
{
2016-06-11 15:56:15 +00:00
List<string> paths;
lock (_timerLock)
{
paths = _affectedPaths.ToList();
}
Logger.LogDebug("Timer stopped.");
2016-05-26 04:11:27 +00:00
DisposeTimer();
Completed?.Invoke(this, EventArgs.Empty);
2016-05-26 04:11:27 +00:00
try
{
2017-08-16 17:30:16 +00:00
ProcessPathChanges(paths.ToList());
2016-05-26 04:11:27 +00:00
}
catch (Exception ex)
{
2018-12-20 12:11:26 +00:00
Logger.LogError(ex, "Error processing directory changes");
2016-05-26 04:11:27 +00:00
}
}
2017-08-16 17:30:16 +00:00
private void ProcessPathChanges(List<string> paths)
2016-05-26 04:11:27 +00:00
{
var itemsToRefresh = paths
2016-09-20 19:43:27 +00:00
.Distinct(StringComparer.OrdinalIgnoreCase)
2016-05-26 04:11:27 +00:00
.Select(GetAffectedBaseItem)
.Where(item => item != null)
2019-02-02 11:27:06 +00:00
.GroupBy(x => x.Id)
.Select(x => x.First());
2016-05-26 04:11:27 +00:00
foreach (var item in itemsToRefresh)
{
2017-10-03 18:39:37 +00:00
if (item is AggregateFolder)
{
continue;
}
Logger.LogInformation("{name} ({path}) will be refreshed.", item.Name, item.Path);
2016-05-26 04:11:27 +00:00
try
{
2017-05-27 07:19:09 +00:00
item.ChangedExternally();
2016-05-26 04:11:27 +00:00
}
catch (IOException ex)
{
2019-01-07 23:27:46 +00:00
// For now swallow and log.
2016-05-26 04:11:27 +00:00
// Research item: If an IOException occurs, the item may be in a disconnected state (media unavailable)
// Should we remove it from it's parent?
2018-12-20 12:11:26 +00:00
Logger.LogError(ex, "Error refreshing {name}", item.Name);
2016-05-26 04:11:27 +00:00
}
catch (Exception ex)
{
2018-12-20 12:11:26 +00:00
Logger.LogError(ex, "Error refreshing {name}", item.Name);
2016-05-26 04:11:27 +00:00
}
}
}
/// <summary>
/// Gets the affected base item.
/// </summary>
/// <param name="path">The path.</param>
/// <returns>BaseItem.</returns>
private BaseItem GetAffectedBaseItem(string path)
{
BaseItem item = null;
while (item == null && !string.IsNullOrEmpty(path))
{
item = LibraryManager.FindByPath(path, null);
path = System.IO.Path.GetDirectoryName(path);
2016-05-26 04:11:27 +00:00
}
if (item != null)
{
// If the item has been deleted find the first valid parent that still exists
while (!Directory.Exists(item.Path) && !File.Exists(item.Path))
2016-05-26 04:11:27 +00:00
{
2018-09-12 17:26:21 +00:00
item = item.GetOwner() ?? item.GetParent();
2016-05-26 04:11:27 +00:00
if (item == null)
{
break;
}
}
}
return item;
}
private void DisposeTimer()
2016-05-26 04:11:27 +00:00
{
lock (_timerLock)
{
if (_timer != null)
{
_timer.Dispose();
2016-09-17 06:08:38 +00:00
_timer = null;
2016-05-26 04:11:27 +00:00
}
}
}
2016-08-11 03:56:01 +00:00
private bool _disposed;
2016-05-26 04:11:27 +00:00
public void Dispose()
{
2016-08-11 03:56:01 +00:00
_disposed = true;
2016-05-26 04:11:27 +00:00
DisposeTimer();
}
}
}