mirror of
				https://github.com/jellyfin/jellyfin.git
				synced 2025-11-03 19:17:24 -05:00 
			
		
		
		
	
		
			
				
	
	
		
			679 lines
		
	
	
		
			23 KiB
		
	
	
	
		
			C#
		
	
	
	
	
	
			
		
		
	
	
			679 lines
		
	
	
		
			23 KiB
		
	
	
	
		
			C#
		
	
	
	
	
	
using System;
 | 
						|
using System.Collections.Generic;
 | 
						|
using System.Diagnostics;
 | 
						|
using System.IO;
 | 
						|
using System.Linq;
 | 
						|
using System.Threading;
 | 
						|
using System.Threading.Tasks;
 | 
						|
using MediaBrowser.Api.Playback;
 | 
						|
using MediaBrowser.Common.Configuration;
 | 
						|
using MediaBrowser.Controller.Configuration;
 | 
						|
using MediaBrowser.Controller.Library;
 | 
						|
using MediaBrowser.Controller.MediaEncoding;
 | 
						|
using MediaBrowser.Controller.Plugins;
 | 
						|
using MediaBrowser.Controller.Session;
 | 
						|
using MediaBrowser.Model.IO;
 | 
						|
using MediaBrowser.Model.Session;
 | 
						|
using Microsoft.Extensions.Logging;
 | 
						|
 | 
						|
