GitHub Actions Runner

2026-03-15 09:20:44 -04:00 · 2019-10-10 00:52:42 -04:00
commit c8afc84840
1255 changed files with 198670 additions and 0 deletions
--- a/src/Runner.Common/TraceManager.cs
+++ b/src/Runner.Common/TraceManager.cs
@@ -0,0 +1,88 @@
+using GitHub.Runner.Common.Util;
+using System;
+using System.Collections.Concurrent;
+using System.Diagnostics;
+using GitHub.DistributedTask.Logging;
+using GitHub.Runner.Sdk;
+
+namespace GitHub.Runner.Common
+{
+    public interface ITraceManager : IDisposable
+    {
+        SourceSwitch Switch { get; }
+        Tracing this[string name] { get; }
+    }
+
+    public sealed class TraceManager : ITraceManager
+    {
+        private readonly ConcurrentDictionary<string, Tracing> _sources = new ConcurrentDictionary<string, Tracing>(StringComparer.OrdinalIgnoreCase);
+        private readonly HostTraceListener _hostTraceListener;
+        private TraceSetting _traceSetting;
+        private ISecretMasker _secretMasker;
+
+        public TraceManager(HostTraceListener traceListener, ISecretMasker secretMasker)
+            : this(traceListener, new TraceSetting(), secretMasker)
+        {
+        }
+
+        public TraceManager(HostTraceListener traceListener, TraceSetting traceSetting, ISecretMasker secretMasker)
+        {
+            // Validate and store params.
+            ArgUtil.NotNull(traceListener, nameof(traceListener));
+            ArgUtil.NotNull(traceSetting, nameof(traceSetting));
+            ArgUtil.NotNull(secretMasker, nameof(secretMasker));
+            _hostTraceListener = traceListener;
+            _traceSetting = traceSetting;
+            _secretMasker = secretMasker;
+
+            Switch = new SourceSwitch("GitHubActionsRunnerSwitch")
+            {
+                Level = _traceSetting.DefaultTraceLevel.ToSourceLevels()
+            };
+        }
+
+        public SourceSwitch Switch { get; private set; }
+
+        public Tracing this[string name]
+        {
+            get
+            {
+                return _sources.GetOrAdd(name, key => CreateTraceSource(key));
+            }
+        }
+
+        public void Dispose()
+        {
+            Dispose(true);
+            GC.SuppressFinalize(this);
+        }
+
+        private void Dispose(bool disposing)
+        {
+            if (disposing)
+            {
+                foreach (Tracing traceSource in _sources.Values)
+                {
+                    traceSource.Dispose();
+                }
+
+                _sources.Clear();
+            }
+        }
+
+        private Tracing CreateTraceSource(string name)
+        {
+            SourceSwitch sourceSwitch = Switch;
+
+            TraceLevel sourceTraceLevel;
+            if (_traceSetting.DetailTraceSetting.TryGetValue(name, out sourceTraceLevel))
+            {
+                sourceSwitch = new SourceSwitch("GitHubActionsRunnerSubSwitch")
+                {
+                    Level = sourceTraceLevel.ToSourceLevels()
+                };
+            }
+            return new Tracing(name, _secretMasker, sourceSwitch, _hostTraceListener);
+        }
+    }
+}