Allow to cancel resolveMainMethod codelens command

Author: testforstephen

src/debugCodeLensProvider.ts

@@ -102,7 +102,7 @@ class DebugCodeLensProvider implements vscode.CodeLensProvider {
 
     public async provideCodeLenses(document: vscode.TextDocument, token: vscode.CancellationToken): Promise<vscode.CodeLens[]> {
         try {
-            const mainMethods: IMainMethod[] = await resolveMainMethod(document.uri);
+            const mainMethods: IMainMethod[] = await resolveMainMethod(document.uri, token);
             return _.flatten(mainMethods.map((method) => {
                 return [
                     new vscode.CodeLens(method.range, {

src/languageServerPlugin.ts

@@ -33,8 +33,8 @@ export interface ILaunchValidationResponse {
     readonly proposals?: IMainClassOption[];
 }
 
-export async function resolveMainMethod(uri: vscode.Uri): Promise<IMainMethod[]> {
-    return <IMainMethod[]> await commands.executeJavaLanguageServerCommand(commands.JAVA_RESOLVE_MAINMETHOD, uri.toString());
+export async function resolveMainMethod(uri: vscode.Uri, token: vscode.CancellationToken): Promise<IMainMethod[]> {
+    return <IMainMethod[]> await commands.executeJavaLanguageServerCommand(commands.JAVA_RESOLVE_MAINMETHOD, uri.toString(), token);
 }
 
 export function startDebugSession() {
@@ -108,3 +108,7 @@ export function fetchPlatformSettings(): any {
 export async function resolveClassFilters(patterns: string[]): Promise<string[]> {
     return <string[]> await commands.executeJavaLanguageServerCommand(commands.JAVA_RESOLVE_CLASSFILTERS, ...patterns);
 }
+
+export async function resolveSourceUri(sourceLine: string): Promise<string> {
+    return <string> await commands.executeJavaLanguageServerCommand(commands.JAVA_RESOLVE_SOURCEURI, sourceLine);
+}