2013-09-10 20:56:00 +02:00
|
|
|
|
using MediaBrowser.Controller.Entities;
|
|
|
|
|
using MediaBrowser.Controller.Entities.Audio;
|
|
|
|
|
using MediaBrowser.Controller.Library;
|
|
|
|
|
using MediaBrowser.Model.Logging;
|
|
|
|
|
using System;
|
|
|
|
|
using System.Collections.Generic;
|
|
|
|
|
using System.Linq;
|
|
|
|
|
using System.Threading;
|
|
|
|
|
using System.Threading.Tasks;
|
|
|
|
|
|
|
|
|
|
namespace MediaBrowser.Server.Implementations.Library.Validators
|
|
|
|
|
{
|
|
|
|
|
class GenresValidator
|
|
|
|
|
{
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// The _library manager
|
|
|
|
|
/// </summary>
|
2013-09-17 04:08:18 +02:00
|
|
|
|
private readonly ILibraryManager _libraryManager;
|
2013-09-10 20:56:00 +02:00
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// The _user manager
|
|
|
|
|
/// </summary>
|
|
|
|
|
private readonly IUserManager _userManager;
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// The _logger
|
|
|
|
|
/// </summary>
|
|
|
|
|
private readonly ILogger _logger;
|
|
|
|
|
|
2013-09-17 04:08:18 +02:00
|
|
|
|
public GenresValidator(ILibraryManager libraryManager, IUserManager userManager, ILogger logger)
|
2013-09-10 20:56:00 +02:00
|
|
|
|
{
|
|
|
|
|
_libraryManager = libraryManager;
|
|
|
|
|
_userManager = userManager;
|
|
|
|
|
_logger = logger;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// Runs the specified progress.
|
|
|
|
|
/// </summary>
|
|
|
|
|
/// <param name="progress">The progress.</param>
|
|
|
|
|
/// <param name="cancellationToken">The cancellation token.</param>
|
|
|
|
|
/// <returns>Task.</returns>
|
|
|
|
|
public async Task Run(IProgress<double> progress, CancellationToken cancellationToken)
|
|
|
|
|
{
|
|
|
|
|
var userLibraries = _userManager.Users
|
2013-09-27 14:24:28 +02:00
|
|
|
|
.Select(i => new Tuple<Guid, IList<BaseItem>>(i.Id, i.RootFolder.GetRecursiveChildren(i, m => !(m is IHasMusicGenres) && !(m is Game))))
|
2013-09-10 20:56:00 +02:00
|
|
|
|
.ToList();
|
|
|
|
|
|
2013-09-11 19:54:59 +02:00
|
|
|
|
var masterDictionary = new Dictionary<string, Dictionary<Guid, Dictionary<CountType, int>>>(StringComparer.OrdinalIgnoreCase);
|
2013-09-10 20:56:00 +02:00
|
|
|
|
|
|
|
|
|
// Populate counts of items
|
2013-09-11 19:54:59 +02:00
|
|
|
|
//SetItemCounts(null, allLibraryItems, masterDictionary);
|
2013-09-10 20:56:00 +02:00
|
|
|
|
|
|
|
|
|
progress.Report(2);
|
|
|
|
|
|
|
|
|
|
var numComplete = 0;
|
|
|
|
|
|
|
|
|
|
foreach (var lib in userLibraries)
|
|
|
|
|
{
|
|
|
|
|
SetItemCounts(lib.Item1, lib.Item2, masterDictionary);
|
|
|
|
|
|
|
|
|
|
numComplete++;
|
|
|
|
|
double percent = numComplete;
|
|
|
|
|
percent /= userLibraries.Count;
|
|
|
|
|
percent *= 8;
|
|
|
|
|
|
|
|
|
|
progress.Report(percent);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
progress.Report(10);
|
|
|
|
|
|
2013-09-11 19:54:59 +02:00
|
|
|
|
var count = masterDictionary.Count;
|
2013-09-10 20:56:00 +02:00
|
|
|
|
numComplete = 0;
|
|
|
|
|
|
2013-09-11 19:54:59 +02:00
|
|
|
|
foreach (var name in masterDictionary.Keys)
|
2013-09-10 20:56:00 +02:00
|
|
|
|
{
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
await UpdateItemByNameCounts(name, cancellationToken, masterDictionary[name]).ConfigureAwait(false);
|
|
|
|
|
}
|
2013-09-24 23:06:21 +02:00
|
|
|
|
catch (OperationCanceledException)
|
|
|
|
|
{
|
|
|
|
|
// Don't clutter the log
|
2013-10-18 21:08:48 +02:00
|
|
|
|
break;
|
2013-09-24 23:06:21 +02:00
|
|
|
|
}
|
2013-09-10 20:56:00 +02:00
|
|
|
|
catch (Exception ex)
|
|
|
|
|
{
|
|
|
|
|
_logger.ErrorException("Error updating counts for {0}", ex, name);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
numComplete++;
|
|
|
|
|
double percent = numComplete;
|
2013-09-11 19:54:59 +02:00
|
|
|
|
percent /= count;
|
2013-09-10 20:56:00 +02:00
|
|
|
|
percent *= 90;
|
|
|
|
|
|
|
|
|
|
progress.Report(percent + 10);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
progress.Report(100);
|
|
|
|
|
}
|
|
|
|
|
|
2013-09-11 19:54:59 +02:00
|
|
|
|
private async Task UpdateItemByNameCounts(string name, CancellationToken cancellationToken, Dictionary<Guid, Dictionary<CountType, int>> counts)
|
2013-09-10 20:56:00 +02:00
|
|
|
|
{
|
2013-09-17 04:08:18 +02:00
|
|
|
|
var itemByName = _libraryManager.GetGenre(name);
|
2013-09-10 20:56:00 +02:00
|
|
|
|
|
2013-09-11 19:54:59 +02:00
|
|
|
|
foreach (var libraryId in counts.Keys)
|
2013-09-10 20:56:00 +02:00
|
|
|
|
{
|
|
|
|
|
var itemCounts = CountHelpers.GetCounts(counts[libraryId]);
|
|
|
|
|
|
2013-09-11 19:54:59 +02:00
|
|
|
|
itemByName.UserItemCounts[libraryId] = itemCounts;
|
2013-09-10 20:56:00 +02:00
|
|
|
|
}
|
2013-09-17 04:08:18 +02:00
|
|
|
|
|
|
|
|
|
await itemByName.RefreshMetadata(cancellationToken).ConfigureAwait(false);
|
2013-09-10 20:56:00 +02:00
|
|
|
|
}
|
|
|
|
|
|
2013-09-11 19:54:59 +02:00
|
|
|
|
private void SetItemCounts(Guid userId, IEnumerable<BaseItem> allItems, Dictionary<string, Dictionary<Guid, Dictionary<CountType, int>>> masterDictionary)
|
2013-09-10 20:56:00 +02:00
|
|
|
|
{
|
|
|
|
|
foreach (var media in allItems)
|
|
|
|
|
{
|
|
|
|
|
var names = media
|
|
|
|
|
.Genres
|
|
|
|
|
.Distinct(StringComparer.OrdinalIgnoreCase)
|
|
|
|
|
.ToList();
|
|
|
|
|
|
|
|
|
|
CountHelpers.SetItemCounts(userId, media, names, masterDictionary);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|