public override void Write(IChannelHandlerContext ctx, object msg, IPromise promise)
{
if (msg is IHttpResponse response)
{
var headers = response.Headers;
if (WebSocketExtensionUtil.IsWebsocketUpgrade(headers))
{
if (_validExtensions is object)
{
string headerValue = null;
if (headers.TryGet(HttpHeaderNames.SecWebsocketExtensions, out ICharSequence value))
{
headerValue = value?.ToString();
}
for (int i = 0; i < _validExtensions.Count; i++)
{
WebSocketExtensionData extensionData = _validExtensions[i].NewReponseData();
headerValue = WebSocketExtensionUtil.AppendExtension(headerValue,
extensionData.Name, extensionData.Parameters);
}
_ = promise.Task.ContinueWith(s_switchWebSocketExtensionHandlerAction, Tuple.Create(ctx, _validExtensions), TaskContinuationOptions.ExecuteSynchronously);
if (headerValue is object)
{
_ = headers.Set(HttpHeaderNames.SecWebsocketExtensions, headerValue);
}
}
}
_ = promise.Task.ContinueWith(s_removeWebSocketExtensionHandlerAction, Tuple.Create(ctx, this), TaskContinuationOptions.ExecuteSynchronously);
}
base.Write(ctx, msg, promise);
}
public override Task WriteAsync(IChannelHandlerContext ctx, object msg)
{
Action <Task> continuationAction = null;
if (msg is IHttpResponse response &&
WebSocketExtensionUtil.IsWebsocketUpgrade(response.Headers) &&
this.validExtensions != null)
{
string headerValue = null;
if (response.Headers.TryGet(HttpHeaderNames.SecWebsocketExtensions, out ICharSequence value))
{
headerValue = value?.ToString();
}
foreach (IWebSocketServerExtension extension in this.validExtensions)
{
WebSocketExtensionData extensionData = extension.NewReponseData();
headerValue = WebSocketExtensionUtil.AppendExtension(headerValue,
extensionData.Name, extensionData.Parameters);
}
continuationAction = promise =>
{
if (promise.Status == TaskStatus.RanToCompletion)
{
foreach (IWebSocketServerExtension extension in this.validExtensions)
{
WebSocketExtensionDecoder decoder = extension.NewExtensionDecoder();
WebSocketExtensionEncoder encoder = extension.NewExtensionEncoder();
ctx.Channel.Pipeline.AddAfter(ctx.Name, decoder.GetType().Name, decoder);
ctx.Channel.Pipeline.AddAfter(ctx.Name, encoder.GetType().Name, encoder);
}
}
ctx.Channel.Pipeline.Remove(ctx.Name);
};
if (headerValue != null)
{
response.Headers.Set(HttpHeaderNames.SecWebsocketExtensions, headerValue);
}
}
return(continuationAction == null
? base.WriteAsync(ctx, msg)
: base.WriteAsync(ctx, msg)
.ContinueWith(continuationAction, TaskContinuationOptions.ExecuteSynchronously));
}
public override void Write(IChannelHandlerContext ctx, object msg, IPromise promise)
{
if (msg is IHttpRequest request && WebSocketExtensionUtil.IsWebsocketUpgrade(request.Headers))
{
string headerValue = null;
if (request.Headers.TryGet(HttpHeaderNames.SecWebsocketExtensions, out ICharSequence value))
{
headerValue = value.ToString();
}
foreach (IWebSocketClientExtensionHandshaker extensionHandshaker in this.extensionHandshakers)
{
WebSocketExtensionData extensionData = extensionHandshaker.NewRequestData();
headerValue = WebSocketExtensionUtil.AppendExtension(headerValue,
extensionData.Name, extensionData.Parameters);
}
_ = request.Headers.Set(HttpHeaderNames.SecWebsocketExtensions, headerValue);
}
base.Write(ctx, msg, promise);
}
