Merge pull request #332 from liuhll/fix-hosting-env

KestrelHost通过环境变量ASPNETCORE_ENVIRONMENT指定开发环境
fanliang11 · Jul 12, 2019 · 7876ac6 · 7876ac6
2 parents 9a0d2e1 + 78d7fb6
commit 7876ac6
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 0 deletions.
diff --git a/src/Surging.Core/Surging.Core.CPlatform/Configurations/RuntimeEnvironment.cs 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
@@ -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
@@ -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);