C# 9 新特性：代碼生成器、編譯時反射-有解無憂

前言

今天 .NET 官方博客宣布 C# 9 Source Generators 第一個預覽版發布，這是一個用戶已經喊了快 5 年特性，今天終于發布了，

簡介

Source Generators 顧名思義代碼生成器，它允許開發者在代碼編譯程序中獲取查看用戶代碼并且生成新的 C# 代碼參與編譯程序，并且可以很好的與代碼分析器集成提供 Intellisense、除錯資訊和報錯資訊，可以用它來做代碼生成，因此也相當于是一個加強版本的編譯時反射，

使用 Source Generators，可以做到這些事情：

獲取一個 Compilation 物件，這個物件表示了所有正在編譯的用戶代碼，你可以從中獲取 AST 和語意模型等資訊
可以向 Compilation 物件中插入新的代碼，讓編譯器連同已有的用戶代碼一起編譯

Source Generators 作為編譯程序中的一個階段執行：

編譯運行 -> [分析源代碼 -> 生成新代碼] -> 將生成的新代碼添加入編譯程序 -> 編譯繼續，

上述流程中，中括號包括的內容即為 Source Generators 所參與的階段和能做到的事情，

作用

.NET 明明具備運行時反射和動態 IL 織入功能，那這個 Source Generators 有什么用呢？

編譯時反射 - 0 運行時開銷

拿 ASP.NET Core 舉例，啟動一個 ASP.NET Core 應用時，首先會通過運行時反射來發現 Controllers、Services 等的型別定義，然后在請求管道中需要通過運行時反射獲取其建構式資訊以便于進行依賴注入，然而運行時反射開銷很大，即使快取了型別簽名，對于剛剛啟動后的應用也無任何幫助作用，而且不利于做 AOT 編譯，

Source Generators 將可以讓 ASP.NET Core 所有的型別發現、依賴注入等在編譯時就全部完成并編譯到最終的程式集當中，最終做到 0 運行時反射使用，不僅利于 AOT 編譯，而且運行時 0 開銷，

除了上述作用之外，gRPC 等也可以利用此功能在編譯時織入代碼參與編譯，不需要再利用任何的 MSBuild Task 做代碼生成啦！

另外，甚至還可以讀取 XML、JSON 直接生成 C# 代碼參與編譯，DTO 撰寫全自動化都是沒問題的，

AOT 編譯

Source Generators 的另一個作用是可以幫助消除 AOT 編譯優化的主要障礙，

許多框架和庫都大量使用反射，例如System.Text.Json、System.Text.RegularExpressions、ASP.NET Core 和 WPF 等等，它們在運行時從用戶代碼中發現型別，這些非常不利于 AOT 編譯優化，因為為了使反射能夠正常作業，必須將大量額外甚至可能不需要的型別元資料編譯到最終的原生映像當中，

有了 Source Generators 之后，只需要做編譯時代碼生成便可以避免大部分的運行時反射的使用，讓 AOT 編譯優化工具能夠更好的運行，

例子

`INotifyPropertyChanged`

寫過 WPF 或 UWP 的都知道，在 ViewModel 中為了使屬性變更可被發現，需要實作 INotifyPropertyChanged 介面，并且在每一個需要的屬性的 setter 處觸發屬性更改事件：

class MyViewModel : INotifyPropertyChanged
{
    public event PropertyChangedEventHandler? PropertyChanged;

    private string _text;
    public string Text
    {
        get => _text;
        set
        {
            _text = value;
            PropertyChanged?.Invoke(this, new PropertyChangedEventArgs(nameof(Text)));
        }
    }
}

當屬性多了之后將會非常繁瑣，先前 C# 引入了 CallerMemberName 用于簡化屬性較多時候的情況：

class MyViewModel : INotifyPropertyChanged
{
    public event PropertyChangedEventHandler? PropertyChanged;

    private string _text;
    public string Text
    {
        get => _text;
        set
        {
            _text = value;
            OnPropertyChanged();
        }
    }

    protected virtual void OnPropertyChanged([CallerMemberName] string? propertyName = null)
    {
        PropertyChanged?.Invoke(this, new PropertyChangedEventArgs(propertyName));
    }
}

即，用 CallerMemberName 指示引數，在編譯時自動填充呼叫方的成員名稱，

但是還是不方便，

如今有了 Source Generators，我們可以在編譯時生成代碼做到這一點了，

為了實作 Source Generators，我們需要寫個實作了 ISourceGenerator 并且標注了 Generator 的型別，

完整的 Source Generators 代碼如下：

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using Microsoft.CodeAnalysis;
using Microsoft.CodeAnalysis.CSharp;
using Microsoft.CodeAnalysis.CSharp.Syntax;
using Microsoft.CodeAnalysis.Text;

