[Nullable Context] Microsoft.Diagnostics.Monitoring.WebApi fixes

src/Microsoft.Diagnostics.Monitoring.WebApi/Controllers/DiagController.cs

@@ -35,9 +35,7 @@ public partial class DiagController : DiagnosticsControllerBase
     {
         private const TraceProfile DefaultTraceProfiles = TraceProfile.Cpu | TraceProfile.Http | TraceProfile.Metrics | TraceProfile.GcCollect;
 
-#nullable disable
         private readonly IOptions<DiagnosticPortOptions> _diagnosticPortOptions;
-#nullable restore
         private readonly IOptions<CallStacksOptions> _callStacksOptions;
         private readonly IOptions<ParameterCapturingOptions> _parameterCapturingOptions;
         private readonly IOptionsMonitor<GlobalCounterOptions> _counterOptions;
@@ -54,7 +52,7 @@ public partial class DiagController : DiagnosticsControllerBase
         public DiagController(IServiceProvider serviceProvider, ILogger<DiagController> logger)
             : base(serviceProvider.GetRequiredService<IDiagnosticServices>(), serviceProvider.GetRequiredService<IEgressOperationStore>(), logger)
         {
-            _diagnosticPortOptions = serviceProvider.GetService<IOptions<DiagnosticPortOptions>>();
+            _diagnosticPortOptions = serviceProvider.GetRequiredService<IOptions<DiagnosticPortOptions>>();
             _callStacksOptions = serviceProvider.GetRequiredService<IOptions<CallStacksOptions>>();
             _parameterCapturingOptions = serviceProvider.GetRequiredService<IOptions<ParameterCapturingOptions>>();
             _counterOptions = serviceProvider.GetRequiredService<IOptionsMonitor<GlobalCounterOptions>>();

src/Microsoft.Diagnostics.Monitoring.WebApi/Controllers/MetricsController.cs

@@ -21,17 +21,15 @@ public class MetricsController : ControllerBase
         private const string ArtifactType_Metrics = "metrics";
 
         private readonly ILogger<MetricsController> _logger;
-#nullable disable
         private readonly MetricsStoreService _metricsStore;
-#nullable restore
         private readonly MetricsOptions _metricsOptions;
 
         public MetricsController(ILogger<MetricsController> logger,
             IServiceProvider serviceProvider,
             IOptions<MetricsOptions> metricsOptions)
         {
             _logger = logger;
-            _metricsStore = serviceProvider.GetService<MetricsStoreService>();
+            _metricsStore = serviceProvider.GetRequiredService<MetricsStoreService>();
             _metricsOptions = metricsOptions.Value;
         }
 

src/Microsoft.Diagnostics.Monitoring.WebApi/TaskCompletionSourceExtensions.cs

@@ -1,8 +1,6 @@
 // Licensed to the .NET Foundation under one or more agreements.
 // The .NET Foundation licenses this file to you under the MIT license.
 
-#nullable disable
-
 using System.Threading;
 using System.Threading.Tasks;
 
@@ -22,15 +20,15 @@ internal static class TaskCompletionSourceExtensions
         /// </remarks>
         public static async Task<T> WithCancellation<T>(this TaskCompletionSource<T> source, CancellationToken token)
         {
-            using (token.Register(source => ((TaskCompletionSource<T>)source).TrySetCanceled(token), source))
+            using (token.Register(() => source.TrySetCanceled(token)))
             {
                 return await source.Task.ConfigureAwait(false);
             }
         }
 
         public static async Task WithCancellation(this TaskCompletionSource source, CancellationToken token)
         {
-            using (token.Register(source => ((TaskCompletionSource)source).TrySetCanceled(token), source))
+            using (token.Register(() => source.TrySetCanceled(token)))
             {
                 await source.Task.ConfigureAwait(false);
             }