namespace MediaBrowser.Api
 | 
						|
{
 | 
						|
    /// <summary>
 | 
						|
    /// Class ServerEntryPoint.
 | 
						|
    /// </summary>
 | 
						|
    public class ApiEntryPoint : IServerEntryPoint
 | 
						|
    {
 | 
						|
        /// <summary>
 | 
						|
        /// The instance.
 | 
						|
        /// </summary>
 | 
						|
        public static ApiEntryPoint Instance;
 | 
						|
 | 
						|
        /// <summary>
 | 
						|
        /// The logger.
 | 
						|
        /// </summary>
 | 
						|
        private ILogger _logger;
 | 
						|
 | 
						|
        /// <summary>
 | 
						|
        /// The configuration manager.
 | 
						|
        /// </summary>
 | 
						|
        private IServerConfigurationManager _serverConfigurationManager;
 | 
						|
 | 
						|
        private readonly ISessionManager _sessionManager;
 | 
						|
        private readonly IFileSystem _fileSystem;
 | 
						|
        private readonly IMediaSourceManager _mediaSourceManager;
 | 
						|
 | 
						|
        /// <summary>
 | 
						|
        /// The active transcoding jobs
 | 
						|
        /// </summary>
 | 
						|
        private readonly List<TranscodingJob> _activeTranscodingJobs = new List<TranscodingJob>();
 | 
						|
 | 
						|
        private readonly Dictionary<string, SemaphoreSlim> _transcodingLocks =
 | 
						|
            new Dictionary<string, SemaphoreSlim>();
 | 
						|
 | 
						|
        private bool _disposed = false;
 | 
						|
 | 
						|
        /// <summary>
 | 
						|
        /// Initializes a new instance of the <see cref="ApiEntryPoint" /> class.
 | 
						|
        /// </summary>
 | 
						|
        /// <param name="logger">The logger.</param>
 | 
						|
        /// <param name="sessionManager">The session manager.</param>
 | 
						|
        /// <param name="config">The configuration.</param>
 | 
						|
        /// <param name="fileSystem">The file system.</param>
 | 
						|
        /// <param name="mediaSourceManager">The media source manager.</param>
 | 
						|
        public ApiEntryPoint(
 | 
						|
            ILogger<ApiEntryPoint> logger,
 | 
						|
            ISessionManager sessionManager,
 | 
						|
            IServerConfigurationManager config,
 | 
						|
            IFileSystem fileSystem,
 | 
						|
            IMediaSourceManager mediaSourceManager)
 | 
						|
        {
 | 
						|
            _logger = logger;
 | 
						|
            _sessionManager = sessionManager;
 | 
						|
            _serverConfigurationManager = config;
 | 
						|
            _fileSystem = fileSystem;
 | 
						|
            _mediaSourceManager = mediaSourceManager;
 | 
						|
 | 
						|
            _sessionManager.PlaybackProgress += OnPlaybackProgress;
 | 
						|
            _sessionManager.PlaybackStart += OnPlaybackStart;
 | 
						|
 | 
						|
            Instance = this;
 | 
						|
        }
 | 
						|
 | 
						|
        public static string[] Split(string value, char separator, bool removeEmpty)
 | 
						|
        {
 | 
						|
            if (string.IsNullOrWhiteSpace(value))
 | 
						|
            {
 | 
						|
                return Array.Empty<string>();
 | 
						|
            }
 | 
						|
 | 
						|
            return removeEmpty
 | 
						|
                ? value.Split(new[] { separator }, StringSplitOptions.RemoveEmptyEntries)
 | 
						|
                : value.Split(separator);
 | 
						|
        }
 | 
						|
 | 
						|
        public SemaphoreSlim GetTranscodingLock(string outputPath)
 | 
						|
        {
 | 
						|
            lock (_transcodingLocks)
 | 
						|
            {
 | 
						|
                if (!_transcodingLocks.TryGetValue(outputPath, out SemaphoreSlim result))
 | 
						|
                {
 | 
						|
                    result = new SemaphoreSlim(1, 1);
 | 
						|
                    _transcodingLocks[outputPath] = result;
 | 
						|
                }
 | 
						|
 | 
						|
                return result;
 | 
						|
            }
 | 
						|
        }
 | 
						|
 | 
						|
        private void OnPlaybackStart(object sender, PlaybackProgressEventArgs e)
 | 
						|
        {
 | 
						|
            if (!string.IsNullOrWhiteSpace(e.PlaySessionId))
 | 
						|
            {
 | 
						|
                PingTranscodingJob(e.PlaySessionId, e.IsPaused);
 | 
						|
            }
 | 
						|
        }
 | 
						|
 | 
						|
        private void OnPlaybackProgress(object sender, PlaybackProgressEventArgs e)
 | 
						|
        {
 | 
						|
            if (!string.IsNullOrWhiteSpace(e.PlaySessionId))
 | 
						|
            {
 | 
						|
                PingTranscodingJob(e.PlaySessionId, e.IsPaused);
 | 
						|
            }
 | 
						|
        }
 | 
						|
 | 
						|
        /// <summary>
 | 
						|
        /// Runs this instance.
 | 
						|
        /// </summary>
 | 
						|
        public Task RunAsync()
 | 
						|
        {
 | 
						|
            try
 | 
						|
            {
 | 
						|
                DeleteEncodedMediaCache();
 | 
						|
            }
 | 
						|
            catch (Exception ex)
 | 
						|
            {
 | 
						|
                _logger.LogError(ex, "Error deleting encoded media cache");
 | 
						|
            }
 | 
						|
 | 
						|
            return Task.CompletedTask;
 | 
						|
        }
 | 
						|
 | 
						|
        /// <summary>
 | 
						|
        /// Deletes the encoded media cache.
 | 
						|
        /// </summary>
 | 
						|
        private void DeleteEncodedMediaCache()
 | 
						|
        {
 | 
						|
            var path = _serverConfigurationManager.GetTranscodePath();
 | 
						|
            if (!Directory.Exists(path))
 | 
						|
            {
 | 
						|
                return;
 | 
						|
            }
 | 
						|
 | 
						|
            foreach (var file in _fileSystem.GetFilePaths(path, true))
 | 
						|
            {
 | 
						|
                _fileSystem.DeleteFile(file);
 | 
						|
            }
 | 
						|
        }
 | 
						|
 | 
						|
        /// <inheritdoc />
 | 
						|
        public void Dispose()
 | 
						|
        {
 | 
						|
            Dispose(true);
 | 
						|
            GC.SuppressFinalize(this);
 | 
						|
        }
 | 
						|
 | 
						|
        /// <summary>
 | 
						|
        /// Releases unmanaged and - optionally - managed resources.
 | 
						|
        /// </summary>
 | 
						|
        /// <param name="dispose"><c>true</c> to release both managed and unmanaged resources; <c>false</c> to release only unmanaged resources.</param>
 | 
						|
        protected virtual void Dispose(bool dispose)
 | 
						|
        {
 | 
						|
            if (_disposed)
 | 
						|
            {
 | 
						|
                return;
 | 
						|
            }
 | 
						|
 | 
						|
            if (dispose)
 | 
						|
            {
 | 
						|
                // TODO: dispose
 | 
						|
            }
 | 
						|
 | 
						|
            var jobs = _activeTranscodingJobs.ToList();
 | 
						|
            var jobCount = jobs.Count;
 | 
						|
 | 
						|
            IEnumerable<Task> GetKillJobs()
 | 
						|
            {
 | 
						|
                foreach (var job in jobs)
 | 
						|
                {
 | 
						|
                    yield return KillTranscodingJob(job, false, path => true);
 | 
						|
                }
 | 
						|
            }
 | 
						|
 | 
						|
            // Wait for all processes to be killed
 | 
						|
            if (jobCount > 0)
 | 
						|
            {
 | 
						|
                Task.WaitAll(GetKillJobs().ToArray());
 | 
						|
            }
 | 
						|
 | 
						|
            _activeTranscodingJobs.Clear();
 | 
						|
            _transcodingLocks.Clear();
 | 
						|
 | 
						|
            _sessionManager.PlaybackProgress -= OnPlaybackProgress;
 | 
						|
            _sessionManager.PlaybackStart -= OnPlaybackStart;
 | 
						|
 | 
						|
            _disposed = true;
 | 
						|
        }
 | 
						|
 | 
						|
 | 
						|
        /// <summary>
 | 
						|
        /// Called when [transcode beginning].
 | 
						|
        /// </summary>
 | 
						|
        /// <param name="path">The path.</param>
 | 
						|
        /// <param name="playSessionId">The play session identifier.</param>
 | 
						|
        /// <param name="liveStreamId">The live stream identifier.</param>
 | 
						|
        /// <param name="transcodingJobId">The transcoding job identifier.</param>
 | 
						|
        /// <param name="type">The type.</param>
 | 
						|
        /// <param name="process">The process.</param>
 | 
						|
        /// <param name="deviceId">The device id.</param>
 | 
						|
        /// <param name="state">The state.</param>
 | 
						|
        /// <param name="cancellationTokenSource">The cancellation token source.</param>
 | 
						|
        /// <returns>TranscodingJob.</returns>
 | 
						|
        public 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(_logger)
 | 
						|
                {
 | 
						|
                    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;
 | 
						|
            }
 | 
						|
        }
 | 
						|
 | 
						|
        public void ReportTranscodingProgress(TranscodingJob job, StreamState state, TimeSpan? transcodingPosition, float? framerate, double? percentComplete, long? bytesTranscoded, int? bitRate)
 | 
						|
        {
 | 
						|
            var ticks = transcodingPosition?.Ticks;
 | 
						|
 | 
						|
            if (job != 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;
 | 
						|
 | 
						|
                _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 = string.Equals(state.OutputAudioCodec, "copy", StringComparison.OrdinalIgnoreCase),
 | 
						|
                    IsVideoDirect = string.Equals(state.OutputVideoCodec, "copy", StringComparison.OrdinalIgnoreCase),
 | 
						|
                    TranscodeReasons = state.TranscodeReasons
 | 
						|
                });
 | 
						|
            }
 | 
						|
        }
 | 
						|
 | 
						|
        /// <summary>
 | 
						|
        /// <summary>
 | 
						|
        /// The progressive
 | 
						|
        /// </summary>
 | 
						|
        /// Called when [transcode failed to start].
 | 
						|
        /// </summary>
 | 
						|
        /// <param name="path">The path.</param>
 | 
						|
        /// <param name="type">The type.</param>
 | 
						|
        /// <param name="state">The state.</param>
 | 
						|
        public 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 != null)
 | 
						|
                {
 | 
						|
                    _activeTranscodingJobs.Remove(job);
 | 
						|
                }
 | 
						|
            }
 | 
						|
 | 
						|
            lock (_transcodingLocks)
 | 
						|
            {
 | 
						|
                _transcodingLocks.Remove(path);
 | 
						|
            }
 | 
						|
 | 
						|
            if (!string.IsNullOrWhiteSpace(state.Request.DeviceId))
 | 
						|
            {
 | 
						|
                _sessionManager.ClearTranscodingInfo(state.Request.DeviceId);
 | 
						|
            }
 | 
						|
        }
 | 
						|
 | 
						|
        /// <summary>
 | 
						|
        /// Determines whether [has active transcoding job] [the specified path].
 | 
						|
        /// </summary>
 | 
						|
        /// <param name="path">The path.</param>
 | 
						|
        /// <param name="type">The type.</param>
 | 
						|
        /// <returns><c>true</c> if [has active transcoding job] [the specified path]; otherwise, <c>false</c>.</returns>
 | 
						|
        public bool HasActiveTranscodingJob(string path, TranscodingJobType type)
 | 
						|
        {
 | 
						|
            return GetTranscodingJob(path, type) != null;
 | 
						|
        }
 | 
						|
 | 
						|
        public TranscodingJob GetTranscodingJob(string path, TranscodingJobType type)
 | 
						|
        {
 | 
						|
            lock (_activeTranscodingJobs)
 | 
						|
            {
 | 
						|
                return _activeTranscodingJobs.FirstOrDefault(j => j.Type == type && string.Equals(j.Path, path, StringComparison.OrdinalIgnoreCase));
 | 
						|
            }
 | 
						|
        }
 | 
						|
 | 
						|
        public TranscodingJob GetTranscodingJob(string playSessionId)
 | 
						|
        {
 | 
						|
            lock (_activeTranscodingJobs)
 | 
						|
            {
 | 
						|
                return _activeTranscodingJobs.FirstOrDefault(j => string.Equals(j.PlaySessionId, playSessionId, StringComparison.OrdinalIgnoreCase));
 | 
						|
            }
 | 
						|
        }
 | 
						|
 | 
						|
        /// <summary>
 | 
						|
        /// Called when [transcode begin request].
 | 
						|
        /// </summary>
 | 
						|
        /// <param name="path">The path.</param>
 | 
						|
        /// <param name="type">The type.</param>
 | 
						|
        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 == null)
 | 
						|
                {
 | 
						|
                    return null;
 | 
						|
                }
 | 
						|
 | 
						|
                OnTranscodeBeginRequest(job);
 | 
						|
 | 
						|
                return job;
 | 
						|
            }
 | 
						|
        }
 | 
						|
 | 
						|
        public void OnTranscodeBeginRequest(TranscodingJob job)
 | 
						|
        {
 | 
						|
            job.ActiveRequestCount++;
 | 
						|
 | 
						|
            if (string.IsNullOrWhiteSpace(job.PlaySessionId) || job.Type == TranscodingJobType.Progressive)
 | 
						|
            {
 | 
						|
                job.StopKillTimer();
 | 
						|
            }
 | 
						|
        }
 | 
						|
 | 
						|
        public void OnTranscodeEndRequest(TranscodingJob job)
 | 
						|
        {
 | 
						|
            job.ActiveRequestCount--;
 | 
						|
            _logger.LogDebug("OnTranscodeEndRequest job.ActiveRequestCount={0}", job.ActiveRequestCount);
 | 
						|
            if (job.ActiveRequestCount <= 0)
 | 
						|
            {
 | 
						|
                PingTimer(job, false);
 | 
						|
            }
 | 
						|
        }
 | 
						|
 | 
						|
        internal void PingTranscodingJob(string playSessionId, bool? isUserPaused)
 | 
						|
        {
 | 
						|
            if (string.IsNullOrEmpty(playSessionId))
 | 
						|
            {
 | 
						|
                throw new ArgumentNullException(nameof(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();
 | 
						|
            }
 | 
						|
        }
 | 
						|
 | 
						|
        /// <summary>
 | 
						|
        /// Called when [transcode kill timer stopped].
 | 
						|
        /// </summary>
 | 
						|
        /// <param name="state">The state.</param>
 | 
						|
        private async void OnTranscodeKillTimerStopped(object state)
 | 
						|
        {
 | 
						|
            var job = (TranscodingJob)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);
 | 
						|
        }
 | 
						|
 | 
						|
        /// <summary>
 | 
						|
        /// Kills the single transcoding job.
 | 
						|
        /// </summary>
 | 
						|
        /// <param name="deviceId">The device id.</param>
 | 
						|
        /// <param name="playSessionId">The play session identifier.</param>
 | 
						|
        /// <param name="deleteFiles">The delete files.</param>
 | 
						|
        /// <returns>Task.</returns>
 | 
						|
        internal Task KillTranscodingJobs(string deviceId, string playSessionId, Func<string, bool> deleteFiles)
 | 
						|
        {
 | 
						|
            return KillTranscodingJobs(j => string.IsNullOrWhiteSpace(playSessionId)
 | 
						|
                ? string.Equals(deviceId, j.DeviceId, StringComparison.OrdinalIgnoreCase)
 | 
						|
                : string.Equals(playSessionId, j.PlaySessionId, StringComparison.OrdinalIgnoreCase), deleteFiles);
 | 
						|
        }
 | 
						|
 | 
						|
        /// <summary>
 | 
						|
        /// Kills the transcoding jobs.
 | 
						|
        /// </summary>
 | 
						|
        /// <param name="killJob">The kill job.</param>
 | 
						|
        /// <param name="deleteFiles">The delete files.</param>
 | 
						|
        /// <returns>Task.</returns>
 | 
						|
        private Task KillTranscodingJobs(Func<TranscodingJob, bool> killJob, 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(killJob));
 | 
						|
            }
 | 
						|
 | 
						|
            if (jobs.Count == 0)
 | 
						|
            {
 | 
						|
                return Task.CompletedTask;
 | 
						|
            }
 | 
						|
 | 
						|
            IEnumerable<Task> GetKillJobs()
 | 
						|
            {
 | 
						|
                foreach (var job in jobs)
 | 
						|
                {
 | 
						|
                    yield return KillTranscodingJob(job, false, deleteFiles);
 | 
						|
                }
 | 
						|
            }
 | 
						|
 | 
						|
            return Task.WhenAll(GetKillJobs());
 | 
						|
        }
 | 
						|
 | 
						|
        /// <summary>
 | 
						|
        /// Kills the transcoding job.
 | 
						|
        /// </summary>
 | 
						|
        /// <param name="job">The job.</param>
 | 
						|
        /// <param name="closeLiveStream">if set to <c>true</c> [close live stream].</param>
 | 
						|
        /// <param name="delete">The delete.</param>
 | 
						|
        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)
 | 
						|
                {
 | 
						|
                    job.CancellationTokenSource.Cancel();
 | 
						|
                }
 | 
						|
            }
 | 
						|
 | 
						|
            lock (_transcodingLocks)
 | 
						|
            {
 | 
						|
                _transcodingLocks.Remove(job.Path);
 | 
						|
            }
 | 
						|
 | 
						|
            lock (job.ProcessLock)
 | 
						|
            {
 | 
						|
                job.TranscodingThrottler?.Stop().GetAwaiter().GetResult();
 | 
						|
 | 
						|
                var process = job.Process;
 | 
						|
 | 
						|
                var hasExited = job.HasExited;
 | 
						|
 | 
						|
                if (!hasExited)
 | 
						|
                {
 | 
						|
                    try
 | 
						|
                    {
 | 
						|
                        _logger.LogInformation("Stopping ffmpeg process with q command for {Path}", job.Path);
 | 
						|
 | 
						|
                        process.StandardInput.WriteLine("q");
 | 
						|
 | 
						|
                        // Need to wait because killing is asynchronous
 | 
						|
                        if (!process.WaitForExit(5000))
 | 
						|
                        {
 | 
						|
                            _logger.LogInformation("Killing ffmpeg process for {Path}", job.Path);
 | 
						|
                            process.Kill();
 | 
						|
                        }
 | 
						|
                    }
 | 
						|
                    catch (InvalidOperationException)
 | 
						|
                    {
 | 
						|
                    }
 | 
						|
                }
 | 
						|
            }
 | 
						|
 | 
						|
            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);
 | 
						|
            }
 | 
						|
        }
 | 
						|
 | 
						|
        /// <summary>
 | 
						|
        /// Deletes the progressive partial stream files.
 | 
						|
        /// </summary>
 | 
						|
        /// <param name="outputFilePath">The output file path.</param>
 | 
						|
        private void DeleteProgressivePartialStreamFiles(string outputFilePath)
 | 
						|
        {
 | 
						|
            if (File.Exists(outputFilePath))
 | 
						|
            {
 | 
						|
                _fileSystem.DeleteFile(outputFilePath);
 | 
						|
            }
 | 
						|
        }
 | 
						|
 | 
						|
        /// <summary>
 | 
						|
        /// Deletes the HLS partial stream files.
 | 
						|
        /// </summary>
 | 
						|
        /// <param name="outputFilePath">The output file path.</param>
 | 
						|
        private void DeleteHlsPartialStreamFiles(string outputFilePath)
 | 
						|
        {
 | 
						|
            var directory = Path.GetDirectoryName(outputFilePath);
 | 
						|
            var name = Path.GetFileNameWithoutExtension(outputFilePath);
 | 
						|
 | 
						|
            var filesToDelete = _fileSystem.GetFilePaths(directory)
 | 
						|
                .Where(f => f.IndexOf(name, StringComparison.OrdinalIgnoreCase) != -1);
 | 
						|
 | 
						|
            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>(4)).Add(ex);
 | 
						|
                    _logger.LogError(ex, "Error deleting HLS file {Path}", file);
 | 
						|
                }
 | 
						|
            }
 | 
						|
 | 
						|
            if (exs != null)
 | 
						|
            {
 | 
						|
                throw new AggregateException("Error deleting HLS files", exs);
 | 
						|
            }
 | 
						|
        }
 | 
						|
    }
 | 
						|
}
 |