Throw if cancellation requested.

Author: ChebanovDD

src/UnityUxmlGenerator.UnityPackage/Assets/Plugins/UnityUxmlGenerator/Runtime/UnityUxmlGenerator.dll

@@ -81,7 +81,7 @@ private static MemberDeclarationSyntax[] GetTraitsClassMembers(GeneratorExecutio
                     typeToCast: IdentifierName(capture.ClassName)))
         };
 
-        initMethodBody.AddRange(GetAttributeValueAssignments(capture));
+        initMethodBody.AddRange(GetAttributeValueAssignments(context, capture));
 
         var initMethod = MethodWidget(
             identifier: "Init",
@@ -115,6 +115,8 @@ private static IEnumerable<MemberDeclarationSyntax> GetAttributeFields(Generator
 
         foreach (var (property, uxmlAttributeDefaultValue) in capture.Properties)
         {
+            context.CancellationToken.ThrowIfCancellationRequested();
+
             fields.Add(GetAttributeFieldDeclaration(GetAttributeInfo(context, property, uxmlAttributeDefaultValue)));
         }
 
@@ -222,12 +224,15 @@ private static FieldDeclarationSyntax GetAttributeFieldDeclaration(UxmlAttribute
         );
     }
 
-    private static IEnumerable<StatementSyntax> GetAttributeValueAssignments(UxmlTraitsCapture capture)
+    private static IEnumerable<StatementSyntax> GetAttributeValueAssignments(GeneratorExecutionContext context,
+        UxmlTraitsCapture capture)
     {
         var attributeValueAssignments = new List<StatementSyntax>();
 
         foreach (var (property, _) in capture.Properties)
         {
+            context.CancellationToken.ThrowIfCancellationRequested();
+
             var propertyName = property.GetName();
             var fieldName = propertyName.ToPrivateFieldName();
 

src/UnityUxmlGenerator/UxmlGenerator.Traits.cs

@@ -26,18 +26,23 @@ public void Execute(GeneratorExecutionContext context)
         context.AddSource($"{nameof(UxmlElementClassName)}.g.cs", GenerateUxmlElementAttribute());
         context.AddSource($"{nameof(UxmlAttributeClassName)}.g.cs", GenerateUxmlAttributeAttribute());
 
-        if (context.SyntaxReceiver is not VisualElementReceiver receiver)
+        if (context.SyntaxReceiver is not VisualElementReceiver receiver ||
+            context.CancellationToken.IsCancellationRequested)
         {
             return;
         }
 
         foreach (var uxmlElement in receiver.UxmlFactoryReceiver.Captures)
         {
+            context.CancellationToken.ThrowIfCancellationRequested();
+
             AddSource(context, uxmlElement, GenerateUxmlFactory(uxmlElement));
         }
 
         foreach (var capture in receiver.UxmlTraitsReceiver.Captures)
         {
+            context.CancellationToken.ThrowIfCancellationRequested();
+
             AddSource(context, capture.Value, GenerateUxmlTraits(context, capture.Value));
         }