CachedResponseStream.cs

Go to the documentation of this file.

using System;
using System.IO;
using System.Net.Http;
using System.Threading.Tasks;
 
namespace Tgstation.Server.Common.Http
{
    public sealed class CachedResponseStream : Stream
    {
        readonly HttpContent responseContent;
 
        readonly Stream responseStream;
 
        public static async ValueTask<CachedResponseStream> Create(HttpResponseMessage response)
        {
            if (response == null)
                throw new ArgumentNullException(nameof(response));
 
            using (response)
            {
                var content = response.Content;
                response.Content = null;
                try
                {
                    // don't cry about the missing CancellationToken overload: https://github.com/dotnet/runtime/issues/916
                    var responseStream = await content.ReadAsStreamAsync().ConfigureAwait(false);
                    return new CachedResponseStream(content, responseStream);
                }
                catch
                {
                    content.Dispose();
                    throw;
                }
            }
        }
 
        public override bool CanRead => responseStream.CanRead;
 
        public override bool CanSeek => responseStream.CanSeek;
 
        public override bool CanWrite => responseStream.CanWrite;
 
        public override long Length => responseStream.Length;
 
        public override long Position
        {
            get => responseStream.Position;
            set => responseStream.Position = value;
        }
 
        public override void Flush() => responseStream.Flush();
 
        public override int Read(byte[] buffer, int offset, int count) => responseStream.Read(buffer, offset, count);
 
        public override long Seek(long offset, SeekOrigin origin) => responseStream.Seek(offset, origin);
 
        public override void SetLength(long value) => responseStream.SetLength(value);
 
        public override void Write(byte[] buffer, int offset, int count) => responseStream.Write(buffer, offset, count);
 
        protected override void Dispose(bool disposing)
        {
            base.Dispose(disposing);
            if (!disposing)
                return;
 
            responseContent.Dispose();
            responseStream.Dispose();
        }
 
        CachedResponseStream(HttpContent responseContent, Stream responseStream)
        {
            this.responseContent = responseContent;
            this.responseStream = responseStream;
        }
    }
}