Improved output for the benchmark profiling wrapper.

Author: tillig

bench/Autofac.BenchmarkProfiling/Program.cs

@@ -1,5 +1,8 @@
-using BenchmarkDotNet.Running;
+using System.Diagnostics;
+using System.Reflection;
+using BenchmarkDotNet.Running;
 using BenchmarkDotNet.Toolchains.InProcess.NoEmit;
+using Autofac.Core;
 
 namespace Autofac.BenchmarkProfiling;
 
@@ -10,6 +13,11 @@ class Program
 {
     static void Main(string[] args)
     {
+        if (!string.IsNullOrEmpty(Environment.GetEnvironmentVariable("AUTOFAC_SCOPE_DIAGNOSTICS")))
+        {
+            AppContext.SetSwitch("Autofac.ScopeIsolatedDiagnostics", true);
+        }
+
         // Pick a benchmark.
         var availableBenchmarks = Benchmarks.BenchmarkSet.All;
 
@@ -98,6 +106,20 @@ void workloadAction(int repeat)
         // Warmup.
         workloadAction(100);
 
+        if (int.TryParse(Environment.GetEnvironmentVariable("AUTOFAC_MEASURE_ITERATIONS"), out var measurementIterations) &&
+            measurementIterations > 0)
+        {
+            var sw = Stopwatch.StartNew();
+            workloadAction(measurementIterations);
+            sw.Stop();
+            var perIteration = sw.Elapsed.TotalMilliseconds / measurementIterations;
+            Console.WriteLine(
+                "[Profiling] Duration: {0} iterations took {1:F2} ms (avg {2:F4} ms)",
+                measurementIterations,
+                sw.Elapsed.TotalMilliseconds,
+                perIteration);
+        }
+
         // Now start a new thread.
         var runThread = new Thread(new ThreadStart(() =>
         {
@@ -112,6 +134,8 @@ void workloadAction(int repeat)
         runThread.Join();
 
         cleanupAction.InvokeSingle();
+
+        LogScopeDiagnosticsIfEnabled();
     }
 
     private static void PrintBenchmarks(Type[] availableBenchmarks)
@@ -137,4 +161,36 @@ private static void PrintCases(BenchmarkRunInfo benchRunInfo)
             }
         }
     }
+
+    private static void LogScopeDiagnosticsIfEnabled()
+    {
+        if (!AppContext.TryGetSwitch("Autofac.ScopeIsolatedDiagnostics", out var enabled) || !enabled)
+        {
+            return;
+        }
+
+        var diagnosticsType = typeof(IComponentRegistry).Assembly.GetType("Autofac.Core.Registration.ScopeIsolatedServiceDiagnostics");
+        var snapshotProperty = diagnosticsType?.GetProperty(
+            "Snapshot",
+            BindingFlags.NonPublic | BindingFlags.Public | BindingFlags.Static);
+
+        if (snapshotProperty?.GetValue(null) is object snapshot)
+        {
+            var cacheHits = (long)(snapshot.GetType().GetProperty("CacheHits")?.GetValue(snapshot) ?? 0L);
+            var cacheMisses = (long)(snapshot.GetType().GetProperty("CacheMisses")?.GetValue(snapshot) ?? 0L);
+            var cacheAdds = (long)(snapshot.GetType().GetProperty("CacheAdds")?.GetValue(snapshot) ?? 0L);
+            var cacheRemovals = (long)(snapshot.GetType().GetProperty("CacheRemovals")?.GetValue(snapshot) ?? 0L);
+            var cachedInitializations = (long)(snapshot.GetType().GetProperty("CachedInitializations")?.GetValue(snapshot) ?? 0L);
+            var discardedInfos = (long)(snapshot.GetType().GetProperty("ServiceInfoDiscarded")?.GetValue(snapshot) ?? 0L);
+
+            Console.WriteLine(
+                "[Profiling] Scope cache stats -> Hits={0}, Misses={1}, Adds={2}, Removes={3}, CachedInit={4}, Discarded={5}",
+                cacheHits,
+                cacheMisses,
+                cacheAdds,
+                cacheRemovals,
+                cachedInitializations,
+                discardedInfos);
+        }
+    }
 }