From e2dcddc5ac43846baea0f9b1a0fc62844dd9ee1d Mon Sep 17 00:00:00 2001 From: Luke Pulverenti Date: Sat, 23 Mar 2013 22:45:00 -0400 Subject: made compression and caching available to plugin api endpoints --- .../HttpServer/RangeRequestWriter.cs | 190 ++++++++++++--------- 1 file changed, 108 insertions(+), 82 deletions(-) (limited to 'MediaBrowser.Server.Implementations/HttpServer/RangeRequestWriter.cs') diff --git a/MediaBrowser.Server.Implementations/HttpServer/RangeRequestWriter.cs b/MediaBrowser.Server.Implementations/HttpServer/RangeRequestWriter.cs index b61e05d0b..a355a2db5 100644 --- a/MediaBrowser.Server.Implementations/HttpServer/RangeRequestWriter.cs +++ b/MediaBrowser.Server.Implementations/HttpServer/RangeRequestWriter.cs @@ -1,6 +1,8 @@ using ServiceStack.Service; +using ServiceStack.ServiceHost; using System; using System.Collections.Generic; +using System.Globalization; using System.IO; using System.Linq; using System.Net; @@ -8,30 +10,105 @@ using System.Threading.Tasks; namespace MediaBrowser.Server.Implementations.HttpServer { - public class RangeRequestWriter : IStreamWriter + public class RangeRequestWriter : IStreamWriter, IHttpResult { /// /// Gets or sets the source stream. /// /// The source stream. private Stream SourceStream { get; set; } - private HttpListenerResponse Response { get; set; } private string RangeHeader { get; set; } private bool IsHeadRequest { get; set; } + private long RangeStart { get; set; } + private long RangeEnd { get; set; } + private long RangeLength { get; set; } + private long TotalContentLength { get; set; } + + /// + /// The _options + /// + private readonly Dictionary _options = new Dictionary(); + + /// + /// The us culture + /// + private static readonly CultureInfo UsCulture = new CultureInfo("en-US"); + + /// + /// Additional HTTP Headers + /// + /// The headers. + public Dictionary Headers + { + get { return _options; } + } + + /// + /// Gets the options. + /// + /// The options. + public IDictionary Options + { + get { return Headers; } + } + /// /// Initializes a new instance of the class. /// /// The range header. - /// The response. /// The source. + /// Type of the content. /// if set to true [is head request]. - public RangeRequestWriter(string rangeHeader, HttpListenerResponse response, Stream source, bool isHeadRequest) + public RangeRequestWriter(string rangeHeader, Stream source, string contentType, bool isHeadRequest) { + if (string.IsNullOrEmpty(contentType)) + { + throw new ArgumentNullException("contentType"); + } + RangeHeader = rangeHeader; - Response = response; SourceStream = source; IsHeadRequest = isHeadRequest; + + ContentType = contentType; + Options["Content-Type"] = contentType; + Options["Accept-Ranges"] = "bytes"; + StatusCode = HttpStatusCode.PartialContent; + + SetRangeValues(); + } + + /// + /// Sets the range values. + /// + private void SetRangeValues() + { + var requestedRange = RequestedRanges.First(); + + TotalContentLength = SourceStream.Length; + + // If the requested range is "0-", we can optimize by just doing a stream copy + if (!requestedRange.Value.HasValue) + { + RangeEnd = TotalContentLength - 1; + } + else + { + RangeEnd = requestedRange.Value.Value; + } + + RangeStart = requestedRange.Key; + RangeLength = 1 + RangeEnd - RangeStart; + + // Content-Length is the length of what we're serving, not the original content + Options["Content-Length"] = RangeLength.ToString(UsCulture); + Options["Content-Range"] = string.Format("bytes {0}-{1}/{2}", RangeStart, RangeEnd, TotalContentLength); + + if (RangeStart > 0) + { + SourceStream.Position = RangeStart; + } } /// @@ -42,7 +119,7 @@ namespace MediaBrowser.Server.Implementations.HttpServer /// Gets the requested ranges. /// /// The requested ranges. - protected IEnumerable> RequestedRanges + protected List> RequestedRanges { get { @@ -83,9 +160,6 @@ namespace MediaBrowser.Server.Implementations.HttpServer /// The response stream. public void WriteTo(Stream responseStream) { - Response.Headers["Accept-Ranges"] = "bytes"; - Response.StatusCode = 206; - var task = WriteToAsync(responseStream); Task.WaitAll(task); @@ -98,94 +172,46 @@ namespace MediaBrowser.Server.Implementations.HttpServer /// Task. private async Task WriteToAsync(Stream responseStream) { - using (var source = SourceStream) + // Headers only + if (IsHeadRequest) { - var requestedRange = RequestedRanges.First(); - - var totalLength = SourceStream.Length; + return; + } + using (var source = SourceStream) + { // If the requested range is "0-", we can optimize by just doing a stream copy - if (!requestedRange.Value.HasValue) + if (RangeEnd == TotalContentLength - 1) { - await ServeCompleteRangeRequest(source, requestedRange, responseStream, totalLength).ConfigureAwait(false); + await source.CopyToAsync(responseStream).ConfigureAwait(false); } + else + { + // Read the bytes we need + var buffer = new byte[Convert.ToInt32(RangeLength)]; + await source.ReadAsync(buffer, 0, buffer.Length).ConfigureAwait(false); - // This will have to buffer a portion of the content into memory - await ServePartialRangeRequest(source, requestedRange.Key, requestedRange.Value.Value, responseStream, totalLength).ConfigureAwait(false); + await responseStream.WriteAsync(buffer, 0, Convert.ToInt32(RangeLength)).ConfigureAwait(false); + } } } - /// - /// Handles a range request of "bytes=0-" - /// This will serve the complete content and add the content-range header - /// - /// The source stream. - /// The requested range. - /// The response stream. - /// Total length of the content. - /// Task. - private Task ServeCompleteRangeRequest(Stream sourceStream, KeyValuePair requestedRange, Stream responseStream, long totalContentLength) - { - var rangeStart = requestedRange.Key; - var rangeEnd = totalContentLength - 1; - var rangeLength = 1 + rangeEnd - rangeStart; + public string ContentType { get; set; } - // Content-Length is the length of what we're serving, not the original content - Response.ContentLength64 = rangeLength; - Response.Headers["Content-Range"] = string.Format("bytes {0}-{1}/{2}", rangeStart, rangeEnd, totalContentLength); + public IRequestContext RequestContext { get; set; } - // Headers only - if (IsHeadRequest) - { - return Task.FromResult(true); - } + public object Response { get; set; } - if (rangeStart > 0) - { - sourceStream.Position = rangeStart; - } + public IContentTypeWriter ResponseFilter { get; set; } - return sourceStream.CopyToAsync(responseStream); - } + public int Status { get; set; } - /// - /// Serves a partial range request - /// - /// The source stream. - /// The range start. - /// The range end. - /// The response stream. - /// Total length of the content. - /// Task. - private async Task ServePartialRangeRequest(Stream sourceStream, long rangeStart, long rangeEnd, Stream responseStream, long totalContentLength) + public HttpStatusCode StatusCode { - var rangeLength = 1 + rangeEnd - rangeStart; - - // Content-Length is the length of what we're serving, not the original content - Response.ContentLength64 = rangeLength; - Response.Headers["Content-Range"] = string.Format("bytes {0}-{1}/{2}", rangeStart, rangeEnd, totalContentLength); - - // Headers only - if (IsHeadRequest) - { - return; - } - - sourceStream.Position = rangeStart; - - // Fast track to just copy the stream to the end - if (rangeEnd == totalContentLength - 1) - { - await sourceStream.CopyToAsync(responseStream).ConfigureAwait(false); - } - else - { - // Read the bytes we need - var buffer = new byte[Convert.ToInt32(rangeLength)]; - await sourceStream.ReadAsync(buffer, 0, buffer.Length).ConfigureAwait(false); - - await responseStream.WriteAsync(buffer, 0, Convert.ToInt32(rangeLength)).ConfigureAwait(false); - } + get { return (HttpStatusCode)Status; } + set { Status = (int)value; } } + + public string StatusDescription { get; set; } } } -- cgit v1.2.3