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

151 lines
4.7 KiB
C#
Raw Normal View History

2014-09-03 02:30:24 +00:00
using System.Threading;
2014-08-17 05:38:13 +00:00
using MediaBrowser.Common.IO;
2013-02-27 04:19:05 +00:00
using MediaBrowser.Model.Logging;
2013-12-07 15:52:38 +00:00
using ServiceStack.Web;
2014-09-03 02:30:24 +00:00
using System;
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-08-17 05:38:13 +00:00
catch (Exception ex)
2013-02-27 04:19:05 +00:00
{
2014-08-17 05:38:13 +00:00
Logger.ErrorException("Error streaming media. The client has most likely disconnected or transcoding has failed.", ex);
2013-03-11 04:04:08 +00:00
throw;
2013-02-27 04:19:05 +00:00
}
finally
{
2014-09-03 02:30:24 +00:00
if (_job != null)
{
ApiEntryPoint.Instance.OnTranscodeEndRequest(_job);
}
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-09-03 02:30:24 +00:00
private long _bytesWritten = 0;
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)
{
2014-09-03 02:30:24 +00:00
await CopyToAsyncInternal(fs, outputStream, 81920, CancellationToken.None).ConfigureAwait(false);
2013-02-27 04:19:05 +00:00
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;
}
}
}
2014-09-03 02:30:24 +00:00
private async Task CopyToAsyncInternal(Stream source, Stream destination, int bufferSize, CancellationToken cancellationToken)
{
byte[] array = new byte[bufferSize];
int count;
while ((count = await source.ReadAsync(array, 0, array.Length, cancellationToken).ConfigureAwait(false)) != 0)
{
await destination.WriteAsync(array, 0, count, cancellationToken).ConfigureAwait(false);
_bytesWritten += count;
if (_job != null)
{
_job.BytesDownloaded = Math.Max(_job.BytesDownloaded ?? _bytesWritten, _bytesWritten);
}
}
}
2013-02-27 04:19:05 +00:00
}
}