.NET性能优化-使用SourceGenerator-Logger记录日志

技术分享 2年前 (2022-08-02) 0 999+

前言

在现在许许多多的应用系统中，日志非常关键，它即是排查问题的强力工具，也是程序员居家旅行工作甩锅必备良品。
在团队中编码中，我们都要求对于那些会变更数据的接口、调用第三方的接口记录请求和响应参数，另外在关键的代码路径记录后续可供诊断的日志信息。
如果使用了微软官方的日志记录框架Microsoft.Extensions.Logging，我们通常会像下面代码一样记录日志。
这是我们经常会遇到的日志记录场景，其中会记录一些外部传入的参数。带参数的日志会有频繁的字符串拼接必然会使用更多的内存对GC造成更大的压力。当系统处理的请求越来越多的时候，日志记录就很可能会成为瓶颈。

// 创建日志记录类，分别使用不同的方式来记录日志 var logger = LoggerFactory.Create(l => l.AddConsole()).CreateLogger(typeof(OrderLogger)); var orderLogger = new OrderLogger(logger); var member = new Member("8888","Justin Yu"); orderLogger.LogByStringInterpolation(member, DateTime.Now); orderLogger.LogByStructure(member, DateTime.Now); OrderLogger.LogBySourceGenerator(logger, member, DateTime.Now);  /// <summary>   /// 会员   /// </summary>   /// <param name="MemberId">会员Id</param>   /// <param name="Name">会员名</param>   public record Member(string MemberId, string Name);  /// <summary> /// 订单日志记录类 /// 需要使用Source Generator 所以类型为partial /// </summary> public partial class OrderLogger {     private readonly ILogger _logger;          public OrderLogger(ILogger logger)     {         _logger = logger;     }      /// <summary>     /// 使用字符串插值记录     /// </summary>     public void LogByStringInterpolation(Member member, DateTime now)=>         _logger.LogInformation($"会员[{member}]在[{now:yyyy-MM-dd HH:mm:ss}]充值了一个小目标");      /// <summary>     /// 使用参数化记录     /// </summary>     public void LogByStructure(Member member, DateTime now) =>         _logger.LogInformation("会员[{Member}]在[{Now:yyyy-MM-dd HH:mm:ss}]充值了一个小目标", member, now);      /// <summary>     /// 使用源代码生成     /// </summary>     [LoggerMessage(         EventId = 0,         Level = LogLevel.Information,          Message = "会员[{member}]在[{Now:yyyy-MM-dd HH:mm:ss}]充值了一个小目标")]     public static partial void LogBySourceGenerator(ILogger logger, Member member, DateTime now); }

然后运行一下代码看看日志记录的结果，它们都能正常的输出我们想要的日志，那么究竟性能上有什么差别呢？让我们看看第二节的内容。

性能对比

我们这里实际上是测试这三种方式参数化日志记录的性能，为了避免日志输出到外部Console、File可能影响我们的测试结果，我们自己编写一个什么都不做的ILogger实现。

public class EmptyLogger : ILogger   {       public void Log<TState>(LogLevel logLevel, EventId eventId, TState state, Exception? exception, Func<TState, Exception?, string> formatter)       { }          public bool IsEnabled(LogLevel logLevel) => true;          public IDisposable BeginScope<TState>(TState state) => default;   }

然后使用下面代码进行Benchmark。

[HtmlExporter] [MemoryDiagnoser]   [Orderer(SummaryOrderPolicy.FastestToSlowest)] public class Benchmark   {       private static readonly ILogger Logger = new EmptyLogger();     private readonly OrderLogger _orderLogger = new(Logger);     private readonly Member _user = new("8888","Justin Yu");       private readonly DateTime _now = DateTime.UtcNow;          [Benchmark]       public void LogByStringInterpolation() => _orderLogger.LogByStringInterpolation(_user, _now);          [Benchmark]       public void LogByStructure() => _orderLogger.LogByStructure(_user, _now);          [Benchmark]       public void LogBySourceGenerator() => OrderLogger.LogBySourceGenerator(_logger, _user, _now);   }

跑分结果如下所示，可以看到使用了SourceGenerator的日志记录方式遥遥领先；另外也不要小看那么几十纳秒和Byte的差别，在高并发的系统中就是积少成多会带来不错的性能提升。

那么为什么使用SourceGenerator会有这样的效果？我们来看看SourceGenerator到底生成了什么。

SourceGeneratorSourceGenerator

关于SourceGenerator

SourceGenerator（下文使用SG替代）并不是什么新的东西，在.NET5时就已经支持，不过由于在那时还是比较新的技术，没有太多的应用场景。引用微软文档上的图片，SourceGenerator就是在编译时通过Roslyn来分析源代码然后生成新源代码参与编译，更详细的信息可以在文末找到链接。

