FastGithub/FastGithub.ReverseProxy/ReverseProxyApplicationBuilderExtensions.cs
2021-07-25 00:20:32 +08:00

35 lines
1.2 KiB
C#

using FastGithub.ReverseProxy;
using Microsoft.AspNetCore.Builder;
using Microsoft.Extensions.DependencyInjection;
namespace FastGithub
{
/// <summary>
/// https反向代理的中间件扩展
/// </summary>
public static class ReverseProxyApplicationBuilderExtensions
{
/// <summary>
/// 使用请求日志中间件
/// </summary>
/// <param name="app"></param>
/// <returns></returns>
public static IApplicationBuilder UseRequestLogging(this IApplicationBuilder app)
{
var middlware = app.ApplicationServices.GetRequiredService<RequestLoggingMilldeware>();
return app.Use(next => context => middlware.InvokeAsync(context, next));
}
/// <summary>
/// 使用https反向代理中间件
/// </summary>
/// <param name="app"></param>
/// <returns></returns>
public static IApplicationBuilder UseHttpsReverseProxy(this IApplicationBuilder app)
{
var middleware = app.ApplicationServices.GetRequiredService<ReverseProxyMiddleware>();
return app.Use(next => context => middleware.InvokeAsync(context, next));
}
}
}