add telemetry to autocomplete providers

Author: Deep Furiya

src/autocomplete/ConditionCompletionProvider.ts

@@ -4,6 +4,7 @@ import { TopLevelSection } from '../context/ContextType';
 import { getEntityMap } from '../context/SectionContextBuilder';
 import { SyntaxTreeManager } from '../context/syntaxtree/SyntaxTreeManager';
 import { LoggerFactory } from '../telemetry/LoggerFactory';
+import { Measure } from '../telemetry/TelemetryDecorator';
 import { getFuzzySearchFunction } from '../utils/FuzzySearchUtil';
 import { CompletionProvider } from './CompletionProvider';
 import { createCompletionItem } from './CompletionUtils';
@@ -21,6 +22,7 @@ export class ConditionCompletionProvider implements CompletionProvider {
 
     public constructor(private readonly syntaxTreeManager: SyntaxTreeManager) {}
 
+    @Measure({ name: 'getConditionCompletions' })
     public getCompletions(context: Context, params: CompletionParams): CompletionItem[] | undefined {
         const syntaxTree = this.syntaxTreeManager.getSyntaxTree(params.textDocument.uri);
         if (!syntaxTree) {

src/autocomplete/IntrinsicFunctionArgumentCompletionProvider.ts

@@ -10,6 +10,7 @@ import { SyntaxTreeManager } from '../context/syntaxtree/SyntaxTreeManager';
 import { DocumentManager } from '../document/DocumentManager';
 import { SchemaRetriever } from '../schema/SchemaRetriever';
 import { LoggerFactory } from '../telemetry/LoggerFactory';
+import { Measure } from '../telemetry/TelemetryDecorator';
 import { getFuzzySearchFunction } from '../utils/FuzzySearchUtil';
 import {
     determineGetAttPosition,
@@ -55,6 +56,7 @@ export class IntrinsicFunctionArgumentCompletionProvider implements CompletionPr
         private readonly documentManager: DocumentManager,
     ) {}
 
+    @Measure({ name: 'getIntrinsicFunctionArgumentCompletions' })
     getCompletions(context: Context, params: CompletionParams): CompletionItem[] | undefined {
         const syntaxTree = this.syntaxTreeManager.getSyntaxTree(params.textDocument.uri);
         if (!syntaxTree) {

src/autocomplete/IntrinsicFunctionCompletionProvider.ts

@@ -2,6 +2,7 @@ import { CompletionItem, CompletionItemKind, CompletionParams, TextEdit } from '
 import { Context } from '../context/Context';
 import { Intrinsics } from '../context/ContextType';
 import { DocumentType } from '../document/Document';
+import { Measure } from '../telemetry/TelemetryDecorator';
 import { getFuzzySearchFunction } from '../utils/FuzzySearchUtil';
 import { CompletionProvider } from './CompletionProvider';
 import { createCompletionItem, createReplacementRange } from './CompletionUtils';
@@ -16,6 +17,7 @@ export class IntrinsicFunctionCompletionProvider implements CompletionProvider {
         ignoreLocation: false,
     });
 
+    @Measure({ name: 'getIntrinsicFunctionCompletions' })
     getCompletions(context: Context, params: CompletionParams): CompletionItem[] | undefined {
         const triggerChar = params.context?.triggerCharacter ?? '';
         const isYaml = context.documentType !== DocumentType.JSON;

src/autocomplete/ParameterTypeValueCompletionProvider.ts

@@ -2,6 +2,7 @@ import { CompletionItem, CompletionItemKind, CompletionParams } from 'vscode-lan
 import { Context } from '../context/Context';
 import { PARAMETER_TYPES } from '../context/semantic/parameter/ParameterType';
 import { LoggerFactory } from '../telemetry/LoggerFactory';
+import { Measure } from '../telemetry/TelemetryDecorator';
 import { getFuzzySearchFunction } from '../utils/FuzzySearchUtil';
 import { CompletionProvider } from './CompletionProvider';
 import { createCompletionItem } from './CompletionUtils';
@@ -17,6 +18,7 @@ export class ParameterTypeValueCompletionProvider implements CompletionProvider
         ignoreLocation: false,
     });
 
+    @Measure({ name: 'getParameterTypeValueCompletions' })
     public getCompletions(context: Context, _: CompletionParams): CompletionItem[] | undefined {
         const items = this.getParameterTypesAsCompletionItems();
 

src/autocomplete/ResourcePropertyCompletionProvider.ts

@@ -28,6 +28,7 @@ import { CommonNodeTypes } from '../context/syntaxtree/utils/TreeSitterTypes';
 import { propertyTypesToMarkdown } from '../hover/HoverFormatter';
 import { PropertyType, ResourceSchema } from '../schema/ResourceSchema';
 import { SchemaRetriever } from '../schema/SchemaRetriever';
+import { Measure } from '../telemetry/TelemetryDecorator';
 import { getFuzzySearchFunction } from '../utils/FuzzySearchUtil';
 import { templatePathToJsonPointerPath } from '../utils/PathUtils';
 import { CompletionItemData, ExtendedCompletionItem } from './CompletionFormatter';
@@ -47,6 +48,7 @@ export class ResourcePropertyCompletionProvider implements CompletionProvider {
 
     constructor(private readonly schemaRetriever: SchemaRetriever) {}
 
+    @Measure({ name: 'getResourcePropertyCompletions' })
     getCompletions(context: Context, _params: CompletionParams): CompletionItem[] | undefined {
         // Use unified property completion method for all scenarios
         const { completions: propertyCompletions, skipFuzzySearch } = this.getPropertyCompletions(context);

src/autocomplete/ResourceSectionCompletionProvider.ts

@@ -6,6 +6,7 @@ import { CfnExternal } from '../server/CfnExternal';
 import { CfnInfraCore } from '../server/CfnInfraCore';
 import { CfnLspProviders } from '../server/CfnLspProviders';
 import { LoggerFactory } from '../telemetry/LoggerFactory';
+import { Measure } from '../telemetry/TelemetryDecorator';
 import { CompletionProvider } from './CompletionProvider';
 import { ResourceEntityCompletionProvider } from './ResourceEntityCompletionProvider';
 import { ResourcePropertyCompletionProvider } from './ResourcePropertyCompletionProvider';
@@ -29,6 +30,7 @@ export class ResourceSectionCompletionProvider implements CompletionProvider {
         private readonly resourceProviders = createResourceCompletionProviders(core, external, providers),
     ) {}
 
+    @Measure({ name: 'getResourceSectionCompletions' })
     getCompletions(
         context: Context,
         params: CompletionParams,

src/autocomplete/ResourceTypeCompletionProvider.ts

@@ -1,6 +1,7 @@
 import { CompletionItem, CompletionItemKind, CompletionParams, TextEdit } from 'vscode-languageserver';
 import { Context } from '../context/Context';
 import { SchemaRetriever } from '../schema/SchemaRetriever';
+import { Measure } from '../telemetry/TelemetryDecorator';
 import { getFuzzySearchFunction } from '../utils/FuzzySearchUtil';
 import { CompletionProvider } from './CompletionProvider';
 import { createCompletionItem, createReplacementRange } from './CompletionUtils';
@@ -10,6 +11,7 @@ export class ResourceTypeCompletionProvider implements CompletionProvider {
 
     constructor(private readonly schemaRetriever: SchemaRetriever) {}
 
+    @Measure({ name: 'getResourceTypeCompletions' })
     getCompletions(context: Context, _params: CompletionParams): CompletionItem[] | undefined {
         const resourceTypeCompletions = this.getResourceTypeCompletions(context);
         return this.fuzzySearch(resourceTypeCompletions, context.text);

src/autocomplete/TopLevelSectionCompletionProvider.ts

@@ -5,6 +5,7 @@ import { SyntaxTreeManager } from '../context/syntaxtree/SyntaxTreeManager';
 import { DocumentType } from '../document/Document';
 import { DocumentManager } from '../document/DocumentManager';
 import { LoggerFactory } from '../telemetry/LoggerFactory';
+import { Measure } from '../telemetry/TelemetryDecorator';
 import { getFuzzySearchFunction } from '../utils/FuzzySearchUtil';
 import { applySnippetIndentation } from '../utils/IndentationUtils';
 import { CompletionFormatter, ExtendedCompletionItem } from './CompletionFormatter';
@@ -84,6 +85,7 @@ ${CompletionFormatter.getIndentPlaceholder(1)}\${1:ConditionName}: $2`,
         private readonly documentManager: DocumentManager,
     ) {}
 
+    @Measure({ name: 'getTopLevelSectionCompletions' })
     getCompletions(context: Context, params: CompletionParams): CompletionItem[] | undefined {
         // Get both regular and snippet completions
         const stringCompletions = this.getTopLevelSectionCompletions();