.Net Core 你必须知道的source-generators

Chester Chen

发布于 2024-03-25 13:58:36

41300

代码可运行

文章被收录于专栏：chester技术分享chester技术分享

运行总次数：0

代码可运行

源生成器是 C# 9 中引入的一项功能，允许在编译过程中动态生成代码。

它们直接与 C# 编译器集成(Roslyn)并在编译时运行，分析源代码并根据分析结果生成附加代码。

源生成器提供了一种简化的自动化代码生成方法，无需外部工具或单独的预编译步骤。

通过无缝集成到编译过程中，源生成器可以提高生产力、减少错误并实现更高效的开发工作流程。

如何使用

创建 .NET 控制台应用程序。此示例使用 .NET 6。将 Program 类替换为以下代码。

namespace ConsoleApp;

partial class Program
{
    static void Main(string[] args)
    {
        HelloFrom("Generated Code");
    }

    static partial void HelloFrom(string name);
}

接下来，我们将创建一个源生成器项目来实现 partial void HelloFrom 方法对应项。

创建一个以 netstandard2.0 目标框架名字对象为目标的 .NET 类库。添加以下 NuGet 包

<ItemGroup>
    <PackageReference Include="Microsoft.CodeAnalysis.CSharp" Version="4.8.0" PrivateAssets="all" />
    <PackageReference Include="Microsoft.CodeAnalysis.Analyzers" Version="3.3.4" PrivateAssets="all" />
</ItemGroup>

然后，创建一个名为 HelloSourceGenerator.cs 的新 C# 文件，该文件指定你自己的源生成器，如下所示：

using Microsoft.CodeAnalysis;

namespace SourceGeneratorInCSharp
{
    [Generator]
    public class SourceGenerator : ISourceGenerator
    {
        public void Execute(GeneratorExecutionContext context)
        {
            // Find the main method
            var mainMethod = context.Compilation.GetEntryPoint(context.CancellationToken);

            // Build up the source code
            string source = $@"// <auto-generated/>
using System;

namespace {mainMethod.ContainingNamespace.ToDisplayString()}
{{
    public static partial class {mainMethod.ContainingType.Name}
    {{
        static partial void HelloFrom(string name) =>
            Console.WriteLine($""Generator says: Hi from '{{name}}'"");
    }}
}}
";
            var typeName = mainMethod.ContainingType.Name;

            // Add the source code to the compilation
            context.AddSource($"{typeName}.g.cs", source);
        }

        public void Initialize(GeneratorInitializationContext context)
        {
            // No initialization required for this one
        }
    }
}

现在，我们有一个正常运行的生成器，但需要将其引用到控制台应用程序。

新引用不是传统的项目引用，必须手动编辑以包含 OutputItemType 和 ReferenceOutputAssembly 属性。

<ItemGroup>
    <ProjectReference Include="..\SourceGeneratorInCSharp\SourceGeneratorInCSharp.csproj"
                        OutputItemType="Analyzer"
                        ReferenceOutputAssembly="false" /> 
</ItemGroup>

现在，运行控制台应用程序时，应会看到生成的代码运行并打印到屏幕。

控制台应用程序本身不实现 HelloFrom 方法，而是在编译过程中从源生成器项目生成的源。