它主要是为了在编译时生成代码，减少开发者写一些重复代码所花费的时间，另外也有着比反射和运行时生成代码(Emit)更高的性能和更低的使用门槛。
另外一个方面就是为了Native AOT，比如Dapper、EF等为代表严重依赖运行时代码生成的ORM框架，在Native AOT中其实是不可用的，引入SourceGenerator以后就可以在运行时生成代码，为Native AOT加上这些框架的支持。
目前在.NET框架中已经有Logging、System.Text.Json、Regex等组件已经支持SourceGenerator，都有着较大的性能提成。

SourceGeneratorSourceGenerator

SG为Logger生成了什么？

SG生成的代码是可以输出到文件中的，只需要在项目文件中的<PropertyGroup>内加入这样的一个配置，设置值为True。

<EmitCompilerGeneratedFiles>true</EmitCompilerGeneratedFiles>

然后我们就可以在下面的目录中找到SG为我们生成的代码。路径在项目文件同级目录，根据Debug模式和Release模式如下所示：
.NET性能优化-使用SourceGenerator-Logger记录日志
打开文件夹以后，可以看到有一个*.g.cs结尾的文件，这就是由SG生成的文件。

使用IDE打开这个文件，就可以看到为我们生成的代码是什么样的(如下所示)。可以看到SG为我们填充了LogBySourceGenerator的主体部分，实现高性能的秘密之一就是使用了LoggerMessage.Define，它的所需要的内存分配要少的许多。

*.g.csLogBySourceGeneratorLoggerMessage.Define

// <auto-generated/> #nullable enable partial class OrderLogger  {     [global::System.CodeDom.Compiler.GeneratedCodeAttribute("Microsoft.Extensions.Logging.Generators", "6.0.5.2210")]     private static readonly global::System.Action<global::Microsoft.Extensions.Logging.ILogger, global::Member, global::System.DateTime, global::System.Exception?> __LogBySourceGeneratorCallback =         global::Microsoft.Extensions.Logging.LoggerMessage.Define<global::Member, global::System.DateTime>(global::Microsoft.Extensions.Logging.LogLevel.Information, new global::Microsoft.Extensions.Logging.EventId(0, nameof(LogBySourceGenerator)), "会员[{member}]在[{Now:yyyy-MM-dd HH:mm:ss}]充值了一个小目标", new global::Microsoft.Extensions.Logging.LogDefineOptions() { SkipEnabledCheck = true });      [global::System.CodeDom.Compiler.GeneratedCodeAttribute("Microsoft.Extensions.Logging.Generators", "6.0.5.2210")]     public static partial void LogBySourceGenerator(global::Microsoft.Extensions.Logging.ILogger logger, global::Member member, global::System.DateTime now)     {         if (logger.IsEnabled(global::Microsoft.Extensions.Logging.LogLevel.Information))         {             __LogBySourceGeneratorCallback(logger, member, now, null);         }     } }

而填充的规则是由上面代码中的特性决定的，如下摘抄出来的代码。

[LoggerMessage(     EventId = 0,     Level = LogLevel.Information,      Message = "会员[{member}]在[{Now:yyyy-MM-dd HH:mm:ss}]充值了一个小目标")] public static partial void LogBySourceGenerator(ILogger logger, Member member, DateTime now);

至于SG是如何生成代码的不在本文的讨论范围，大家有兴趣的可以查看下方的源码链接。
https://github.com/dotnet/runtime/tree/main/src/libraries/Microsoft.Extensions.Logging.Abstractions/gen

编译时警告

这是SG另外一个很不错的功能，我们在软件开发过程中经常会犯一些错误，导致不能正常的使用框架的功能或者产生一些BUG。比如：
按照Logger的约定来说，为了方便SG生成代码，我们需要将类和方法名变为partial，假设我忘记了这个，写了如下的代码：

/// <summary>   /// 源代码生成 - 忘记写partial   /// </summary>   [LoggerMessage(       EventId = 0,       Level = LogLevel.Information,    Message = "会员[{member}]在[{Now:yyyy-MM-dd HH:mm:ss}]充值了一个小目标")]   public static void LogBySourceGenerator(ILogger logger, Member member, DateTime now);

就会有下面这样的错误，提醒你必须加partial关键字。

再如，在参数化的模版里面有merber和now两个参数，我们编写代码的时候像下方一样少传了now参数。

merbernownow

/// <summary>   /// 源代码生成   /// </summary>   [LoggerMessage(       EventId = 0,       Level = LogLevel.Information,    Message = "会员[{member}]在[{Now:yyyy-MM-dd HH:mm:ss}]充值了一个小目标")]   public static partial void LogBySourceGenerator(ILogger logger, Member member);

那么SG构建时便可以提醒你，模版中的Now没有听对应的参数。

这功能会非常有用，能节省很多问题排查的时间。