test

Author: filzrev

src/BenchmarkDotNet/EventProcessors/CompositeEventProcessor.cs

@@ -1,6 +1,7 @@
 using System;
 using System.Collections.Generic;
 using BenchmarkDotNet.Extensions;
+using BenchmarkDotNet.Loggers;
 using BenchmarkDotNet.Reports;
 using BenchmarkDotNet.Running;
 using BenchmarkDotNet.Toolchains.Results;
@@ -12,14 +13,24 @@ internal sealed class CompositeEventProcessor : EventProcessor
     {
         private readonly HashSet<EventProcessor> eventProcessors;
 
-        public CompositeEventProcessor(BenchmarkRunInfo[] benchmarkRunInfos)
+        public CompositeEventProcessor(BenchmarkRunInfo[] benchmarkRunInfos, CompositeLogger compositeLogger)
         {
             var eventProcessors = new HashSet<EventProcessor>();
 
             foreach (var info in benchmarkRunInfos)
                 eventProcessors.AddRange(info.Config.GetEventProcessors());
 
+            // Try add log grouping event processor when running on GitHub Actions or Azure Pipelines.
+            if (IsGitHubAction() || IsAzurePipelines())
+            {
+                if (compositeLogger.TryGetConsoleLogger(out var consoleLogger))
+                    eventProcessors.Add(new LogGroupingEventProcessor(consoleLogger));
+            }
+
             this.eventProcessors = eventProcessors;
+
+            static bool IsGitHubAction() => Environment.GetEnvironmentVariable("GITHUB_ACTION") == "true";
+            static bool IsAzurePipelines() => Environment.GetEnvironmentVariable("TF_BUILD") == "True";
         }
 
         public override void OnStartValidationStage()

src/BenchmarkDotNet/EventProcessors/LogGroupingEventProcessor.cs

@@ -0,0 +1,69 @@
+using System;
+using System.Collections.Generic;
+using BenchmarkDotNet.Extensions;
+using BenchmarkDotNet.Loggers;
+using BenchmarkDotNet.Reports;
+using BenchmarkDotNet.Running;
+
+namespace BenchmarkDotNet.EventProcessors
+{
+    internal sealed class LogGroupingEventProcessor : EventProcessor
+    {
+        private readonly ILogger consoleLogger;
+
+        public LogGroupingEventProcessor(ConsoleLogger logger)
+        {
+            consoleLogger = logger;
+        }
+
+        public override void OnStartValidationStage()
+        {
+            consoleLogger.WriteLine("##[group]ValidationStage");
+        }
+
+        public override void OnEndValidationStage()
+        {
+            consoleLogger.WriteLine("##[endgroup]");
+        }
+
+        public override void OnStartBuildStage(IReadOnlyList<BuildPartition> partitions)
+        {
+            consoleLogger.WriteLine("##[group]BuildStage");
+        }
+
+        public override void OnEndBuildStage()
+        {
+            consoleLogger.WriteLine("##[endgroup]");
+        }
+
+        public override void OnStartRunStage()
+        {
+            consoleLogger.WriteLine("##[group]RunStage");
+        }
+
+        public override void OnStartRunBenchmarksInType(Type type, IReadOnlyList<BenchmarkCase> benchmarks)
+        {
+            consoleLogger.WriteLine($"##[group]RunBenchmarksInType: {type.GetDisplayName()}");
+        }
+
+        public override void OnEndRunBenchmarksInType(Type type, Summary summary)
+        {
+            consoleLogger.WriteLine("##[endgroup]");
+        }
+
+        public override void OnStartRunBenchmark(BenchmarkCase benchmarkCase)
+        {
+            Console.WriteLine($"##[group]RunBenchmark: {benchmarkCase.DisplayInfo}");
+        }
+
+        public override void OnEndRunBenchmark(BenchmarkCase benchmarkCase, BenchmarkReport report)
+        {
+            Console.WriteLine("##[endgroup]");
+        }
+
+        public override void OnEndRunStage()
+        {
+            Console.WriteLine("##[endgroup]");
+        }
+    }
+}

src/BenchmarkDotNet/Loggers/CompositeLogger.cs

@@ -1,4 +1,5 @@
 using System.Collections.Immutable;
+using System.Linq;
 
 namespace BenchmarkDotNet.Loggers
 {
@@ -57,5 +58,11 @@ public void Flush()
                 }
             }
         }
+
+        public bool TryGetConsoleLogger(out ConsoleLogger consoleLogger)
+        {
+            consoleLogger = loggers.OfType<ConsoleLogger>().FirstOrDefault();
+            return consoleLogger != null;
+        }
     }
 }

src/BenchmarkDotNet/Running/BenchmarkRunnerClean.cs

@@ -55,8 +55,9 @@ internal static Summary[] Run(BenchmarkRunInfo[] benchmarkRunInfos)
             using (var streamLogger = new StreamLogger(GetLogFileStreamWriter(benchmarkRunInfos, logFilePath)))
             {
                 var compositeLogger = CreateCompositeLogger(benchmarkRunInfos, streamLogger);
+
                 using var wakeLock = WakeLock.Request(WakeLock.GetWakeLockType(benchmarkRunInfos), "BenchmarkDotNet Running Benchmarks", streamLogger);
-                var eventProcessor = new CompositeEventProcessor(benchmarkRunInfos);
+                var eventProcessor = new CompositeEventProcessor(benchmarkRunInfos, (CompositeLogger)compositeLogger);
 
                 eventProcessor.OnStartValidationStage();