development/vscode
development/vscode
1
0
Fork 0
mirror of https://github.com/Microsoft/vscode synced 2024-08-27 21:09:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

Pass through CancellationToken

Browse source
This commit is contained in:
Matt Bierner 2022-04-05 11:56:07 -07:00
parent 0496c2b3a7
commit 2154f3d008
No known key found for this signature in database
GPG key ID: 099C331567E11888
1 changed files with 9 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
extensions/markdown-language-features/src/languageFeatures/references.ts
View file

@ -104,7 +104,7 @@ export class MdReferencesProvider extends Disposable implements vscode.Reference
if (header) {
return this.getReferencesToHeader(document, header);
} else {
return this.getReferencesToLinkAtPosition(document, position);
return this.getReferencesToLinkAtPosition(document, position, token);
}
}
@ -141,7 +141,7 @@ export class MdReferencesProvider extends Disposable implements vscode.Reference
return references;
}
private async getReferencesToLinkAtPosition(document: SkinnyTextDocument, position: vscode.Position): Promise<MdReference[]> {
private async getReferencesToLinkAtPosition(document: SkinnyTextDocument, position: vscode.Position, token: vscode.CancellationToken): Promise<MdReference[]> {
const docLinks = await this.linkProvider.getAllLinks(document);
for (const link of docLinks) {
@ -150,11 +150,11 @@ export class MdReferencesProvider extends Disposable implements vscode.Reference
if (link.ref.range.contains(position)) {
return Array.from(this.getReferencesToLinkReference(docLinks, link.ref.text, { resource: document.uri, range: link.ref.range }));
} else if (link.source.hrefRange.contains(position)) {
return this.getReferencesToLink(link);
return this.getReferencesToLink(link, token);
}
} else {
if (link.source.hrefRange.contains(position)) {
return this.getReferencesToLink(link);
return this.getReferencesToLink(link, token);
}
}
}
@ -162,8 +162,11 @@ export class MdReferencesProvider extends Disposable implements vscode.Reference
return [];
}
private async getReferencesToLink(sourceLink: MdLink): Promise<MdReference[]> {
private async getReferencesToLink(sourceLink: MdLink, token: vscode.CancellationToken): Promise<MdReference[]> {
const allLinksInWorkspace = (await this._linkCache.getAll()).flat();
if (token.isCancellationRequested) {
return [];
}
if (sourceLink.href.kind === 'reference') {
return Array.from(this.getReferencesToLinkReference(allLinksInWorkspace, sourceLink.href.ref, { resource: sourceLink.source.resource, range: sourceLink.source.hrefRange }));
@ -182,7 +185,7 @@ export class MdReferencesProvider extends Disposable implements vscode.Reference
}
}
if (!targetDoc) {
if (!targetDoc || token.isCancellationRequested) {
return [];
}