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 =>
{