NLogProviderOptions with support for RegisterServiceProvider

NLog · Oct 2, 2021 · c964fd2 · c964fd2
1 parent 039f13a
commit c964fd2
Show file tree

Hide file tree

Showing 5 changed files with 42 additions and 6 deletions.
diff --git a/src/NLog.Extensions.Hosting/Extensions/ConfigureExtensions.cs b/src/NLog.Extensions.Hosting/Extensions/ConfigureExtensions.cs
@@ -49,7 +49,14 @@ private static void AddNLogLoggerProvider(IServiceCollection services, IConfigur
         private static NLogLoggerProvider CreateNLogLoggerProvider(IServiceProvider serviceProvider, IConfiguration configuration, NLogProviderOptions options)
         {
             NLogLoggerProvider provider = new NLogLoggerProvider(options);
+
             configuration = SetupConfiguration(serviceProvider, configuration);
+
+            if (serviceProvider != null && provider.Options.RegisterServiceProvider)
+            {
+                provider.LogFactory.ServiceRepository.RegisterService(typeof(IServiceProvider), serviceProvider);
+            }
+
             if (configuration != null)
             {
                 provider.Configure(configuration.GetSection("Logging:NLog"));

diff --git a/src/NLog.Extensions.Logging/Extensions/ConfigureExtensions.cs b/src/NLog.Extensions.Logging/Extensions/ConfigureExtensions.cs
@@ -249,7 +249,14 @@ private static NLogLoggerProvider CreateNLogLoggerProvider(IServiceProvider serv
         private static NLogLoggerProvider CreateNLogLoggerProvider(IServiceProvider serviceProvider, IConfiguration configuration, NLogProviderOptions options, LogFactory logFactory)
         {
             NLogLoggerProvider provider = new NLogLoggerProvider(options ?? NLogProviderOptions.Default, logFactory ?? LogManager.LogFactory);
+
             configuration = SetupConfiguration(serviceProvider, configuration);
+
+            if (serviceProvider != null && provider.Options.RegisterServiceProvider)
+            {
+                provider.LogFactory.ServiceRepository.RegisterService(typeof(IServiceProvider), serviceProvider);
+            }
+
             if (configuration != null)
             {
                 provider.Configure(configuration.GetSection("Logging:NLog"));

diff --git a/src/NLog.Extensions.Logging/LayoutRenderers/ConfigSettingLayoutRenderer.cs b/src/NLog.Extensions.Logging/LayoutRenderers/ConfigSettingLayoutRenderer.cs
@@ -33,6 +33,8 @@ namespace NLog.Extensions.Logging
     [ThreadSafe]
     public class ConfigSettingLayoutRenderer : LayoutRenderer
     {
+        private IConfiguration _serviceConfiguration;
+
         /// <summary>
         /// Global Configuration Container
         /// </summary>
@@ -66,14 +68,29 @@ public string Item
         ///</summary>
         public string Default { get; set; }
 
+        /// <inheritdoc/>
+        protected override void InitializeLayoutRenderer()
+        {
+            try
+            {
+                _serviceConfiguration = ResolveService<IConfiguration>();
+            }
+            catch (NLogDependencyResolveException ex)
+            {
+                InternalLogger.Debug("ConfigSetting - IConfiguration could not be resolved, so fallback to DefaultConfiguration: {0}", ex.Message);
+            }
+
+            base.InitializeLayoutRenderer();
+        }
+
         /// <inheritdoc/>
         protected override void Append(StringBuilder builder, LogEventInfo logEvent)
         {
             if (string.IsNullOrEmpty(_itemLookup))
                 return;
 
             string value = null;
-            var configurationRoot = DefaultConfiguration;
+            var configurationRoot = _serviceConfiguration ?? DefaultConfiguration;
             if (configurationRoot != null)
             {
                 value = configurationRoot[_itemLookup];

diff --git a/src/NLog.Extensions.Logging/Logging/NLogProviderOptions.cs b/src/NLog.Extensions.Logging/Logging/NLogProviderOptions.cs
@@ -13,11 +13,8 @@ public class NLogProviderOptions
         public string EventIdSeparator { get; set; } = "_";
 
         /// <summary>
-        /// Skip allocation of <see cref="LogEventInfo.Properties" />-dictionary
+        /// Skip creating "EventId_Id" and "EventId_Name" as <see cref="LogEventInfo.Properties" /> when <c>default(EventId)</c>
         /// </summary>
-        /// <remarks>
-        /// using
-        ///     <c>default(EventId)</c></remarks>
         public bool IgnoreEmptyEventId { get; set; } = true;
 
         /// <summary>
@@ -79,7 +76,7 @@ public class NLogProviderOptions
         public bool RemoveLoggerFactoryFilter { get; set; } = true;
 
         /// <summary>
-        /// Replace Microsoft LoggerFactory with a pure <see cref="NLogLoggerFactory" />, and disables Microsoft Filter Logic and multiple LoggingProvider support.
+        /// Replace Microsoft LoggerFactory with a pure <see cref="NLogLoggerFactory" />, and disables Microsoft Filter Logic and removes other LoggingProviders.
         /// </summary>
         /// <remarks>This option affects the building of service configuration, so assigning it from appsettings.json has no effect (loaded after).</remarks>
         public bool ReplaceLoggerFactory { get; set; }
@@ -98,6 +95,11 @@ public class NLogProviderOptions
         /// </remarks>
         public bool CaptureEntireEventId { get; set; }
 
+        /// <summary>
+        /// Enable NLog Targets and Layouts to perform dependency lookup using the Microsoft Dependency Injection IServiceProvider
+        /// </summary>
+        public bool RegisterServiceProvider { get; set; } = true;
+
         /// <summary>Initializes a new instance NLogProviderOptions with default values.</summary>
         public NLogProviderOptions()
         {

diff --git a/test/NLog.Extensions.Hosting.Tests/ExtensionMethodTests.cs b/test/NLog.Extensions.Hosting.Tests/ExtensionMethodTests.cs
@@ -109,6 +109,9 @@ public void UseNLog_LoadConfigurationFromSection()
 
             var loggerFact = host.Services.GetService<ILoggerFactory>();
             var logger = loggerFact.CreateLogger("logger1");
+
+            ConfigSettingLayoutRenderer.DefaultConfiguration = null;    // See dependency resolving is working
+
             logger.LogError("error1");
 
             var loggerProvider = host.Services.GetService<ILoggerProvider>() as NLogLoggerProvider;