jellyfin/MediaBrowser.Server.Implementations/EntryPoints/UsageEntryPoint.cs

138 lines
4.5 KiB
C#
Raw Normal View History

2014-02-15 22:42:06 +00:00
using MediaBrowser.Common;
2014-05-28 14:21:07 +00:00
using MediaBrowser.Common.Extensions;
2014-02-15 22:42:06 +00:00
using MediaBrowser.Common.Net;
2015-01-02 05:36:27 +00:00
using MediaBrowser.Controller.Library;
2014-02-15 22:42:06 +00:00
using MediaBrowser.Controller.Plugins;
2014-05-28 14:21:07 +00:00
using MediaBrowser.Controller.Session;
2014-02-15 22:42:06 +00:00
using MediaBrowser.Model.Logging;
using System;
2014-05-28 14:21:07 +00:00
using System.Collections.Concurrent;
using System.Collections.Generic;
2014-02-15 22:42:06 +00:00
using System.Threading;
using System.Threading.Tasks;
2016-03-03 05:17:00 +00:00
using MediaBrowser.Controller.Configuration;
2014-02-15 22:42:06 +00:00
namespace MediaBrowser.Server.Implementations.EntryPoints
{
/// <summary>
/// Class UsageEntryPoint
/// </summary>
public class UsageEntryPoint : IServerEntryPoint
{
private readonly IApplicationHost _applicationHost;
private readonly IHttpClient _httpClient;
private readonly ILogger _logger;
2014-05-28 14:21:07 +00:00
private readonly ISessionManager _sessionManager;
2015-01-02 05:36:27 +00:00
private readonly IUserManager _userManager;
2016-03-03 05:17:00 +00:00
private readonly IServerConfigurationManager _config;
2014-02-15 22:42:06 +00:00
2014-05-28 14:21:07 +00:00
private readonly ConcurrentDictionary<Guid, ClientInfo> _apps = new ConcurrentDictionary<Guid, ClientInfo>();
2016-03-03 05:17:00 +00:00
public UsageEntryPoint(ILogger logger, IApplicationHost applicationHost, IHttpClient httpClient, ISessionManager sessionManager, IUserManager userManager, IServerConfigurationManager config)
2014-02-15 22:42:06 +00:00
{
_logger = logger;
_applicationHost = applicationHost;
_httpClient = httpClient;
2014-05-28 14:21:07 +00:00
_sessionManager = sessionManager;
2015-01-04 14:34:15 +00:00
_userManager = userManager;
2016-03-03 05:17:00 +00:00
_config = config;
2014-05-28 14:21:07 +00:00
_sessionManager.SessionStarted += _sessionManager_SessionStarted;
}
void _sessionManager_SessionStarted(object sender, SessionEventArgs e)
{
var session = e.SessionInfo;
if (!string.IsNullOrEmpty(session.Client) &&
2014-05-28 14:25:53 +00:00
!string.IsNullOrEmpty(session.DeviceName) &&
2014-05-28 14:30:55 +00:00
!string.IsNullOrEmpty(session.DeviceId) &&
!string.IsNullOrEmpty(session.ApplicationVersion))
2014-05-28 14:21:07 +00:00
{
var keys = new List<string>
2014-05-28 14:25:53 +00:00
{
session.Client,
session.DeviceName,
2014-05-28 14:30:55 +00:00
session.DeviceId,
session.ApplicationVersion
2014-05-28 14:25:53 +00:00
};
2014-05-28 14:21:07 +00:00
var key = string.Join("_", keys.ToArray()).GetMD5();
_apps.GetOrAdd(key, guid => GetNewClientInfo(session));
}
}
private async void ReportNewSession(ClientInfo client)
{
2016-03-03 05:17:00 +00:00
if (!_config.Configuration.EnableAnonymousUsageReporting)
{
return;
}
2014-05-28 14:21:07 +00:00
try
{
2015-12-30 07:11:58 +00:00
await new UsageReporter(_applicationHost, _httpClient, _userManager, _logger)
2014-05-28 14:21:07 +00:00
.ReportAppUsage(client, CancellationToken.None)
.ConfigureAwait(false);
}
catch (Exception ex)
{
_logger.ErrorException("Error sending anonymous usage statistics.", ex);
}
}
private ClientInfo GetNewClientInfo(SessionInfo session)
{
var info = new ClientInfo
{
AppName = session.Client,
AppVersion = session.ApplicationVersion,
2014-05-28 14:30:55 +00:00
DeviceName = session.DeviceName,
2014-05-28 14:25:53 +00:00
DeviceId = session.DeviceId
2014-05-28 14:21:07 +00:00
};
// Report usage to remote server, except for web client, since we already have data on that
if (!string.Equals(info.AppName, "Dashboard", StringComparison.OrdinalIgnoreCase))
{
ReportNewSession(info);
}
return info;
2014-02-15 22:42:06 +00:00
}
public async void Run()
2014-02-15 22:42:06 +00:00
{
await Task.Delay(5000).ConfigureAwait(false);
OnTimerFired();
2014-02-15 22:42:06 +00:00
}
/// <summary>
/// Called when [timer fired].
/// </summary>
private async void OnTimerFired()
2014-02-15 22:42:06 +00:00
{
2016-03-03 05:17:00 +00:00
if (!_config.Configuration.EnableAnonymousUsageReporting)
{
return;
}
2014-02-15 22:42:06 +00:00
try
{
2015-12-30 07:11:58 +00:00
await new UsageReporter(_applicationHost, _httpClient, _userManager, _logger)
2014-05-28 14:21:07 +00:00
.ReportServerUsage(CancellationToken.None)
.ConfigureAwait(false);
2014-02-15 22:42:06 +00:00
}
catch (Exception ex)
{
_logger.ErrorException("Error sending anonymous usage statistics.", ex);
}
}
public void Dispose()
{
2014-05-28 14:21:07 +00:00
_sessionManager.SessionStarted -= _sessionManager_SessionStarted;
2014-02-15 22:42:06 +00:00
}
}
}