.NET中的代码生成技术:Source Generators入门与实战

Source Generators是.NET 5引入的编译时代码生成技术,通过实现ISourceGenerator接口在编译阶段分析语法树并生成C#代码,1. 具备编译时执行、IDE友好和类型安全优势,2. 可用于自动生成ToString、INotifyPropertyChanged、DTO映射等重复代码,3. 结合Roslyn API收集语法节点并生成源文件,4. 支持通过EmitCompilerGeneratedFiles调试生成结果,提升性能且无运行时开销。

.NET 中的 Source Generators 是一种强大的编译时代码生成技术,它允许你在编译过程中自动生成 C# 代码,从而减少重复劳动、提升性能并增强类型安全。与传统的运行时反射不同,Source Generators 在编译阶段工作,生成的代码会直接嵌入到最终程序集中,不会带来额外的运行时开销。

什么是 Source Generators?

Source Generators 是 .NET 5 引入的一项功能,属于 Compiler-as-a-Service 特性的一部分。它通过实现 ISourceGenerator 接口,在 Roslyn 编译器解析语法树时介入,分析现有代码并注入新的 C# 源文件。

它的核心优势包括:

  • 编译时执行:不依赖运行时反射,避免性能损耗。
  • IDE 友好:生成的代码可在 Visual Studio 或其他编辑器中查看和调试。
  • 类型安全:生成的代码参与完整编译流程,错误能在编译期暴露。

如何创建一个简单的 Source Generator

下面通过一个实际例子展示如何构建一个基础的 Source Generator:自动为带有特定属性的类生成 ToString() 方法。

1. 创建项目结构
新建一个 .NET 类库项目,并修改 .csproj 文件以支持源生成器:


  
    netstandard2.0
    preview
  

2. 定义标记属性
在主项目或共享库中定义一个特性用于标识需要处理的类:

[AttributeUsage(AttributeTargets.Class)]
public class AutoToStringAttribute : Attribute { }

3. 实现 Source Generator
添加一个新类实现 ISourceGenerator

[Generator]
public class ToStringGenerator : ISourceGenerator
{
    public void Initialize(GeneratorInitializationContext context) { }
public void Execute(GeneratorExecutionContext context)
{
    context.RegisterForPostInitialization(pi => pi.AddSource(
        "AutoToString.g.cs",
        SourceText.From(@"

namespace Generated { internal static class Welcome { public static string Get() => ""Hello from Source Generator!""; } }", LanguageNames.CSharp)));

    context.RegisterForSyntaxNotifications(() => new SyntaxReceiver());
}

}

4. 收集目标类并生成代码
使用内部类收集被标记的类:

class SyntaxReceiver : ISyntaxReceiver
{
    public List CandidateClasses { get; } = new();
public void OnVisitSyntaxNode(SyntaxNode syntaxNode)
{
    if (syntaxNode is ClassDeclarationSyntax cds &&
        cds.AttributeLists.SelectMany(al => al.Attributes)
           .Any(a => a.Name.ToString().Contains("AutoToString")))
    {
        CandidateClasses.Add(cds);
    }
}

}

接着在 Execute 中遍历这些类并生成 ToString() 实现(可使用 StringBuilderMicrosoft.CodeAnalysis.Text.SourceText 构建字符串)。

实战场景:DTO 映射、INotifyPropertyChanged、序列化优化

Source Generators 在以下常见场景中非常实用:

  • 自动化属性通知:为 ViewModel 自动生成 INotifyPropertyChanged 代码,无需手动触发事件。
  • DTO 到实体映射:根据命名规则生成高效转换方法,替代 AutoMapper 的反射开销。
  • JSON 序列化辅助:为特定类型生成定制化的 JsonConverter,提高反序列化速度。
  • 路由注册生成:在 Minimal API 或插件系统中,扫描标记控制器并生成路由配置代码。

调试与测试技巧

由于 Source Generators 运行在编译期间,调试相对复杂。推荐做法包括:

  • 使用 context.ReportDiagnostic() 输出日志信息帮助排查问题。
  • 借助 Roslyn SDK 提供的测试工具进行单元测试。
  • 在调用项目中启用 EmitCompilerGeneratedFiles 查看实际生成的代码:
    true
    生成的文件通常位于 obj/Debug/netX.X/generated 目录下。

基本上就这些。掌握 Source Generators 能让你写出更聪明、更高效的 .NET 代码,尤其适合框架开发和高性能场景。虽然学习曲线略陡,但一旦上手,你会发现很多“样板代码”都可以交给编译器自动完成。