using System; using System.Diagnostics; using System.Threading; using MediaBrowser.Model.Dto; using Microsoft.Extensions.Logging; namespace MediaBrowser.Controller.MediaEncoding; /// /// Class TranscodingJob. /// public sealed class TranscodingJob : IDisposable { private readonly ILogger _logger; private readonly object _processLock = new(); private readonly object _timerLock = new(); private Timer? _killTimer; /// /// Initializes a new instance of the class. /// /// Instance of the interface. public TranscodingJob(ILogger logger) { _logger = logger; } /// /// Gets or sets the play session identifier. /// public string? PlaySessionId { get; set; } /// /// Gets or sets the live stream identifier. /// public string? LiveStreamId { get; set; } /// /// Gets or sets a value indicating whether is live output. /// public bool IsLiveOutput { get; set; } /// /// Gets or sets the path. /// public MediaSourceInfo? MediaSource { get; set; } /// /// Gets or sets path. /// public string? Path { get; set; } /// /// Gets or sets the type. /// public TranscodingJobType Type { get; set; } /// /// Gets or sets the process. /// public Process? Process { get; set; } /// /// Gets or sets the active request count. /// public int ActiveRequestCount { get; set; } /// /// Gets or sets device id. /// public string? DeviceId { get; set; } /// /// Gets or sets cancellation token source. /// public CancellationTokenSource? CancellationTokenSource { get; set; } /// /// Gets or sets a value indicating whether has exited. /// public bool HasExited { get; set; } /// /// Gets or sets exit code. /// public int ExitCode { get; set; } /// /// Gets or sets a value indicating whether is user paused. /// public bool IsUserPaused { get; set; } /// /// Gets or sets id. /// public string? Id { get; set; } /// /// Gets or sets framerate. /// public float? Framerate { get; set; } /// /// Gets or sets completion percentage. /// public double? CompletionPercentage { get; set; } /// /// Gets or sets bytes downloaded. /// public long BytesDownloaded { get; set; } /// /// Gets or sets bytes transcoded. /// public long? BytesTranscoded { get; set; } /// /// Gets or sets bit rate. /// public int? BitRate { get; set; } /// /// Gets or sets transcoding position ticks. /// public long? TranscodingPositionTicks { get; set; } /// /// Gets or sets download position ticks. /// public long? DownloadPositionTicks { get; set; } /// /// Gets or sets transcoding throttler. /// public TranscodingThrottler? TranscodingThrottler { get; set; } /// /// Gets or sets transcoding segment cleaner. /// public TranscodingSegmentCleaner? TranscodingSegmentCleaner { get; set; } /// /// Gets or sets last ping date. /// public DateTime LastPingDate { get; set; } /// /// Gets or sets ping timeout. /// public int PingTimeout { get; set; } /// /// Stop kill timer. /// public void StopKillTimer() { lock (_timerLock) { _killTimer?.Change(Timeout.Infinite, Timeout.Infinite); } } /// /// Dispose kill timer. /// public void DisposeKillTimer() { lock (_timerLock) { if (_killTimer is not null) { _killTimer.Dispose(); _killTimer = null; } } } /// /// Start kill timer. /// /// Callback action. public void StartKillTimer(Action callback) { StartKillTimer(callback, PingTimeout); } /// /// Start kill timer. /// /// Callback action. /// Callback interval. public void StartKillTimer(Action callback, int intervalMs) { if (HasExited) { return; } lock (_timerLock) { if (_killTimer is null) { _logger.LogDebug("Starting kill timer at {0}ms. JobId {1} PlaySessionId {2}", intervalMs, Id, PlaySessionId); _killTimer = new Timer(new TimerCallback(callback), this, intervalMs, Timeout.Infinite); } else { _logger.LogDebug("Changing kill timer to {0}ms. JobId {1} PlaySessionId {2}", intervalMs, Id, PlaySessionId); _killTimer.Change(intervalMs, Timeout.Infinite); } } } /// /// Change kill timer if started. /// public void ChangeKillTimerIfStarted() { if (HasExited) { return; } lock (_timerLock) { if (_killTimer is not null) { var intervalMs = PingTimeout; _logger.LogDebug("Changing kill timer to {0}ms. JobId {1} PlaySessionId {2}", intervalMs, Id, PlaySessionId); _killTimer.Change(intervalMs, Timeout.Infinite); } } } /// /// Stops the transcoding job. /// public void Stop() { lock (_processLock) { #pragma warning disable CA1849 // Can't await in lock block TranscodingThrottler?.Stop().GetAwaiter().GetResult(); TranscodingSegmentCleaner?.Stop(); var process = Process; if (!HasExited) { try { _logger.LogInformation("Stopping ffmpeg process with q command for {Path}", Path); process!.StandardInput.WriteLine("q"); // Need to wait because killing is asynchronous. if (!process.WaitForExit(5000)) { _logger.LogInformation("Killing FFmpeg process for {Path}", Path); process.Kill(); } } catch (InvalidOperationException) { } } #pragma warning restore CA1849 } } /// public void Dispose() { Process?.Dispose(); Process = null; _killTimer?.Dispose(); _killTimer = null; CancellationTokenSource?.Dispose(); CancellationTokenSource = null; TranscodingThrottler?.Dispose(); TranscodingThrottler = null; TranscodingSegmentCleaner?.Dispose(); TranscodingSegmentCleaner = null; } }