jellyfin-server/MediaBrowser.Api/Playback/Progressive/ProgressiveStreamWriter.cs

127 lines
3.9 KiB
C#
Raw Normal View History

2013-02-27 04:19:05 +00:00
using MediaBrowser.Common.IO;
using MediaBrowser.Model.Logging;
2013-12-07 15:52:38 +00:00
using ServiceStack.Web;
using System.Collections.Generic;
2013-02-27 04:19:05 +00:00
using System.IO;
using System.Threading.Tasks;
namespace MediaBrowser.Api.Playback.Progressive
{
public class ProgressiveStreamWriter : IStreamWriter, IHasOptions
2013-02-27 04:19:05 +00:00
{
2013-03-11 04:04:08 +00:00
private string Path { get; set; }
private ILogger Logger { get; set; }
private readonly IFileSystem _fileSystem;
2014-07-17 22:21:35 +00:00
private readonly TranscodingJob _job;
2013-03-11 04:04:08 +00:00
/// <summary>
/// The _options
/// </summary>
private readonly IDictionary<string, string> _options = new Dictionary<string, string>();
/// <summary>
/// Gets the options.
/// </summary>
/// <value>The options.</value>
public IDictionary<string, string> Options
{
get { return _options; }
}
2013-03-11 04:04:08 +00:00
/// <summary>
/// Initializes a new instance of the <see cref="ProgressiveStreamWriter" /> class.
/// </summary>
/// <param name="path">The path.</param>
/// <param name="logger">The logger.</param>
2014-06-02 19:32:41 +00:00
/// <param name="fileSystem">The file system.</param>
2014-07-17 22:21:35 +00:00
public ProgressiveStreamWriter(string path, ILogger logger, IFileSystem fileSystem, TranscodingJob job)
2013-03-11 04:04:08 +00:00
{
Path = path;
Logger = logger;
_fileSystem = fileSystem;
2014-07-17 22:21:35 +00:00
_job = job;
2013-03-11 04:04:08 +00:00
}
2013-02-27 04:19:05 +00:00
/// <summary>
/// Writes to.
/// </summary>
/// <param name="responseStream">The response stream.</param>
public void WriteTo(Stream responseStream)
{
var task = WriteToAsync(responseStream);
Task.WaitAll(task);
}
/// <summary>
/// Writes to async.
/// </summary>
/// <param name="responseStream">The response stream.</param>
/// <returns>Task.</returns>
public async Task WriteToAsync(Stream responseStream)
{
try
{
2014-07-17 22:21:35 +00:00
await new ProgressiveFileCopier(_fileSystem, _job)
.StreamFile(Path, responseStream).ConfigureAwait(false);
2013-02-27 04:19:05 +00:00
}
2014-06-20 15:10:40 +00:00
catch
2013-02-27 04:19:05 +00:00
{
2014-06-20 15:10:40 +00:00
Logger.Error("Error streaming media. The client has most likely disconnected or transcoding has failed.");
2013-03-11 04:04:08 +00:00
throw;
2013-02-27 04:19:05 +00:00
}
finally
{
2013-03-08 19:14:09 +00:00
ApiEntryPoint.Instance.OnTranscodeEndRequest(Path, TranscodingJobType.Progressive);
2013-02-27 04:19:05 +00:00
}
}
2014-06-26 17:04:11 +00:00
}
2013-02-27 04:19:05 +00:00
2014-06-26 17:04:11 +00:00
public class ProgressiveFileCopier
{
private readonly IFileSystem _fileSystem;
2014-07-17 22:21:35 +00:00
private readonly TranscodingJob _job;
2014-06-26 17:04:11 +00:00
2014-07-17 22:21:35 +00:00
public ProgressiveFileCopier(IFileSystem fileSystem, TranscodingJob job)
2014-06-26 17:04:11 +00:00
{
_fileSystem = fileSystem;
2014-07-17 22:21:35 +00:00
_job = job;
2014-06-26 17:04:11 +00:00
}
public async Task StreamFile(string path, Stream outputStream)
2013-02-27 04:19:05 +00:00
{
var eofCount = 0;
long position = 0;
using (var fs = _fileSystem.GetFileStream(path, FileMode.Open, FileAccess.Read, FileShare.ReadWrite, true))
2013-02-27 04:19:05 +00:00
{
while (eofCount < 15)
{
await fs.CopyToAsync(outputStream).ConfigureAwait(false);
var fsPosition = fs.Position;
var bytesRead = fsPosition - position;
//Logger.Debug("Streamed {0} bytes from file {1}", bytesRead, path);
2013-02-27 04:19:05 +00:00
if (bytesRead == 0)
{
2014-07-17 22:21:35 +00:00
if (_job == null || _job.HasExited)
{
eofCount++;
}
2013-02-27 04:19:05 +00:00
await Task.Delay(100).ConfigureAwait(false);
}
else
{
eofCount = 0;
}
position = fsPosition;
}
}
}
}
}