770 lines
28 KiB
C#
770 lines
28 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Diagnostics;
|
|
using System.Globalization;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Runtime.CompilerServices;
|
|
using System.Text;
|
|
using System.Text.Json;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
using AsyncKeyedLock;
|
|
using Jellyfin.Data.Enums;
|
|
using Jellyfin.Extensions;
|
|
using MediaBrowser.Common;
|
|
using MediaBrowser.Common.Configuration;
|
|
using MediaBrowser.Common.Extensions;
|
|
using MediaBrowser.Controller.Configuration;
|
|
using MediaBrowser.Controller.Library;
|
|
using MediaBrowser.Controller.MediaEncoding;
|
|
using MediaBrowser.Controller.Session;
|
|
using MediaBrowser.Controller.Streaming;
|
|
using MediaBrowser.Model.Dlna;
|
|
using MediaBrowser.Model.Entities;
|
|
using MediaBrowser.Model.IO;
|
|
using MediaBrowser.Model.MediaInfo;
|
|
using MediaBrowser.Model.Session;
|
|
using Microsoft.Extensions.Logging;
|
|
|
|
namespace MediaBrowser.MediaEncoding.Transcoding;
|
|
|
|
/// <inheritdoc cref="ITranscodeManager"/>
|
|
public sealed class TranscodeManager : ITranscodeManager, IDisposable
|
|
{
|
|
private readonly ILoggerFactory _loggerFactory;
|
|
private readonly ILogger<TranscodeManager> _logger;
|
|
private readonly IFileSystem _fileSystem;
|
|
private readonly IApplicationPaths _appPaths;
|
|
private readonly IServerConfigurationManager _serverConfigurationManager;
|
|
private readonly IUserManager _userManager;
|
|
private readonly ISessionManager _sessionManager;
|
|
private readonly EncodingHelper _encodingHelper;
|
|
private readonly IMediaEncoder _mediaEncoder;
|
|
private readonly IMediaSourceManager _mediaSourceManager;
|
|
private readonly IAttachmentExtractor _attachmentExtractor;
|
|
|
|
private readonly List<TranscodingJob> _activeTranscodingJobs = new();
|
|
private readonly AsyncKeyedLocker<string> _transcodingLocks = new(o =>
|
|
{
|
|
o.PoolSize = 20;
|
|
o.PoolInitialFill = 1;
|
|
});
|
|
|
|
/// <summary>
|
|
/// Initializes a new instance of the <see cref="TranscodeManager"/> class.
|
|
/// </summary>
|
|
/// <param name="loggerFactory">The <see cref="ILoggerFactory"/>.</param>
|
|
/// <param name="fileSystem">The <see cref="IFileSystem"/>.</param>
|
|
/// <param name="appPaths">The <see cref="IApplicationPaths"/>.</param>
|
|
/// <param name="serverConfigurationManager">The <see cref="IServerConfigurationManager"/>.</param>
|
|
/// <param name="userManager">The <see cref="IUserManager"/>.</param>
|
|
/// <param name="sessionManager">The <see cref="ISessionManager"/>.</param>
|
|
/// <param name="encodingHelper">The <see cref="EncodingHelper"/>.</param>
|
|
/// <param name="mediaEncoder">The <see cref="IMediaEncoder"/>.</param>
|
|
/// <param name="mediaSourceManager">The <see cref="IMediaSourceManager"/>.</param>
|
|
/// <param name="attachmentExtractor">The <see cref="IAttachmentExtractor"/>.</param>
|
|
public TranscodeManager(
|
|
ILoggerFactory loggerFactory,
|
|
IFileSystem fileSystem,
|
|
IApplicationPaths appPaths,
|
|
IServerConfigurationManager serverConfigurationManager,
|
|
IUserManager userManager,
|
|
ISessionManager sessionManager,
|
|
EncodingHelper encodingHelper,
|
|
IMediaEncoder mediaEncoder,
|
|
IMediaSourceManager mediaSourceManager,
|
|
IAttachmentExtractor attachmentExtractor)
|
|
{
|
|
_loggerFactory = loggerFactory;
|
|
_fileSystem = fileSystem;
|
|
_appPaths = appPaths;
|
|
_serverConfigurationManager = serverConfigurationManager;
|
|
_userManager = userManager;
|
|
_sessionManager = sessionManager;
|
|
_encodingHelper = encodingHelper;
|
|
_mediaEncoder = mediaEncoder;
|
|
_mediaSourceManager = mediaSourceManager;
|
|
_attachmentExtractor = attachmentExtractor;
|
|
|
|
_logger = loggerFactory.CreateLogger<TranscodeManager>();
|
|
DeleteEncodedMediaCache();
|
|
_sessionManager.PlaybackProgress += OnPlaybackProgress;
|
|
_sessionManager.PlaybackStart += OnPlaybackProgress;
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
public TranscodingJob? GetTranscodingJob(string playSessionId)
|
|
{
|
|
lock (_activeTranscodingJobs)
|
|
{
|
|
return _activeTranscodingJobs.FirstOrDefault(j => string.Equals(j.PlaySessionId, playSessionId, StringComparison.OrdinalIgnoreCase));
|
|
}
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
public TranscodingJob? GetTranscodingJob(string path, TranscodingJobType type)
|
|
{
|
|
lock (_activeTranscodingJobs)
|
|
{
|
|
return _activeTranscodingJobs.FirstOrDefault(j => j.Type == type && string.Equals(j.Path, path, StringComparison.OrdinalIgnoreCase));
|
|
}
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
public void PingTranscodingJob(string playSessionId, bool? isUserPaused)
|
|
{
|
|
ArgumentException.ThrowIfNullOrEmpty(playSessionId);
|
|
|
|
_logger.LogDebug("PingTranscodingJob PlaySessionId={0} isUsedPaused: {1}", playSessionId, isUserPaused);
|
|
|
|
List<TranscodingJob> jobs;
|
|
|
|
lock (_activeTranscodingJobs)
|
|
{
|
|
// This is really only needed for HLS.
|
|
// Progressive streams can stop on their own reliably.
|
|
jobs = _activeTranscodingJobs.Where(j => string.Equals(playSessionId, j.PlaySessionId, StringComparison.OrdinalIgnoreCase)).ToList();
|
|
}
|
|
|
|
foreach (var job in jobs)
|
|
{
|
|
if (isUserPaused.HasValue)
|
|
{
|
|
_logger.LogDebug("Setting job.IsUserPaused to {0}. jobId: {1}", isUserPaused, job.Id);
|
|
job.IsUserPaused = isUserPaused.Value;
|
|
}
|
|
|
|
PingTimer(job, true);
|
|
}
|
|
}
|
|
|
|
private void PingTimer(TranscodingJob job, bool isProgressCheckIn)
|
|
{
|
|
if (job.HasExited)
|
|
{
|
|
job.StopKillTimer();
|
|
return;
|
|
}
|
|
|
|
var timerDuration = 10000;
|
|
|
|
if (job.Type != TranscodingJobType.Progressive)
|
|
{
|
|
timerDuration = 60000;
|
|
}
|
|
|
|
job.PingTimeout = timerDuration;
|
|
job.LastPingDate = DateTime.UtcNow;
|
|
|
|
// Don't start the timer for playback checkins with progressive streaming
|
|
if (job.Type != TranscodingJobType.Progressive || !isProgressCheckIn)
|
|
{
|
|
job.StartKillTimer(OnTranscodeKillTimerStopped);
|
|
}
|
|
else
|
|
{
|
|
job.ChangeKillTimerIfStarted();
|
|
}
|
|
}
|
|
|
|
private async void OnTranscodeKillTimerStopped(object? state)
|
|
{
|
|
var job = state as TranscodingJob ?? throw new ArgumentException($"{nameof(state)} is not of type {nameof(TranscodingJob)}", nameof(state));
|
|
if (!job.HasExited && job.Type != TranscodingJobType.Progressive)
|
|
{
|
|
var timeSinceLastPing = (DateTime.UtcNow - job.LastPingDate).TotalMilliseconds;
|
|
|
|
if (timeSinceLastPing < job.PingTimeout)
|
|
{
|
|
job.StartKillTimer(OnTranscodeKillTimerStopped, job.PingTimeout);
|
|
return;
|
|
}
|
|
}
|
|
|
|
_logger.LogInformation("Transcoding kill timer stopped for JobId {0} PlaySessionId {1}. Killing transcoding", job.Id, job.PlaySessionId);
|
|
|
|
await KillTranscodingJob(job, true, path => true).ConfigureAwait(false);
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
public Task KillTranscodingJobs(string deviceId, string? playSessionId, Func<string, bool> deleteFiles)
|
|
{
|
|
var jobs = new List<TranscodingJob>();
|
|
|
|
lock (_activeTranscodingJobs)
|
|
{
|
|
// This is really only needed for HLS.
|
|
// Progressive streams can stop on their own reliably.
|
|
jobs.AddRange(_activeTranscodingJobs.Where(j => string.IsNullOrWhiteSpace(playSessionId)
|
|
? string.Equals(deviceId, j.DeviceId, StringComparison.OrdinalIgnoreCase)
|
|
: string.Equals(playSessionId, j.PlaySessionId, StringComparison.OrdinalIgnoreCase)));
|
|
}
|
|
|
|
return Task.WhenAll(GetKillJobs());
|
|
|
|
IEnumerable<Task> GetKillJobs()
|
|
{
|
|
foreach (var job in jobs)
|
|
{
|
|
yield return KillTranscodingJob(job, false, deleteFiles);
|
|
}
|
|
}
|
|
}
|
|
|
|
private async Task KillTranscodingJob(TranscodingJob job, bool closeLiveStream, Func<string, bool> delete)
|
|
{
|
|
job.DisposeKillTimer();
|
|
|
|
_logger.LogDebug("KillTranscodingJob - JobId {0} PlaySessionId {1}. Killing transcoding", job.Id, job.PlaySessionId);
|
|
|
|
lock (_activeTranscodingJobs)
|
|
{
|
|
_activeTranscodingJobs.Remove(job);
|
|
|
|
if (job.CancellationTokenSource?.IsCancellationRequested == false)
|
|
{
|
|
#pragma warning disable CA1849 // Can't await in lock block
|
|
job.CancellationTokenSource.Cancel();
|
|
#pragma warning restore CA1849
|
|
}
|
|
}
|
|
|
|
job.Stop();
|
|
|
|
if (delete(job.Path!))
|
|
{
|
|
await DeletePartialStreamFiles(job.Path!, job.Type, 0, 1500).ConfigureAwait(false);
|
|
}
|
|
|
|
if (closeLiveStream && !string.IsNullOrWhiteSpace(job.LiveStreamId))
|
|
{
|
|
try
|
|
{
|
|
await _mediaSourceManager.CloseLiveStream(job.LiveStreamId).ConfigureAwait(false);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.LogError(ex, "Error closing live stream for {Path}", job.Path);
|
|
}
|
|
}
|
|
}
|
|
|
|
private async Task DeletePartialStreamFiles(string path, TranscodingJobType jobType, int retryCount, int delayMs)
|
|
{
|
|
if (retryCount >= 10)
|
|
{
|
|
return;
|
|
}
|
|
|
|
_logger.LogInformation("Deleting partial stream file(s) {Path}", path);
|
|
|
|
await Task.Delay(delayMs).ConfigureAwait(false);
|
|
|
|
try
|
|
{
|
|
if (jobType == TranscodingJobType.Progressive)
|
|
{
|
|
DeleteProgressivePartialStreamFiles(path);
|
|
}
|
|
else
|
|
{
|
|
DeleteHlsPartialStreamFiles(path);
|
|
}
|
|
}
|
|
catch (IOException ex)
|
|
{
|
|
_logger.LogError(ex, "Error deleting partial stream file(s) {Path}", path);
|
|
|
|
await DeletePartialStreamFiles(path, jobType, retryCount + 1, 500).ConfigureAwait(false);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.LogError(ex, "Error deleting partial stream file(s) {Path}", path);
|
|
}
|
|
}
|
|
|
|
private void DeleteProgressivePartialStreamFiles(string outputFilePath)
|
|
{
|
|
if (File.Exists(outputFilePath))
|
|
{
|
|
_fileSystem.DeleteFile(outputFilePath);
|
|
}
|
|
}
|
|
|
|
private void DeleteHlsPartialStreamFiles(string outputFilePath)
|
|
{
|
|
var directory = Path.GetDirectoryName(outputFilePath)
|
|
?? throw new ArgumentException("Path can't be a root directory.", nameof(outputFilePath));
|
|
|
|
var name = Path.GetFileNameWithoutExtension(outputFilePath);
|
|
|
|
var filesToDelete = _fileSystem.GetFilePaths(directory)
|
|
.Where(f => f.Contains(name, StringComparison.OrdinalIgnoreCase));
|
|
|
|
List<Exception>? exs = null;
|
|
foreach (var file in filesToDelete)
|
|
{
|
|
try
|
|
{
|
|
_logger.LogDebug("Deleting HLS file {0}", file);
|
|
_fileSystem.DeleteFile(file);
|
|
}
|
|
catch (IOException ex)
|
|
{
|
|
(exs ??= new List<Exception>()).Add(ex);
|
|
_logger.LogError(ex, "Error deleting HLS file {Path}", file);
|
|
}
|
|
}
|
|
|
|
if (exs is not null)
|
|
{
|
|
throw new AggregateException("Error deleting HLS files", exs);
|
|
}
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
public void ReportTranscodingProgress(
|
|
TranscodingJob job,
|
|
StreamState state,
|
|
TimeSpan? transcodingPosition,
|
|
float? framerate,
|
|
double? percentComplete,
|
|
long? bytesTranscoded,
|
|
int? bitRate)
|
|
{
|
|
var ticks = transcodingPosition?.Ticks;
|
|
|
|
if (job is not null)
|
|
{
|
|
job.Framerate = framerate;
|
|
job.CompletionPercentage = percentComplete;
|
|
job.TranscodingPositionTicks = ticks;
|
|
job.BytesTranscoded = bytesTranscoded;
|
|
job.BitRate = bitRate;
|
|
}
|
|
|
|
var deviceId = state.Request.DeviceId;
|
|
|
|
if (!string.IsNullOrWhiteSpace(deviceId))
|
|
{
|
|
var audioCodec = state.ActualOutputAudioCodec;
|
|
var videoCodec = state.ActualOutputVideoCodec;
|
|
var hardwareAccelerationTypeString = _serverConfigurationManager.GetEncodingOptions().HardwareAccelerationType;
|
|
HardwareEncodingType? hardwareAccelerationType = null;
|
|
if (Enum.TryParse<HardwareEncodingType>(hardwareAccelerationTypeString, out var parsedHardwareAccelerationType))
|
|
{
|
|
hardwareAccelerationType = parsedHardwareAccelerationType;
|
|
}
|
|
|
|
_sessionManager.ReportTranscodingInfo(deviceId, new TranscodingInfo
|
|
{
|
|
Bitrate = bitRate ?? state.TotalOutputBitrate,
|
|
AudioCodec = audioCodec,
|
|
VideoCodec = videoCodec,
|
|
Container = state.OutputContainer,
|
|
Framerate = framerate,
|
|
CompletionPercentage = percentComplete,
|
|
Width = state.OutputWidth,
|
|
Height = state.OutputHeight,
|
|
AudioChannels = state.OutputAudioChannels,
|
|
IsAudioDirect = EncodingHelper.IsCopyCodec(state.OutputAudioCodec),
|
|
IsVideoDirect = EncodingHelper.IsCopyCodec(state.OutputVideoCodec),
|
|
HardwareAccelerationType = hardwareAccelerationType,
|
|
TranscodeReasons = state.TranscodeReasons
|
|
});
|
|
}
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
public async Task<TranscodingJob> StartFfMpeg(
|
|
StreamState state,
|
|
string outputPath,
|
|
string commandLineArguments,
|
|
Guid userId,
|
|
TranscodingJobType transcodingJobType,
|
|
CancellationTokenSource cancellationTokenSource,
|
|
string? workingDirectory = null)
|
|
{
|
|
var directory = Path.GetDirectoryName(outputPath) ?? throw new ArgumentException($"Provided path ({outputPath}) is not valid.", nameof(outputPath));
|
|
Directory.CreateDirectory(directory);
|
|
|
|
await AcquireResources(state, cancellationTokenSource).ConfigureAwait(false);
|
|
|
|
if (state.VideoRequest is not null && !EncodingHelper.IsCopyCodec(state.OutputVideoCodec))
|
|
{
|
|
var user = userId.IsEmpty() ? null : _userManager.GetUserById(userId);
|
|
if (user is not null && !user.HasPermission(PermissionKind.EnableVideoPlaybackTranscoding))
|
|
{
|
|
OnTranscodeFailedToStart(outputPath, transcodingJobType, state);
|
|
|
|
throw new ArgumentException("User does not have access to video transcoding.");
|
|
}
|
|
}
|
|
|
|
ArgumentException.ThrowIfNullOrEmpty(_mediaEncoder.EncoderPath);
|
|
|
|
// If subtitles get burned in fonts may need to be extracted from the media file
|
|
if (state.SubtitleStream is not null && state.SubtitleDeliveryMethod == SubtitleDeliveryMethod.Encode)
|
|
{
|
|
var attachmentPath = Path.Combine(_appPaths.CachePath, "attachments", state.MediaSource.Id);
|
|
if (state.MediaSource.VideoType == VideoType.Dvd || state.MediaSource.VideoType == VideoType.BluRay)
|
|
{
|
|
var concatPath = Path.Join(_appPaths.CachePath, "concat", state.MediaSource.Id + ".concat");
|
|
await _attachmentExtractor.ExtractAllAttachments(concatPath, state.MediaSource, attachmentPath, cancellationTokenSource.Token).ConfigureAwait(false);
|
|
}
|
|
else
|
|
{
|
|
await _attachmentExtractor.ExtractAllAttachments(state.MediaPath, state.MediaSource, attachmentPath, cancellationTokenSource.Token).ConfigureAwait(false);
|
|
}
|
|
|
|
if (state.SubtitleStream.IsExternal && Path.GetExtension(state.SubtitleStream.Path.AsSpan()).Equals(".mks", StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
string subtitlePath = state.SubtitleStream.Path;
|
|
string subtitlePathArgument = string.Format(CultureInfo.InvariantCulture, "file:\"{0}\"", subtitlePath.Replace("\"", "\\\"", StringComparison.Ordinal));
|
|
string subtitleId = subtitlePath.GetMD5().ToString("N", CultureInfo.InvariantCulture);
|
|
|
|
await _attachmentExtractor.ExtractAllAttachmentsExternal(subtitlePathArgument, subtitleId, attachmentPath, cancellationTokenSource.Token).ConfigureAwait(false);
|
|
}
|
|
}
|
|
|
|
var process = new Process
|
|
{
|
|
StartInfo = new ProcessStartInfo
|
|
{
|
|
WindowStyle = ProcessWindowStyle.Hidden,
|
|
CreateNoWindow = true,
|
|
UseShellExecute = false,
|
|
|
|
// Must consume both stdout and stderr or deadlocks may occur
|
|
// RedirectStandardOutput = true,
|
|
RedirectStandardError = true,
|
|
RedirectStandardInput = true,
|
|
FileName = _mediaEncoder.EncoderPath,
|
|
Arguments = commandLineArguments,
|
|
WorkingDirectory = string.IsNullOrWhiteSpace(workingDirectory) ? string.Empty : workingDirectory,
|
|
ErrorDialog = false
|
|
},
|
|
EnableRaisingEvents = true
|
|
};
|
|
|
|
var transcodingJob = OnTranscodeBeginning(
|
|
outputPath,
|
|
state.Request.PlaySessionId,
|
|
state.MediaSource.LiveStreamId,
|
|
Guid.NewGuid().ToString("N", CultureInfo.InvariantCulture),
|
|
transcodingJobType,
|
|
process,
|
|
state.Request.DeviceId,
|
|
state,
|
|
cancellationTokenSource);
|
|
|
|
_logger.LogInformation("{Filename} {Arguments}", process.StartInfo.FileName, process.StartInfo.Arguments);
|
|
|
|
var logFilePrefix = "FFmpeg.Transcode-";
|
|
if (state.VideoRequest is not null
|
|
&& EncodingHelper.IsCopyCodec(state.OutputVideoCodec))
|
|
{
|
|
logFilePrefix = EncodingHelper.IsCopyCodec(state.OutputAudioCodec)
|
|
? "FFmpeg.Remux-"
|
|
: "FFmpeg.DirectStream-";
|
|
}
|
|
|
|
if (state.VideoRequest is null && EncodingHelper.IsCopyCodec(state.OutputAudioCodec))
|
|
{
|
|
logFilePrefix = "FFmpeg.Remux-";
|
|
}
|
|
|
|
var logFilePath = Path.Combine(
|
|
_serverConfigurationManager.ApplicationPaths.LogDirectoryPath,
|
|
$"{logFilePrefix}{DateTime.Now:yyyy-MM-dd_HH-mm-ss}_{state.Request.MediaSourceId}_{Guid.NewGuid().ToString()[..8]}.log");
|
|
|
|
// FFmpeg writes debug/error info to stderr. This is useful when debugging so let's put it in the log directory.
|
|
Stream logStream = new FileStream(
|
|
logFilePath,
|
|
FileMode.Create,
|
|
FileAccess.Write,
|
|
FileShare.Read,
|
|
IODefaults.FileStreamBufferSize,
|
|
FileOptions.Asynchronous);
|
|
|
|
await JsonSerializer.SerializeAsync(logStream, state.MediaSource, cancellationToken: cancellationTokenSource.Token).ConfigureAwait(false);
|
|
var commandLineLogMessageBytes = Encoding.UTF8.GetBytes(
|
|
Environment.NewLine
|
|
+ Environment.NewLine
|
|
+ process.StartInfo.FileName + " " + process.StartInfo.Arguments
|
|
+ Environment.NewLine
|
|
+ Environment.NewLine);
|
|
|
|
await logStream.WriteAsync(commandLineLogMessageBytes, cancellationTokenSource.Token).ConfigureAwait(false);
|
|
|
|
process.Exited += (_, _) => OnFfMpegProcessExited(process, transcodingJob, state);
|
|
|
|
try
|
|
{
|
|
process.Start();
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.LogError(ex, "Error starting FFmpeg");
|
|
OnTranscodeFailedToStart(outputPath, transcodingJobType, state);
|
|
|
|
throw;
|
|
}
|
|
|
|
_logger.LogDebug("Launched FFmpeg process");
|
|
state.TranscodingJob = transcodingJob;
|
|
|
|
// Important - don't await the log task or we won't be able to kill FFmpeg when the user stops playback
|
|
_ = new JobLogger(_logger).StartStreamingLog(state, process.StandardError, logStream);
|
|
|
|
// Wait for the file to exist before proceeding
|
|
var ffmpegTargetFile = state.WaitForPath ?? outputPath;
|
|
_logger.LogDebug("Waiting for the creation of {0}", ffmpegTargetFile);
|
|
while (!File.Exists(ffmpegTargetFile) && !transcodingJob.HasExited)
|
|
{
|
|
await Task.Delay(100, cancellationTokenSource.Token).ConfigureAwait(false);
|
|
}
|
|
|
|
_logger.LogDebug("File {0} created or transcoding has finished", ffmpegTargetFile);
|
|
|
|
if (state.IsInputVideo && transcodingJob.Type == TranscodingJobType.Progressive && !transcodingJob.HasExited)
|
|
{
|
|
await Task.Delay(1000, cancellationTokenSource.Token).ConfigureAwait(false);
|
|
|
|
if (state.ReadInputAtNativeFramerate && !transcodingJob.HasExited)
|
|
{
|
|
await Task.Delay(1500, cancellationTokenSource.Token).ConfigureAwait(false);
|
|
}
|
|
}
|
|
|
|
if (!transcodingJob.HasExited)
|
|
{
|
|
StartThrottler(state, transcodingJob);
|
|
StartSegmentCleaner(state, transcodingJob);
|
|
}
|
|
else if (transcodingJob.ExitCode != 0)
|
|
{
|
|
throw new FfmpegException(string.Format(CultureInfo.InvariantCulture, "FFmpeg exited with code {0}", transcodingJob.ExitCode));
|
|
}
|
|
|
|
_logger.LogDebug("StartFfMpeg() finished successfully");
|
|
|
|
return transcodingJob;
|
|
}
|
|
|
|
private void StartThrottler(StreamState state, TranscodingJob transcodingJob)
|
|
{
|
|
if (EnableThrottling(state))
|
|
{
|
|
transcodingJob.TranscodingThrottler = new TranscodingThrottler(transcodingJob, _loggerFactory.CreateLogger<TranscodingThrottler>(), _serverConfigurationManager, _fileSystem, _mediaEncoder);
|
|
transcodingJob.TranscodingThrottler.Start();
|
|
}
|
|
}
|
|
|
|
private static bool EnableThrottling(StreamState state)
|
|
=> state.InputProtocol == MediaProtocol.File
|
|
&& state.RunTimeTicks.HasValue
|
|
&& state.RunTimeTicks.Value >= TimeSpan.FromMinutes(5).Ticks
|
|
&& state.IsInputVideo
|
|
&& state.VideoType == VideoType.VideoFile;
|
|
|
|
private void StartSegmentCleaner(StreamState state, TranscodingJob transcodingJob)
|
|
{
|
|
if (EnableSegmentCleaning(state))
|
|
{
|
|
transcodingJob.TranscodingSegmentCleaner = new TranscodingSegmentCleaner(transcodingJob, _loggerFactory.CreateLogger<TranscodingSegmentCleaner>(), _serverConfigurationManager, _fileSystem, _mediaEncoder, state.SegmentLength);
|
|
transcodingJob.TranscodingSegmentCleaner.Start();
|
|
}
|
|
}
|
|
|
|
private static bool EnableSegmentCleaning(StreamState state)
|
|
=> state.InputProtocol is MediaProtocol.File or MediaProtocol.Http
|
|
&& state.IsInputVideo
|
|
&& state.TranscodingType == TranscodingJobType.Hls
|
|
&& state.RunTimeTicks.HasValue
|
|
&& state.RunTimeTicks.Value >= TimeSpan.FromMinutes(5).Ticks;
|
|
|
|
private TranscodingJob OnTranscodeBeginning(
|
|
string path,
|
|
string? playSessionId,
|
|
string? liveStreamId,
|
|
string transcodingJobId,
|
|
TranscodingJobType type,
|
|
Process process,
|
|
string? deviceId,
|
|
StreamState state,
|
|
CancellationTokenSource cancellationTokenSource)
|
|
{
|
|
lock (_activeTranscodingJobs)
|
|
{
|
|
var job = new TranscodingJob(_loggerFactory.CreateLogger<TranscodingJob>())
|
|
{
|
|
Type = type,
|
|
Path = path,
|
|
Process = process,
|
|
ActiveRequestCount = 1,
|
|
DeviceId = deviceId,
|
|
CancellationTokenSource = cancellationTokenSource,
|
|
Id = transcodingJobId,
|
|
PlaySessionId = playSessionId,
|
|
LiveStreamId = liveStreamId,
|
|
MediaSource = state.MediaSource
|
|
};
|
|
|
|
_activeTranscodingJobs.Add(job);
|
|
|
|
ReportTranscodingProgress(job, state, null, null, null, null, null);
|
|
|
|
return job;
|
|
}
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
public void OnTranscodeEndRequest(TranscodingJob job)
|
|
{
|
|
job.ActiveRequestCount--;
|
|
_logger.LogDebug("OnTranscodeEndRequest job.ActiveRequestCount={ActiveRequestCount}", job.ActiveRequestCount);
|
|
if (job.ActiveRequestCount <= 0)
|
|
{
|
|
PingTimer(job, false);
|
|
}
|
|
}
|
|
|
|
private void OnTranscodeFailedToStart(string path, TranscodingJobType type, StreamState state)
|
|
{
|
|
lock (_activeTranscodingJobs)
|
|
{
|
|
var job = _activeTranscodingJobs.FirstOrDefault(j => j.Type == type && string.Equals(j.Path, path, StringComparison.OrdinalIgnoreCase));
|
|
|
|
if (job is not null)
|
|
{
|
|
_activeTranscodingJobs.Remove(job);
|
|
}
|
|
}
|
|
|
|
if (!string.IsNullOrWhiteSpace(state.Request.DeviceId))
|
|
{
|
|
_sessionManager.ClearTranscodingInfo(state.Request.DeviceId);
|
|
}
|
|
}
|
|
|
|
private void OnFfMpegProcessExited(Process process, TranscodingJob job, StreamState state)
|
|
{
|
|
job.HasExited = true;
|
|
job.ExitCode = process.ExitCode;
|
|
|
|
ReportTranscodingProgress(job, state, null, null, null, null, null);
|
|
|
|
_logger.LogDebug("Disposing stream resources");
|
|
state.Dispose();
|
|
|
|
if (process.ExitCode == 0)
|
|
{
|
|
_logger.LogInformation("FFmpeg exited with code 0");
|
|
}
|
|
else
|
|
{
|
|
_logger.LogError("FFmpeg exited with code {0}", process.ExitCode);
|
|
}
|
|
|
|
job.Dispose();
|
|
}
|
|
|
|
private async Task AcquireResources(StreamState state, CancellationTokenSource cancellationTokenSource)
|
|
{
|
|
if (state.MediaSource.RequiresOpening && string.IsNullOrWhiteSpace(state.Request.LiveStreamId))
|
|
{
|
|
var liveStreamResponse = await _mediaSourceManager.OpenLiveStream(
|
|
new LiveStreamRequest { OpenToken = state.MediaSource.OpenToken },
|
|
cancellationTokenSource.Token)
|
|
.ConfigureAwait(false);
|
|
var encodingOptions = _serverConfigurationManager.GetEncodingOptions();
|
|
|
|
_encodingHelper.AttachMediaSourceInfo(state, encodingOptions, liveStreamResponse.MediaSource, state.RequestedUrl);
|
|
|
|
if (state.VideoRequest is not null)
|
|
{
|
|
_encodingHelper.TryStreamCopy(state);
|
|
}
|
|
}
|
|
|
|
if (state.MediaSource.BufferMs.HasValue)
|
|
{
|
|
await Task.Delay(state.MediaSource.BufferMs.Value, cancellationTokenSource.Token).ConfigureAwait(false);
|
|
}
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
public TranscodingJob? OnTranscodeBeginRequest(string path, TranscodingJobType type)
|
|
{
|
|
lock (_activeTranscodingJobs)
|
|
{
|
|
var job = _activeTranscodingJobs
|
|
.FirstOrDefault(j => j.Type == type && string.Equals(j.Path, path, StringComparison.OrdinalIgnoreCase));
|
|
|
|
if (job is null)
|
|
{
|
|
return null;
|
|
}
|
|
|
|
job.ActiveRequestCount++;
|
|
if (string.IsNullOrWhiteSpace(job.PlaySessionId) || job.Type == TranscodingJobType.Progressive)
|
|
{
|
|
job.StopKillTimer();
|
|
}
|
|
|
|
return job;
|
|
}
|
|
}
|
|
|
|
private void OnPlaybackProgress(object? sender, PlaybackProgressEventArgs e)
|
|
{
|
|
if (!string.IsNullOrWhiteSpace(e.PlaySessionId))
|
|
{
|
|
PingTranscodingJob(e.PlaySessionId, e.IsPaused);
|
|
}
|
|
}
|
|
|
|
private void DeleteEncodedMediaCache()
|
|
{
|
|
var path = _serverConfigurationManager.GetTranscodePath();
|
|
if (!Directory.Exists(path))
|
|
{
|
|
return;
|
|
}
|
|
|
|
foreach (var file in _fileSystem.GetFilePaths(path, true))
|
|
{
|
|
try
|
|
{
|
|
_fileSystem.DeleteFile(file);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.LogError(ex, "Error deleting encoded media cache file {Path}", path);
|
|
}
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Transcoding lock.
|
|
/// </summary>
|
|
/// <param name="outputPath">The output path of the transcoded file.</param>
|
|
/// <param name="cancellationToken">The cancellation token.</param>
|
|
/// <returns>An <see cref="IDisposable"/>.</returns>
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
public ValueTask<IDisposable> LockAsync(string outputPath, CancellationToken cancellationToken)
|
|
{
|
|
return _transcodingLocks.LockAsync(outputPath, cancellationToken);
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
public void Dispose()
|
|
{
|
|
_sessionManager.PlaybackProgress -= OnPlaybackProgress;
|
|
_sessionManager.PlaybackStart -= OnPlaybackProgress;
|
|
_transcodingLocks.Dispose();
|
|
}
|
|
}
|