diff --git a/GFramework.Generator/generator/enums/EnumExtensionsGenerator.cs b/GFramework.Generator/generator/enums/EnumExtensionsGenerator.cs index 709778a..f0bd865 100644 --- a/GFramework.Generator/generator/enums/EnumExtensionsGenerator.cs +++ b/GFramework.Generator/generator/enums/EnumExtensionsGenerator.cs @@ -10,7 +10,7 @@ namespace GFramework.Generator.generator.enums; [Generator] public class EnumExtensionsGenerator : IIncrementalGenerator { - private const string AttributeFullName = "GFramework.Generator.Attributes.GenerateEnumExtensionsAttribute"; + private const string AttributeFullName = "GFramework.Generator.Attributes.generator.enums.GenerateEnumExtensionsAttribute"; public void Initialize(IncrementalGeneratorInitializationContext context) { diff --git a/GFramework.Generator/generator/logging/LoggerGenerator.cs b/GFramework.Generator/generator/logging/LoggerGenerator.cs index 894d00e..7347a98 100644 --- a/GFramework.Generator/generator/logging/LoggerGenerator.cs +++ b/GFramework.Generator/generator/logging/LoggerGenerator.cs @@ -14,7 +14,7 @@ namespace GFramework.Generator.generator.logging public sealed class LoggerGenerator : IIncrementalGenerator { private const string AttributeMetadataName = - "GFramework.Generator.Attributes.LogAttribute"; + "GFramework.Generator.Attributes.generator.logging.LogAttribute"; /// /// 初始化增量生成器 @@ -85,7 +85,13 @@ namespace GFramework.Generator.generator.logging var attr = classSymbol.GetAttributes() .FirstOrDefault(a => - a.AttributeClass?.ToDisplayString() == AttributeMetadataName); + { + var c = a.AttributeClass; + if (c is null) return false; + + return c.ToDisplayString() == AttributeMetadataName + || c.Name == "LogAttribute"; + }); if (attr is null) {