namespace MySourceGenerator
{
    [Generator]
    public class AutoNotifyGenerator : ISourceGenerator
    {
        private const string attributeText = @"
using System;
namespace AutoNotify
{
    [AttributeUsage(AttributeTargets.Field, Inherited = false, AllowMultiple = false)]
    sealed class AutoNotifyAttribute : Attribute
    {
        public AutoNotifyAttribute()
        {
        }
        public string PropertyName { get; set; }
    }
}
";

        public void Initialize(InitializationContext context)
        {
            // 注冊一個語法接收器，會在每次生成時被創建
            context.RegisterForSyntaxNotifications(() => new SyntaxReceiver());
        }

        public void Execute(SourceGeneratorContext context)
        {
            // 添加 Attrbite 文本
            context.AddSource("AutoNotifyAttribute", SourceText.From(attributeText, Encoding.UTF8));

            // 獲取先前的語法接收器 
            if (!(context.SyntaxReceiver is SyntaxReceiver receiver))
                return;

            // 創建處目標名稱的屬性
            CSharpParseOptions options = (context.Compilation as CSharpCompilation).SyntaxTrees[0].Options as CSharpParseOptions;
            Compilation compilation = context.Compilation.AddSyntaxTrees(CSharpSyntaxTree.ParseText(SourceText.From(attributeText, Encoding.UTF8), options));

            // 獲取新系結的 Attribute，并獲取INotifyPropertyChanged
            INamedTypeSymbol attributeSymbol = compilation.GetTypeByMetadataName("AutoNotify.AutoNotifyAttribute");
            INamedTypeSymbol notifySymbol = compilation.GetTypeByMetadataName("System.ComponentModel.INotifyPropertyChanged");

            // 遍歷欄位，只保留有 AutoNotify 標注的欄位
            List<IFieldSymbol> fieldSymbols = new List<IFieldSymbol>();
            foreach (FieldDeclarationSyntax field in receiver.CandidateFields)
            {
                SemanticModel model = compilation.GetSemanticModel(field.SyntaxTree);
                foreach (VariableDeclaratorSyntax variable in field.Declaration.Variables)
                {
                    // 獲取欄位符號資訊，如果有 AutoNotify 標注則保存
                    IFieldSymbol fieldSymbol = model.GetDeclaredSymbol(variable) as IFieldSymbol;
                    if (fieldSymbol.GetAttributes().Any(ad => ad.AttributeClass.Equals(attributeSymbol, SymbolEqualityComparer.Default)))
                    {
                        fieldSymbols.Add(fieldSymbol);
                    }
                }
            }

            // 按 class 對欄位進行分組，并生成代碼
            foreach (IGrouping<INamedTypeSymbol, IFieldSymbol> group in fieldSymbols.GroupBy(f => f.ContainingType))
            {
                string classSource = ProcessClass(group.Key, group.ToList(), attributeSymbol, notifySymbol, context);
               context.AddSource($"{group.Key.Name}_autoNotify.cs", SourceText.From(classSource, Encoding.UTF8));
            }
        }

