Merge pull request #18025 from heejaechang/codelensfunc

add marker to see code lens perf easier

src/VisualStudio/Core/Next/CodeLens/RemoteCodeLensReferencesService.cs

@@ -7,10 +7,9 @@
 using Microsoft.CodeAnalysis;
 using Microsoft.CodeAnalysis.CodeLens;
 using Microsoft.CodeAnalysis.Host.Mef;
+using Microsoft.CodeAnalysis.Internal.Log;
 using Microsoft.CodeAnalysis.Remote;
 using Microsoft.CodeAnalysis.Remote.Diagnostics;
-using Microsoft.VisualStudio.LanguageServices.Implementation.Extensions;
-using Microsoft.VisualStudio.LanguageServices.Remote;
 
 namespace Microsoft.VisualStudio.LanguageServices.CodeLens
 {
@@ -19,6 +18,8 @@ internal sealed class RemoteCodeLensReferencesService : ICodeLensReferencesServi
     {
         public async Task<ReferenceCount> GetReferenceCountAsync(Solution solution, DocumentId documentId, SyntaxNode syntaxNode, int maxSearchResults,
             CancellationToken cancellationToken)
+        {
+            using (Logger.LogBlock(FunctionId.CodeLens_GetReferenceCountAsync, cancellationToken))
             {
                 if (syntaxNode == null)
                 {
@@ -37,9 +38,12 @@ internal sealed class RemoteCodeLensReferencesService : ICodeLensReferencesServi
                     solution, WellKnownServiceHubServices.CodeAnalysisService_GetReferenceCountAsync,
                     new object[] { new CodeLensArguments(documentId, syntaxNode), maxSearchResults }, cancellationToken).ConfigureAwait(false);
             }
+        }
 
         public async Task<IEnumerable<ReferenceLocationDescriptor>> FindReferenceLocationsAsync(Solution solution, DocumentId documentId, SyntaxNode syntaxNode,
             CancellationToken cancellationToken)
+        {
+            using (Logger.LogBlock(FunctionId.CodeLens_FindReferenceLocationsAsync, cancellationToken))
             {
                 if (syntaxNode == null)
                 {
@@ -58,9 +62,12 @@ internal sealed class RemoteCodeLensReferencesService : ICodeLensReferencesServi
                     solution, WellKnownServiceHubServices.CodeAnalysisService_FindReferenceLocationsAsync,
                     new CodeLensArguments(documentId, syntaxNode), cancellationToken).ConfigureAwait(false);
             }
+        }
 
         public async Task<IEnumerable<ReferenceMethodDescriptor>> FindReferenceMethodsAsync(Solution solution, DocumentId documentId, SyntaxNode syntaxNode,
             CancellationToken cancellationToken)
+        {
+            using (Logger.LogBlock(FunctionId.CodeLens_FindReferenceMethodsAsync, cancellationToken))
             {
                 if (syntaxNode == null)
                 {
@@ -79,9 +86,12 @@ internal sealed class RemoteCodeLensReferencesService : ICodeLensReferencesServi
                     solution, WellKnownServiceHubServices.CodeAnalysisService_FindReferenceMethodsAsync,
                     new CodeLensArguments(documentId, syntaxNode), cancellationToken).ConfigureAwait(false);
             }
+        }
 
         public async Task<string> GetFullyQualifiedName(Solution solution, DocumentId documentId, SyntaxNode syntaxNode,
             CancellationToken cancellationToken)
+        {
+            using (Logger.LogBlock(FunctionId.CodeLens_GetFullyQualifiedName, cancellationToken))
             {
                 if (syntaxNode == null)
                 {
@@ -101,4 +111,5 @@ internal sealed class RemoteCodeLensReferencesService : ICodeLensReferencesServi
                     new CodeLensArguments(documentId, syntaxNode), cancellationToken).ConfigureAwait(false);
             }
         }
+    }
 }

src/Workspaces/Core/Portable/Log/FunctionId.cs

@@ -366,5 +366,9 @@ internal enum FunctionId
         RemoteHost_Connect,
         RemoteHost_Disconnect,
         CodeAnalysisService_GetTodoCommentsAsync,
+        CodeLens_GetReferenceCountAsync,
+        CodeLens_FindReferenceLocationsAsync,
+        CodeLens_FindReferenceMethodsAsync,
+        CodeLens_GetFullyQualifiedName,
     }
 }