C# attributes如何实现代码生成

C# attributes 是一种用于为元素添加元数据的特性。通过在属性中添加特定的标记，可以为类、方法、属性等添加额外的信息。在代码生成中，可以使用 attributes 来标记需要生成代码的元素，并编写代码生成器来根据这些标记来生成相应的代码。

以下是一个简单的示例，演示如何使用 attributes 实现代码生成：

[CodeGenerator("MyCodeGenerator")]
public class MyClass
{
    [GenerateMethod]
    public void MyMethod()
    {
        Console.WriteLine("Generated code");
    }
}

public class CodeGeneratorAttribute : Attribute
{
    public string GeneratorName { get; set; }

    public CodeGeneratorAttribute(string generatorName)
    {
        GeneratorName = generatorName;
    }
}

public class GenerateMethodAttribute : Attribute
{
    // This attribute doesn't need any parameters
}

public class CodeGenerator
{
    public static void GenerateCode(object obj)
    {
        Type type = obj.GetType();
        CodeGeneratorAttribute generatorAttribute = (CodeGeneratorAttribute)Attribute.GetCustomAttribute(type, typeof(CodeGeneratorAttribute));

        if (generatorAttribute != null)
        {
            // Check if the generator name matches the expected value
            if (generatorAttribute.GeneratorName == "MyCodeGenerator")
            {
                MethodInfo[] methods = type.GetMethods();
                foreach (MethodInfo method in methods)
                {
                    if (Attribute.IsDefined(method, typeof(GenerateMethodAttribute)))
                    {
                        // Generate code based on the method
                        Console.WriteLine($"Generated code for method {method.Name}");
                    }
                }
            }
        }
    }
}

class Program
{
    static void Main()
    {
        MyClass myClass = new MyClass();
        CodeGenerator.GenerateCode(myClass);
    }
}

在上面的示例中，我们定义了两个自定义 attributes：CodeGeneratorAttribute 和 GenerateMethodAttribute。我们为 MyClass 类添加了 CodeGenerator attribute，并为 MyMethod 方法添加了 GenerateMethod attribute。然后，我们使用 CodeGenerator 类来生成代码，根据这些 attributes 来生成相应的代码。

当运行程序时，将输出以下内容：

Generated code for method MyMethod

这样，通过使用 attributes，我们可以轻松地为元素添加标记，并根据这些标记来生成相应的代码。