Update SemanticModel.cs

Draco-lang · Sep 28, 2023 · 1834caf · 1834caf
1 parent f15ae77
commit 1834caf
Showing 1 changed file with 12 additions and 6 deletions.
diff --git a/src/Draco.Compiler/Api/Semantics/SemanticModel.cs b/src/Draco.Compiler/Api/Semantics/SemanticModel.cs
@@ -4,6 +4,7 @@
 using System.Collections.Immutable;
 using System.Diagnostics.CodeAnalysis;
 using System.Linq;
+using System.Threading.Tasks;
 using Draco.Compiler.Api.Diagnostics;
 using Draco.Compiler.Api.Syntax;
 using Draco.Compiler.Internal;
@@ -61,6 +62,7 @@ private ImmutableArray<Diagnostic> GetDiagnostics(SourceSpan? span = null)
             ? this.Tree.PreOrderTraverse()
             : this.Tree.TraverseSubtreesIntersectingSpan(span.Value);
 
+        var tasks = new List<Task>();
         var addedImportBinders = new HashSet<ImportBinder>();
 
         foreach (var syntaxNode in syntaxNodes)
@@ -81,17 +83,20 @@ private ImmutableArray<Diagnostic> GetDiagnostics(SourceSpan? span = null)
             case CompilationUnitSyntax:
             case FunctionDeclarationSyntax:
             {
-                containingSymbol?.Bind(this);
+                tasks.Add(Task.Run(() => containingSymbol?.Bind(this)));
                 break;
             }
             // NOTE: Only globals need binding
             case VariableDeclarationSyntax when containingSymbol is SourceModuleSymbol containingModule:
             {
-                // We need to search for this global
-                var globalSymbol = containingModule.Members
-                    .OfType<SourceGlobalSymbol>()
-                    .Single(s => s.DeclaringSyntax == syntaxNode);
-                globalSymbol.Bind(this);
+                tasks.Add(Task.Run(() =>
+                {
+                    // We need to search for this global
+                    var globalSymbol = containingModule.Members
+                        .OfType<SourceGlobalSymbol>()
+                        .Single(s => s.DeclaringSyntax == syntaxNode);
+                    globalSymbol.Bind(this);
+                }));
                 break;
             }
             case ImportDeclarationSyntax:
@@ -110,6 +115,7 @@ private ImmutableArray<Diagnostic> GetDiagnostics(SourceSpan? span = null)
             }
         }
 
+        Task.WaitAll(tasks.ToArray());
         return this.DiagnosticBag.ToImmutableArray();
     }