Files
csharp/src/KubernetesClient/WatcherDelegatingHandler.cs

69 lines
2.4 KiB
C#
Raw Normal View History

2017-10-17 03:59:30 +08:00
using System.IO;
using System.Linq;
2017-10-17 03:59:30 +08:00
using System.Net;
using System.Net.Http;
using System.Threading;
using System.Threading.Tasks;
using Microsoft.AspNetCore.WebUtilities;
2017-10-17 03:59:30 +08:00
namespace k8s
{
2017-10-24 04:36:38 +08:00
/// <summary>
/// This HttpDelegatingHandler is to rewrite the response and return first line to autorest client
/// then use WatchExt to create a watch object which interact with the replaced http response to get watch works.
/// </summary>
2017-10-17 03:59:30 +08:00
internal class WatcherDelegatingHandler : DelegatingHandler
{
protected override async Task<HttpResponseMessage> SendAsync(HttpRequestMessage request,
CancellationToken cancellationToken)
{
var originResponse = await base.SendAsync(request, cancellationToken);
if (originResponse.IsSuccessStatusCode)
{
var query = QueryHelpers.ParseQuery(request.RequestUri.Query);
if (query.TryGetValue("watch", out var values) && values.Any(v => v == "true"))
2017-10-17 03:59:30 +08:00
{
originResponse.Content = new LineSeparatedHttpContent(originResponse.Content);
}
}
return originResponse;
}
internal class LineSeparatedHttpContent : HttpContent
{
private readonly HttpContent _originContent;
private Stream _originStream;
public LineSeparatedHttpContent(HttpContent originContent)
{
_originContent = originContent;
}
internal StreamReader StreamReader { get; private set; }
protected override async Task SerializeToStreamAsync(Stream stream, TransportContext context)
{
_originStream = await _originContent.ReadAsStreamAsync();
StreamReader = new StreamReader(_originStream);
var firstLine = await StreamReader.ReadLineAsync();
var writer = new StreamWriter(stream);
// using (writer) // leave open
{
await writer.WriteAsync(firstLine);
await writer.FlushAsync();
}
}
protected override bool TryComputeLength(out long length)
{
length = 0;
return false;
}
}
}
}