        private string ProcessClass(INamedTypeSymbol classSymbol, List<IFieldSymbol> fields, ISymbol attributeSymbol, ISymbol notifySymbol, SourceGeneratorContext context)
        {
            if (!classSymbol.ContainingSymbol.Equals(classSymbol.ContainingNamespace, SymbolEqualityComparer.Default))
            {
                // TODO: 必須在頂層，產生診斷資訊
                return null;
            }

            string namespaceName = classSymbol.ContainingNamespace.ToDisplayString();

            // 開始構建要生成的代碼
            StringBuilder source = new StringBuilder($@"
namespace {namespaceName}
{{
    public partial class {classSymbol.Name} : {notifySymbol.ToDisplayString()}
    {{
");

            // 如果型別還沒有實作 INotifyPropertyChanged 則添加實作
            if (!classSymbol.Interfaces.Contains(notifySymbol))
            {
                source.Append("public event System.ComponentModel.PropertyChangedEventHandler PropertyChanged;");
            }

            // 生成屬性
            foreach (IFieldSymbol fieldSymbol in fields)
            {
                ProcessField(source, fieldSymbol, attributeSymbol);
            }

            source.Append("} }");
            return source.ToString();
        }

        private void ProcessField(StringBuilder source, IFieldSymbol fieldSymbol, ISymbol attributeSymbol)
        {
            // 獲取欄位名稱
            string fieldName = fieldSymbol.Name;
            ITypeSymbol fieldType = fieldSymbol.Type;

            // 獲取 AutoNotify Attribute 和相關的資料
            AttributeData attributeData = https://www.cnblogs.com/hez2010/p/fieldSymbol.GetAttributes().Single(ad => ad.AttributeClass.Equals(attributeSymbol, SymbolEqualityComparer.Default));
            TypedConstant overridenNameOpt = attributeData.NamedArguments.SingleOrDefault(kvp => kvp.Key == "PropertyName").Value;

            string propertyName = chooseName(fieldName, overridenNameOpt);
            if (propertyName.Length == 0 || propertyName == fieldName)
            {
                //TODO: 無法處理，產生診斷資訊
                return;
            }

            source.Append($@"
public {fieldType} {propertyName} 
{{
    get 
    {{
        return this.{fieldName};
    }}
    set
    {{
        this.{fieldName} = value;
        this.PropertyChanged?.Invoke(this, new System.ComponentModel.PropertyChangedEventArgs(nameof({propertyName})));
    }}
}}
");

            string chooseName(string fieldName, TypedConstant overridenNameOpt)
            {
                if (!overridenNameOpt.IsNull)
                {
                    return overridenNameOpt.Value.ToString();
                }

                fieldName = fieldName.TrimStart('_');
                if (fieldName.Length == 0)
                    return string.Empty;

                if (fieldName.Length == 1)
                    return fieldName.ToUpper();

                return fieldName.Substring(0, 1).ToUpper() + fieldName.Substring(1);
            }

        }

        // 語法接收器，將在每次生成代碼時被按需創建
        class SyntaxReceiver : ISyntaxReceiver
        {
            public List<FieldDeclarationSyntax> CandidateFields { get; } = new List<FieldDeclarationSyntax>();

            // 編譯中在訪問每個語法節點時被呼叫，我們可以檢查節點并保存任何對生成有用的資訊
            public void OnVisitSyntaxNode(SyntaxNode syntaxNode)
            {
                // 將具有至少一個 Attribute 的任何欄位作為候選
                if (syntaxNode is FieldDeclarationSyntax fieldDeclarationSyntax
                    && fieldDeclarationSyntax.AttributeLists.Count > 0)
                {
                    CandidateFields.Add(fieldDeclarationSyntax);
                }
            }
        }
    }
}

有了上述代碼生成器之后，以后我們只需要這樣寫 ViewModel 就會自動生成通知介面的事件觸發呼叫：

public partial class MyViewModel
{
    [AutoNotify]
    private string _text = "private field text";

    [AutoNotify(PropertyName = "Count")]
    private int _amount = 5;
}

上述代碼將會在編譯時自動生成以下代碼參與編譯：

public partial class MyViewModel : System.ComponentModel.INotifyPropertyChanged
{
    public event System.ComponentModel.PropertyChangedEventHandler PropertyChanged;

    public string Text
    {
        get 
        {
            return this._text;
        }
        set
        {
            this._text = value;
            this.PropertyChanged?.Invoke(this, new System.ComponentModel.PropertyChangedEventArgs(nameof(Text)));
        }
    }

    public int Count
    {
        get 
        {
            return this._amount;
        }
        set
        {
            this._amount = value;
            this.PropertyChanged?.Invoke(this, new System.ComponentModel.PropertyChangedEventArgs(nameof(Count)));
        }
    }
}

非常方便！

使用時，將 Source Generators 部分作為一個獨立的 .NET Standard 2.0 程式集（暫時不支持 2.1），用以下方式引入到你的專案即可：

<ItemGroup>
  <Analyzer Include="..\MySourceGenerator\bin\$(Configuration)\netstandard2.0\MySourceGenerator.dll" />
</ItemGroup>

<ItemGroup>
  <ProjectReference Include="..\MySourceGenerator\MySourceGenerator.csproj" />
</ItemGroup>

注意需要最新的 .NET 5 preview（寫文章時還在 artifacts 里沒正式 release），并指定語言版本為 preview：

<PropertyGroup>
  <LangVersion>preview</LangVersion>
</PropertyGroup>

另外，Source Generators 需要引入兩個 nuget 包：

<ItemGroup>
  <PackageReference Include="Microsoft.CodeAnalysis.CSharp.Workspaces" Version="3.6.0-3.final" PrivateAssets="all" />
  <PackageReference Include="Microsoft.CodeAnalysis.Analyzers" Version="3.0.0" PrivateAssets="all" />
</ItemGroup>

限制

Source Generators 僅能用于訪問和生成代碼，但是不能修改已有代碼，這有一定原因是出于安全考量，

檔案

Source Generators 處于早期預覽階段，docs.microsoft.com 上暫時沒有相關檔案，關于它的檔案請訪問在 roslyn 倉庫中的檔案：

設計檔案

使用檔案

后記

目前 Source Generators 仍處于非常早期的預覽階段，API 后期還可能會有很大的改動，因此現階段不要用于生產，

另外，關于與 IDE 的集成、診斷資訊、斷點除錯資訊等的開發也在進行中，請期待后續的 preview 版本吧，

轉載請註明出處，本文鏈接：https://www.uj5u.com/net/44848.html

標籤：C#

上一篇：關于.NetCore與.Netframework 對于DataSet的序列化與反序列化問題的探討.

下一篇：從GC的SuppressFinalize方法帶你深刻認識Finalize底層運行機制