feat: perform warmup emit when connection is established

rdavisau · Jun 8, 2022 · 5b663ba · 5b663ba
1 parent 3254a01
commit 5b663ba
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 0 deletions.
diff --git a/src/components/tbc.host/Components/FileEnvironment/FileEnvironment.cs b/src/components/tbc.host/Components/FileEnvironment/FileEnvironment.cs
@@ -126,6 +126,10 @@ public async Task SetupReferenceTracking()
 
                 foreach (var refSeen in existingReferences.Select(x => x.AssemblyLocation))
                     seen.Add(refSeen);
+
+                // allow a little time for references to come back <:-)
+                // should have client tell us when all current dependencies have been sent
+                Task.Delay(TimeSpan.FromSeconds(.33)).ContinueWith(_ => IncrementalCompiler.DoWarmup());
             }
 
             if (targetHello.UseDependencyCache && targetHello.ApplicationIdentifier is { } appIdentifier)

diff --git a/src/components/tbc.host/Components/IncrementalCompiler/IIncrementalCompiler.cs b/src/components/tbc.host/Components/IncrementalCompiler/IIncrementalCompiler.cs
@@ -17,5 +17,6 @@ public interface IIncrementalCompiler
         void ClearTrees();
         void PrintTrees(bool withDetail);
         string TryResolvePrimaryType(string typeHint);
+        void DoWarmup();
     }
 }
diff --git a/src/components/tbc.host/Components/IncrementalCompiler/IncrementalCompiler.cs b/src/components/tbc.host/Components/IncrementalCompiler/IncrementalCompiler.cs
@@ -295,6 +295,26 @@ private bool TryApplyCompilationFixers(EmitResult result, CSharpCompilation curr
                 return false;
             }
         }
+
+        public void DoWarmup()
+        {
+            var sw = Stopwatch.StartNew();
+            try
+            {
+                StageFile(new ChangedFile {
+                    Path = "warmup",
+                    Contents = "public class Warm { public string Warmup { get; set; } }",
+                    ChangedAt = DateTimeOffset.UtcNow });
+
+                foreach (var c in _compilationFixers.Values)
+                    c.TryFix(CurrentCompilation, new(), out _);
+            }
+            catch (Exception ex) { Logger.LogError(ex, "Warmup"); }
+
+            TryRemoveTree("warmup");
+
+            Logger.LogInformation("Warmup emit completed in {Elapsed}", sw.Elapsed);
+        }
 
         private void WithCompilation(Func<CSharpCompilation, CSharpCompilation> action)
         {