From e45cc4d28de0340cf594b35890c526408ba5dc79 Mon Sep 17 00:00:00 2001
From: xRain <xrain@simcu.com>
Date: Fri, 5 Aug 2022 02:20:27 +0800
Subject: [PATCH] fix logger

---
 SimApiExtensions.cs | 16 +++++++++-------
 1 file changed, 9 insertions(+), 7 deletions(-)

diff --git a/SimApiExtensions.cs b/SimApiExtensions.cs
index 24023d2..b1c7b6e 100644
--- a/SimApiExtensions.cs
+++ b/SimApiExtensions.cs
@@ -22,7 +22,14 @@ namespace SimApi
         {
             var simApiOptions = new SimApiOptions();
             options?.Invoke(simApiOptions);
-
+            if (simApiOptions.EnableLogger)
+            {
+                builder.AddLogging(logger =>
+                {
+                    logger.ClearProviders();
+                    logger.AddProvider(new SimApiLoggerProvider());
+                });
+            }
             // 是否使用 AUTH
             if (simApiOptions.EnableSimApiAuth)
             {
@@ -190,13 +197,8 @@ namespace SimApi
         public static IApplicationBuilder UseSimApi(this IApplicationBuilder builder)
         {
             var options = builder.ApplicationServices.GetRequiredService<SimApiOptions>();
-            if (options.EnableLogger)
-            {
-                builder.ApplicationServices.GetRequiredService<ILoggerFactory>()
-                    .AddProvider(new SimApiLoggerProvider());
-            }
 
-            var logger = builder.ApplicationServices.GetRequiredService<ILogger<SimApiLogger>>();
+            var logger = builder.ApplicationServices.GetRequiredService<ILogger>();
 
             if (options.EnableForwardHeaders)
             {