From 388db6f061536e6050cdd88ee9eae72f4ebc11c8 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=99=88=E5=9B=BD=E4=BC=9F?= <366193849@qq.com>
Date: Wed, 3 Nov 2021 13:44:01 +0800
Subject: [PATCH] UseServerHeader

---
 .../ApplicationBuilderExtensions.cs               | 15 +++++++++++++++
 FastGithub/Startup.cs                             |  4 +++-
 2 files changed, 18 insertions(+), 1 deletion(-)

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
     /// </summary>
     public static class ApplicationBuilderExtensions
     {
+        /// <summary>
+        /// 使用服务头
+        /// </summary>
+        /// <param name="app"></param>
+        /// <param name="serverHeader"></param>
+        /// <returns></returns>
+        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;
+            });
+        }
+
         /// <summary>
         /// 使用http代理中间件
         /// </summary>
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<IOptions<FastGithubOptions>>().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 =>
                 {