69 lines
2.3 KiB
C#
69 lines
2.3 KiB
C#
// <copyright>
|
|
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// </copyright>
|
|
|
|
namespace System.ServiceModel.Channels
|
|
{
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Net;
|
|
using System.Net.Http;
|
|
using System.Runtime;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
|
|
abstract class WebSocketConnectionHandler : HttpMessageHandler
|
|
{
|
|
protected internal virtual HttpResponseMessage AcceptWebSocket(HttpRequestMessage request, CancellationToken cancellationToken)
|
|
{
|
|
if (this.AcceptWebSocket(request))
|
|
{
|
|
return GetWebSocketAcceptedResponseMessage(request);
|
|
}
|
|
else
|
|
{
|
|
return GetUpgradeRequiredResponseMessage(request);
|
|
}
|
|
}
|
|
|
|
protected internal virtual bool AcceptWebSocket(HttpRequestMessage request)
|
|
{
|
|
return true;
|
|
}
|
|
|
|
protected static HttpResponseMessage GetUpgradeRequiredResponseMessage(HttpRequestMessage request)
|
|
{
|
|
HttpResponseMessage response = new HttpResponseMessage(HttpStatusCode.UpgradeRequired);
|
|
response.RequestMessage = request;
|
|
return response;
|
|
}
|
|
|
|
protected static HttpResponseMessage GetBadRequestResponseMessage(HttpRequestMessage request)
|
|
{
|
|
HttpResponseMessage response = new HttpResponseMessage(HttpStatusCode.BadRequest);
|
|
response.RequestMessage = request;
|
|
return response;
|
|
}
|
|
|
|
protected static HttpResponseMessage GetWebSocketAcceptedResponseMessage(HttpRequestMessage request)
|
|
{
|
|
HttpResponseMessage message = new HttpResponseMessage(HttpStatusCode.SwitchingProtocols);
|
|
message.RequestMessage = request;
|
|
return message;
|
|
}
|
|
|
|
protected override Task<HttpResponseMessage> SendAsync(HttpRequestMessage request, CancellationToken cancellationToken)
|
|
{
|
|
if (request == null)
|
|
{
|
|
throw FxTrace.Exception.ArgumentNull("request");
|
|
}
|
|
|
|
return Task.Factory.StartNew(
|
|
() => { return this.AcceptWebSocket(request, cancellationToken); },
|
|
cancellationToken);
|
|
}
|
|
}
|
|
}
|