Skip to content

Commit

Permalink
Trace API: give indicator of response success status (#3010)
Browse files Browse the repository at this point in the history
Signed-off-by: Jinbo Wang <jinbwan@microsoft.com>
  • Loading branch information
testforstephen committed Mar 21, 2023
1 parent 6dbd1bf commit 9240d7c
Show file tree
Hide file tree
Showing 2 changed files with 18 additions and 5 deletions.
19 changes: 14 additions & 5 deletions src/TracingLanguageClient.ts
Original file line number Diff line number Diff line change
Expand Up @@ -19,12 +19,12 @@ export class TracingLanguageClient extends LanguageClient {
const startAt: number = performance.now();
return super.start().then(value => {
if (isFirstTimeStart) {
this.fireTraceEvent("initialize", startAt);
this.fireTraceEvent("initialize", startAt, undefined);
}
return value;
}, reason => {
if (isFirstTimeStart) {
this.fireTraceEvent("initialize", startAt, reason);
this.fireTraceEvent("initialize", startAt, undefined, reason);
}
throw reason;
});
Expand All @@ -45,10 +45,10 @@ export class TracingLanguageClient extends LanguageClient {
const startAt: number = performance.now();
const requestType: string = this.getRequestType(method, ...args);
return this.sendRequest0(method, ...args).then(value => {
this.fireTraceEvent(requestType, startAt);
this.fireTraceEvent(requestType, startAt, this.getResultLength(value));
return value;
}, reason => {
this.fireTraceEvent(requestType, startAt, reason);
this.fireTraceEvent(requestType, startAt, undefined, reason);
throw reason;
});
}
Expand Down Expand Up @@ -88,12 +88,21 @@ export class TracingLanguageClient extends LanguageClient {
return requestType;
}

private fireTraceEvent(type: string, startAt: number, reason?: any): void {
private fireTraceEvent(type: string, startAt: number, resultLength: number | undefined, reason?: any): void {
const duration: number = performance.now() - startAt;
requestEventEmitter.fire({
type,
duration,
error: reason,
resultLength,
});
}

private getResultLength(value: any): number | undefined {
if (!value) {
return 0;
}

return value?.length ?? value?.items?.length;
}
}
4 changes: 4 additions & 0 deletions src/extension.api.ts
Original file line number Diff line number Diff line change
Expand Up @@ -86,6 +86,10 @@ export interface TraceEvent {
* Error that occurs while processing a request.
*/
error?: any;
/**
* The number of results returned by a response.
*/
resultLength?: number | undefined;
}

export const extensionApiVersion = '0.8';
Expand Down

0 comments on commit 9240d7c

Please sign in to comment.