diff --git a/FastGithub.HttpServer/ApplicationBuilderExtensions.cs b/FastGithub.HttpServer/ApplicationBuilderExtensions.cs index 1262795..7bdc75c 100644 --- a/FastGithub.HttpServer/ApplicationBuilderExtensions.cs +++ b/FastGithub.HttpServer/ApplicationBuilderExtensions.cs @@ -9,6 +9,21 @@ namespace FastGithub /// public static class ApplicationBuilderExtensions { + /// + /// 使用服务头 + /// + /// + /// + /// + public static IApplicationBuilder UseServerHeader(this IApplicationBuilder app, string serverHeader = nameof(FastGithub)) + { + return app.Use(next => async context => + { + await next(context); + context.Response.Headers.Server = serverHeader; + }); + } + /// /// 使用http代理中间件 /// diff --git a/FastGithub/Startup.cs b/FastGithub/Startup.cs index 5dab33d..927655f 100644 --- a/FastGithub/Startup.cs +++ b/FastGithub/Startup.cs @@ -57,14 +57,16 @@ namespace FastGithub var httpProxyPort = app.ApplicationServices.GetRequiredService>().Value.HttpProxyPort; app.MapWhen(context => context.Connection.LocalPort == httpProxyPort, appBuilder => { + appBuilder.UseServerHeader(); appBuilder.UseHttpProxy(); }); app.MapWhen(context => context.Connection.LocalPort != httpProxyPort, appBuilder => { + appBuilder.UseServerHeader(); appBuilder.UseRequestLogging(); appBuilder.UseHttpReverseProxy(); - + appBuilder.UseRouting(); appBuilder.UseEndpoints(endpoint => {