forked from redhat-developer/vscode-java
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathTracingLanguageClient.ts
More file actions
146 lines (129 loc) · 4.69 KB
/
TracingLanguageClient.ts
File metadata and controls
146 lines (129 loc) · 4.69 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
import { performance } from "perf_hooks";
import { Event, EventEmitter } from "vscode";
import { CancellationToken, LanguageClient, LanguageClientOptions, ProtocolRequestType, ProtocolRequestType0, RequestType, RequestType0, ServerOptions } from "vscode-languageclient/node";
import { TraceEvent } from "./extension.api";
const requestStartEventEmitter = new EventEmitter<TraceEvent>();
const requestEndEventEmitter = new EventEmitter<TraceEvent>();
export const onWillRequestStart: Event<TraceEvent> = requestStartEventEmitter.event;
export const onDidRequestEnd: Event<TraceEvent> = requestEndEventEmitter.event;
export class TracingLanguageClient extends LanguageClient {
private isStarted: boolean = false;
private isSyntaxServer: boolean = false;
constructor(id: string, name: string, serverOptions: ServerOptions, clientOptions: LanguageClientOptions, forceDebug?: boolean) {
super(id, name, serverOptions, clientOptions, forceDebug);
this.isSyntaxServer = name?.includes("Syntax Server");
}
start(): Promise<void> {
const isFirstTimeStart: boolean = !this.isStarted;
this.isStarted = true;
const startAt: number = performance.now();
if (isFirstTimeStart) {
this.fireRequestStartTraceEvent("initialize");
}
return super.start().then(value => {
if (isFirstTimeStart) {
this.fireSuccessTraceEvent("initialize", startAt, undefined);
}
return value;
}, reason => {
if (isFirstTimeStart) {
this.fireFailureTraceEvent("initialize", startAt, reason);
}
throw reason;
});
}
stop(timeout?: number): Promise<void> {
this.isStarted = false;
return super.stop(timeout);
}
sendRequest<R, PR, E, RO>(type: ProtocolRequestType0<R, PR, E, RO>, token?: CancellationToken): Promise<R>;
sendRequest<P, R, PR, E, RO>(type: ProtocolRequestType<P, R, PR, E, RO>, params: P, token?: CancellationToken): Promise<R>;
sendRequest<R, E>(type: RequestType0<R, E>, token?: CancellationToken): Promise<R>;
sendRequest<P, R, E>(type: RequestType<P, R, E>, params: P, token?: CancellationToken): Promise<R>;
sendRequest<R>(method: string, token?: CancellationToken): Promise<R>;
sendRequest<R>(method: string, param: any, token?: CancellationToken): Promise<R>;
sendRequest(method: any, ...args) {
const startAt: number = performance.now();
const requestType: string = this.getRequestType(method, ...args);
let data: any = undefined;
if (args?.[0]?.context?.triggerKind !== undefined) {
data = {
triggerKind: args[0].context.triggerKind,
triggerCharacter: args[0].context.triggerCharacter,
};
}
this.fireRequestStartTraceEvent(requestType);
return this.sendRequest0(method, ...args).then((value: any) => {
if (data && value?.itemDefaults?.data?.completionKinds) {
// Include the completionKinds from the completion response.
data.completionKinds = value.itemDefaults.data.completionKinds;
}
this.fireSuccessTraceEvent(requestType, startAt, this.getResultLength(value), data);
return value;
}, reason => {
this.fireFailureTraceEvent(requestType, startAt, reason, data);
throw reason;
});
}
private sendRequest0(method: any, ...args) {
if (!args || !args.length) {
return super.sendRequest(method);
}
const first = args[0];
const last = args[args.length - 1];
if (CancellationToken.is(last)) {
if (first === last) {
return super.sendRequest(method, last);
} else {
return super.sendRequest(method, first, last);
}
}
return super.sendRequest(method, first);
}
private getRequestType(method: any, ...args): string {
let requestType: string;
if (typeof method === 'string' || method instanceof String) {
requestType = String(method);
} else {
requestType = method?.method;
}
if (requestType === "workspace/executeCommand") {
if (args?.[0]?.command) {
requestType = `workspace/executeCommand/${args[0].command}`;
}
}
return requestType;
}
private fireRequestStartTraceEvent(type: string): void {
requestStartEventEmitter.fire({
type,
fromSyntaxServer: !!this.isSyntaxServer,
});
}
private fireSuccessTraceEvent(type: string, startAt: number, resultLength: number | undefined, data?: any): void {
const duration: number = performance.now() - startAt;
requestEndEventEmitter.fire({
type,
duration,
resultLength,
data,
fromSyntaxServer: !!this.isSyntaxServer,
});
}
private fireFailureTraceEvent(type: string, startAt: number, error: any, data?: any): void {
const duration: number = performance.now() - startAt;
requestEndEventEmitter.fire({
type,
duration,
error,
data,
fromSyntaxServer: !!this.isSyntaxServer,
});
}
private getResultLength(value: any): number | undefined {
if (!value) {
return 0;
}
return value?.length ?? value?.items?.length;
}
}