Use array builder pools

Author: chsienki

src/Compiler/Microsoft.NET.Sdk.Razor.SourceGenerators/Microsoft.NET.Sdk.Razor.SourceGenerators.csproj

@@ -19,6 +19,7 @@
   </ItemGroup>
 
   <ItemGroup>
+    <ProjectReference Include="..\..\Shared\Microsoft.AspNetCore.Razor.Utilities.Shared\Microsoft.AspNetCore.Razor.Utilities.Shared.csproj" />
     <ProjectReference Include="..\Microsoft.AspNetCore.Mvc.Razor.Extensions\src\Microsoft.AspNetCore.Mvc.Razor.Extensions.csproj" />
     <ProjectReference Include="..\Microsoft.AspNetCore.Razor.Language\src\Microsoft.AspNetCore.Razor.Language.csproj" />
     <ProjectReference Include="..\Microsoft.CodeAnalysis.Razor\src\Microsoft.CodeAnalysis.Razor.csproj" />

src/Compiler/Microsoft.NET.Sdk.Razor.SourceGenerators/RazorSourceGenerator.cs

@@ -7,6 +7,7 @@
 using System.IO;
 using System.Linq;
 using Microsoft.AspNetCore.Razor.Language;
+using Microsoft.AspNetCore.Razor.PooledObjects;
 using Microsoft.CodeAnalysis;
 using Microsoft.CodeAnalysis.CSharp;
 using Microsoft.CodeAnalysis.CSharp.Syntax;
@@ -185,7 +186,7 @@ public void Initialize(IncrementalGeneratorInitializationContext context)
                     var tagHelperFeature = new StaticCompilationTagHelperFeature();
                     var discoveryProjectEngine = GetDiscoveryProjectEngine(compilation.References.ToImmutableArray(), tagHelperFeature);
 
-                    var descriptors = ImmutableArray.CreateBuilder<TagHelperDescriptor>();
+                    using var pool = ArrayBuilderPool<TagHelperDescriptor>.GetPooledObject(out var descriptors);
                     tagHelperFeature.Compilation = compilation;
                     foreach (var reference in compilation.References)
                     {
@@ -212,8 +213,8 @@ public void Initialize(IncrementalGeneratorInitializationContext context)
                         return ImmutableArray<TagHelperDescriptor>.Empty;
                     }
 
-                    var allTagHelpers = ImmutableArray.CreateBuilder<TagHelperDescriptor>(count);
-                    allTagHelpers.AddRange(tagHelpersFromCompilation);
+                    using var pool = ArrayBuilderPool<TagHelperDescriptor>.GetPooledObject(out var allTagHelpers);
+					allTagHelpers.AddRange(tagHelpersFromCompilation);
                     allTagHelpers.AddRange(tagHelpersFromReferences);
                     allTagHelpers.AddRange(tagHelpersFromComponents);
 

src/Compiler/Microsoft.NET.Sdk.Razor.SourceGenerators/StaticCompilationTagHelperFeature.cs

@@ -5,6 +5,7 @@
 using System.Collections.Immutable;
 using System.Linq;
 using Microsoft.AspNetCore.Razor.Language;
+using Microsoft.AspNetCore.Razor.PooledObjects;
 using Microsoft.CodeAnalysis;
 using Microsoft.CodeAnalysis.Razor;
 
@@ -21,8 +22,8 @@ public ImmutableArray<TagHelperDescriptor> GetDescriptors()
                 return ImmutableArray<TagHelperDescriptor>.Empty;
             }
 
-            var results = ImmutableArray.CreateBuilder<TagHelperDescriptor>();
-            var context = TagHelperDescriptorProviderContext.Create(results);
+			using var pool = ArrayBuilderPool<TagHelperDescriptor>.GetPooledObject(out var results);
+			var context = TagHelperDescriptorProviderContext.Create(results);
             context.SetCompilation(Compilation);
             if (TargetSymbol is not null)
             {