Skip to content

Commit bfd35b1

Browse files
committed
Fix telemetry service calls
1 parent 7cdee0c commit bfd35b1

File tree

1 file changed

+7
-8
lines changed

1 file changed

+7
-8
lines changed

src/handlers/StackHandler.ts

Lines changed: 7 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -53,13 +53,12 @@ import { handleLspError } from '../utils/Errors';
5353
import { parseWithPrettyError } from '../utils/ZodErrorWrapper';
5454

5555
const log = LoggerFactory.getLogger('StackHandler');
56-
const telemetry = TelemetryService.instance.get('StackHandler');
5756

5857
export function getParametersHandler(
5958
components: ServerComponents,
6059
): RequestHandler<TemplateUri, GetParametersResult, void> {
6160
return (rawParams) => {
62-
return telemetry.measure('getParameters', () => {
61+
return TelemetryService.instance.get('StackHandler').measure('getParameters', () => {
6362
try {
6463
const params = parseWithPrettyError(parseTemplateUriParams, rawParams);
6564
const syntaxTree = components.syntaxTreeManager.getSyntaxTree(params);
@@ -87,7 +86,7 @@ export function createValidationHandler(
8786
components: ServerComponents,
8887
): RequestHandler<CreateValidationParams, CreateStackActionResult, void> {
8988
return async (rawParams) => {
90-
return await telemetry.measureAsync('createValidation', async () => {
89+
return await TelemetryService.instance.get('StackHandler').measureAsync('createValidation', async () => {
9190
try {
9291
const params = parseWithPrettyError(parseCreateValidationParams, rawParams);
9392
return await components.validationWorkflowService.start(params);
@@ -102,7 +101,7 @@ export function createDeploymentHandler(
102101
components: ServerComponents,
103102
): RequestHandler<CreateDeploymentParams, CreateStackActionResult, void> {
104103
return async (rawParams) => {
105-
return await telemetry.measureAsync('createDeployment', async () => {
104+
return await TelemetryService.instance.get('StackHandler').measureAsync('createDeployment', async () => {
106105
try {
107106
const params = parseWithPrettyError(parseCreateDeploymentParams, rawParams);
108107
return await components.deploymentWorkflowService.start(params);
@@ -169,7 +168,7 @@ export function deleteChangeSetHandler(
169168
components: ServerComponents,
170169
): RequestHandler<DeleteChangeSetParams, CreateStackActionResult, void> {
171170
return async (rawParams) => {
172-
return await telemetry.measureAsync('deleteChangeSet', async () => {
171+
return await TelemetryService.instance.get('StackHandler').measureAsync('deleteChangeSet', async () => {
173172
try {
174173
const params = parseWithPrettyError(parseDeleteChangeSetParams, rawParams);
175174
return await components.changeSetDeletionWorkflowService.start(params);
@@ -210,7 +209,7 @@ export function getCapabilitiesHandler(
210209
components: ServerComponents,
211210
): RequestHandler<TemplateUri, GetCapabilitiesResult, void> {
212211
return async (rawParams) => {
213-
return await telemetry.measureAsync('getCapabilities', async () => {
212+
return await TelemetryService.instance.get('StackHandler').measureAsync('getCapabilities', async () => {
214213
try {
215214
const params = parseWithPrettyError(parseTemplateUriParams, rawParams);
216215
const document = components.documentManager.get(params);
@@ -299,7 +298,7 @@ export function listStacksHandler(
299298
components: ServerComponents,
300299
): RequestHandler<ListStacksParams, ListStacksResult, void> {
301300
return async (params: ListStacksParams): Promise<ListStacksResult> => {
302-
return await telemetry.measureAsync('listStacks', async () => {
301+
return await TelemetryService.instance.get('StackHandler').measureAsync('listStacks', async () => {
303302
try {
304303
if (params.statusToInclude?.length && params.statusToExclude?.length) {
305304
throw new Error('Cannot specify both statusToInclude and statusToExclude');
@@ -321,7 +320,7 @@ export function listChangeSetsHandler(
321320
components: ServerComponents,
322321
): RequestHandler<ListChangeSetParams, ListChangeSetResult, void> {
323322
return async (params: ListChangeSetParams): Promise<ListChangeSetResult> => {
324-
return await telemetry.measureAsync('listChangeSets', async () => {
323+
return await TelemetryService.instance.get('StackHandler').measureAsync('listChangeSets', async () => {
325324
try {
326325
const result = await components.cfnService.listChangeSets(params.stackName, params.nextToken);
327326
return {

0 commit comments

Comments
 (0)