From 78d7fb6f92a722eb861cb5d8068127edc3f88187 Mon Sep 17 00:00:00 2001
From: liuhll <1029765111@qq.com>
Date: Fri, 12 Jul 2019 13:44:46 +0800
Subject: [PATCH] =?UTF-8?q?KestrelHost=E9=80=9A=E8=BF=87=E7=8E=AF=E5=A2=83?=
 =?UTF-8?q?=E5=8F=98=E9=87=8FASPNETCORE=5FENVIRONMENT=E6=8C=87=E5=AE=9A?=
 =?UTF-8?q?=E5=BC=80=E5=8F=91=E7=8E=AF=E5=A2=83?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../Configurations/RuntimeEnvironment.cs        | 17 +++++++++++++++++
 .../Configurations/SurgingServerOptions.cs      |  2 ++
 .../ServiceHostBuilderExtensions.cs             |  1 +
 3 files changed, 20 insertions(+)
 create mode 100644 src/Surging.Core/Surging.Core.CPlatform/Configurations/RuntimeEnvironment.cs

diff --git a/src/Surging.Core/Surging.Core.CPlatform/Configurations/RuntimeEnvironment.cs b/src/Surging.Core/Surging.Core.CPlatform/Configurations/RuntimeEnvironment.cs
new file mode 100644
index 000000000..fdc85ad79
--- /dev/null
+++ b/src/Surging.Core/Surging.Core.CPlatform/Configurations/RuntimeEnvironment.cs
@@ -0,0 +1,17 @@
+using System;
+using System.Collections.Generic;
+using System.Text;
+
+namespace Surging.Core.CPlatform.Configurations
+{
+    public enum RuntimeEnvironment
+    {
+        Development = 0,
+
+        Test,
+
+        Staging,
+
+        Production,
+    }
+}
diff --git a/src/Surging.Core/Surging.Core.CPlatform/Configurations/SurgingServerOptions.cs b/src/Surging.Core/Surging.Core.CPlatform/Configurations/SurgingServerOptions.cs
index 177fd5750..2c96ffd77 100644
--- a/src/Surging.Core/Surging.Core.CPlatform/Configurations/SurgingServerOptions.cs
+++ b/src/Surging.Core/Surging.Core.CPlatform/Configurations/SurgingServerOptions.cs
@@ -47,5 +47,7 @@ public  partial class SurgingServerOptions: ServiceCommand
         public string NotRelatedAssemblyFiles { get; set; }
 
         public string RelatedAssemblyFiles { get; set; } = "";
+
+        public RuntimeEnvironment Environment { get; set; } = RuntimeEnvironment.Production;
     }
 }
diff --git a/src/Surging.Core/Surging.Core.CPlatform/ServiceHostBuilderExtensions.cs b/src/Surging.Core/Surging.Core.CPlatform/ServiceHostBuilderExtensions.cs
index 4d99e80c9..9663cf61e 100644
--- a/src/Surging.Core/Surging.Core.CPlatform/ServiceHostBuilderExtensions.cs
+++ b/src/Surging.Core/Surging.Core.CPlatform/ServiceHostBuilderExtensions.cs
@@ -29,6 +29,7 @@ public static IServiceHostBuilder UseServer(this IServiceHostBuilder hostBuilder
         {
             return hostBuilder.MapServices(async mapper =>
             {
+                Environment.SetEnvironmentVariable("ASPNETCORE_ENVIRONMENT", AppConfig.ServerOptions.Environment.ToString());
                 BuildServiceEngine(mapper);
                 await mapper.Resolve<IServiceCommandManager>().SetServiceCommandsAsync();
                 string serviceToken = mapper.Resolve<IServiceTokenGenerator>().GeneratorToken(token);