Use middlewares instead of Routing

This commit is contained in:
Claus Vium 2019-02-26 10:23:58 +01:00
parent f3e7bc0573
commit 194da8416b
2 changed files with 79 additions and 82 deletions

View File

@ -657,25 +657,21 @@ namespace Emby.Server.Implementations
ReceiveBufferSize = 0x10000 ReceiveBufferSize = 0x10000
}); });
app.UseRouter(r => app.Use(ExecuteWebsocketHandlerAsync);
{ app.Use(ExecuteHttpHandlerAsync);
// TODO all the verbs, but really MVC...
r.MapGet("/{*localpath}", ExecuteHandler);
r.MapPut("/{*localpath}", ExecuteHandler);
r.MapPost("/{*localpath}", ExecuteHandler);
r.MapDelete("/{*localpath}", ExecuteHandler);
r.MapVerb("HEAD", "/{*localpath}", ExecuteHandler);
});
}) })
.Build(); .Build();
} }
public async Task ExecuteHandler(HttpRequest request, Microsoft.AspNetCore.Http.HttpResponse response, RouteData data) public async Task ExecuteWebsocketHandlerAsync(HttpContext context, Func<Task> next)
{ {
var ctx = request.HttpContext; if (!context.WebSockets.IsWebSocketRequest)
if (ctx.WebSockets.IsWebSocketRequest)
{ {
await ((HttpListenerHost)HttpServer)._websocketlistener.ProcessWebSocketRequest(ctx).ConfigureAwait(false); await next();
return;
}
await ((HttpListenerHost)HttpServer)._websocketlistener.ProcessWebSocketRequest(context).ConfigureAwait(false);
// try // try
// { // {
// var endpoint = ctx.Request.Path.ToString(); // var endpoint = ctx.Request.Path.ToString();
@ -733,9 +729,20 @@ namespace Emby.Server.Implementations
// ctx.Response.StatusCode = 500; // ctx.Response.StatusCode = 500;
// } // }
} }
public async Task ExecuteHttpHandlerAsync(HttpContext context, Func<Task> next)
{
if (context.WebSockets.IsWebSocketRequest)
{
await next();
return;
}
var request = context.Request;
var response = context.Response;
var localPath = context.Request.Path.ToString().TrimStart('/');
var req = new WebSocketSharpRequest(request, response, request.Path, Logger); var req = new WebSocketSharpRequest(request, response, request.Path, Logger);
await ((HttpListenerHost)HttpServer).RequestHandler(req, request.GetDisplayUrl(), request.Host.ToString(), data.Values["localpath"].ToString(), CancellationToken.None).ConfigureAwait(false); await ((HttpListenerHost)HttpServer).RequestHandler(req, request.GetDisplayUrl(), request.Host.ToString(), localPath, CancellationToken.None).ConfigureAwait(false);
} }
protected virtual IHttpClient CreateHttpClient() protected virtual IHttpClient CreateHttpClient()

View File

@ -28,18 +28,8 @@ namespace Emby.Server.Implementations.SocketSharp
public SharpWebSocket(SocketHttpListener.WebSocket socket, ILogger logger) public SharpWebSocket(SocketHttpListener.WebSocket socket, ILogger logger)
{ {
if (socket == null) _logger = logger ?? throw new ArgumentNullException(nameof(logger));
{ WebSocket = socket ?? throw new ArgumentNullException(nameof(socket));
throw new ArgumentNullException(nameof(socket));
}
if (logger == null)
{
throw new ArgumentNullException(nameof(logger));
}
_logger = logger;
WebSocket = socket;
socket.OnMessage += OnSocketMessage; socket.OnMessage += OnSocketMessage;
socket.OnClose += OnSocketClose; socket.OnClose += OnSocketClose;