Improve generator perf (#161)

ImmediatePlatform · Feb 2, 2025 · e7ea085 · e7ea085
1 parent 5ce226d
commit e7ea085
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/Immediate.Validations.Generators/ImmediateValidationsGenerator.cs b/src/Immediate.Validations.Generators/ImmediateValidationsGenerator.cs
@@ -1,4 +1,5 @@
 using Microsoft.CodeAnalysis;
+using Microsoft.CodeAnalysis.CSharp.Syntax;
 using Scriban;
 
 namespace Immediate.Validations.Generators;
@@ -11,8 +12,8 @@ public void Initialize(IncrementalGeneratorInitializationContext context)
 		var validations = context.SyntaxProvider
 			.ForAttributeWithMetadataName(
 				"Immediate.Validations.Shared.ValidateAttribute",
-				(_, _) => true,
-				(ctx, ct) => new ValidateTargetTransformer(ctx, ct).Transform()
+				predicate: (node, _) => node is TypeDeclarationSyntax,
+				transform: (ctx, ct) => new ValidateTargetTransformer(ctx, ct).Transform()
 			)
 			.Where(m => m != null);