|
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.
#nullable disable
using System;
using System.Collections.Generic;
using System.Collections.Immutable;
using System.Linq;
using System.Runtime.InteropServices;
using System.Threading;
using Microsoft.CodeAnalysis.Editor.Shared.Utilities;
using Microsoft.CodeAnalysis.Host;
using Microsoft.CodeAnalysis.Internal.Log;
using Microsoft.CodeAnalysis.Options;
using Microsoft.CodeAnalysis.Remote;
using Microsoft.VisualStudio.ComponentModelHost;
using Microsoft.VisualStudio.LanguageServices;
using Microsoft.VisualStudio.LanguageServices.Implementation;
using Microsoft.VisualStudio.LanguageServices.Implementation.Options;
namespace Roslyn.VisualStudio.DiagnosticsWindow.OptionsPages
{
[Guid(Guids.RoslynOptionPagePerformanceLoggersIdString)]
internal class PerformanceLoggersPage : AbstractOptionPage
{
private IGlobalOptionService _globalOptions;
private IThreadingContext _threadingContext;
private SolutionServices _workspaceServices;
protected override AbstractOptionPageControl CreateOptionPage(IServiceProvider serviceProvider, OptionStore optionStore)
{
if (_globalOptions == null)
{
var componentModel = (IComponentModel)serviceProvider.GetService(typeof(SComponentModel));
_globalOptions = componentModel.GetService<IGlobalOptionService>();
_threadingContext = componentModel.GetService<IThreadingContext>();
var workspace = componentModel.GetService<VisualStudioWorkspace>();
_workspaceServices = workspace.Services.SolutionServices;
}
return new InternalOptionsControl(FunctionIdOptions.GetOptions(), optionStore);
}
protected override void OnApply(PageApplyEventArgs e)
{
base.OnApply(e);
SetLoggers(_globalOptions, _threadingContext, _workspaceServices);
}
public static void SetLoggers(IGlobalOptionService globalOptions, IThreadingContext threadingContext, SolutionServices workspaceServices)
{
var loggerTypeNames = GetLoggerTypes(globalOptions).ToImmutableArray();
// update loggers in VS
var isEnabled = FunctionIdOptions.CreateFunctionIsEnabledPredicate(globalOptions);
SetRoslynLogger(loggerTypeNames, () => new EtwLogger(isEnabled));
SetRoslynLogger(loggerTypeNames, () => new TraceLogger(isEnabled));
SetRoslynLogger(loggerTypeNames, () => new OutputWindowLogger(isEnabled));
// update loggers in remote process
var client = threadingContext.JoinableTaskFactory.Run(() => RemoteHostClient.TryGetClientAsync(workspaceServices, CancellationToken.None));
if (client != null)
{
var functionIds = Enum.GetValues(typeof(FunctionId)).Cast<FunctionId>().Where(isEnabled).ToImmutableArray();
threadingContext.JoinableTaskFactory.Run(async () => _ = await client.TryInvokeAsync<IRemoteProcessTelemetryService>(
(service, cancellationToken) => service.EnableLoggingAsync(loggerTypeNames, functionIds, cancellationToken),
CancellationToken.None).ConfigureAwait(false));
}
}
private static IEnumerable<string> GetLoggerTypes(IGlobalOptionService globalOptions)
{
if (globalOptions.GetOption(LoggerOptionsStorage.EtwLoggerKey))
{
yield return nameof(EtwLogger);
}
if (globalOptions.GetOption(LoggerOptionsStorage.TraceLoggerKey))
{
yield return nameof(TraceLogger);
}
if (globalOptions.GetOption(LoggerOptionsStorage.OutputWindowLoggerKey))
{
yield return nameof(OutputWindowLogger);
}
}
private static void SetRoslynLogger<T>(ImmutableArray<string> loggerTypeNames, Func<T> creator) where T : ILogger
{
if (loggerTypeNames.Contains(typeof(T).Name))
{
Logger.SetLogger(AggregateLogger.AddOrReplace(creator(), Logger.GetLogger(), l => l is T));
}
else
{
Logger.SetLogger(AggregateLogger.Remove(Logger.GetLogger(), l => l is T));
}
}
}
}
|