From d6497d3db17303db96044f5453141aa20bb3df6d Mon Sep 17 00:00:00 2001 From: Matt Bierner Date: Thu, 23 Jun 2022 18:00:05 -0700 Subject: [PATCH] Use ResourceMap (#153051) Replaces `Set` of uris with a `ResourceMap` --- extensions/markdown-language-features/src/workspace.ts | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/extensions/markdown-language-features/src/workspace.ts b/extensions/markdown-language-features/src/workspace.ts index f059ddd0d4d..9fdef64076d 100644 --- a/extensions/markdown-language-features/src/workspace.ts +++ b/extensions/markdown-language-features/src/workspace.ts @@ -61,7 +61,7 @@ export class VsCodeMdWorkspace extends Disposable implements IMdWorkspace { async getAllMarkdownDocuments(): Promise { const maxConcurrent = 20; - const foundFiles = new Set(); + const foundFiles = new ResourceMap(); const limiter = new Limiter(maxConcurrent); // Add files on disk @@ -70,7 +70,7 @@ export class VsCodeMdWorkspace extends Disposable implements IMdWorkspace { return limiter.queue(async () => { const doc = await this.getOrLoadMarkdownDocument(resource); if (doc) { - foundFiles.add(doc.uri.toString()); + foundFiles.set(resource); } return doc; }); @@ -78,7 +78,7 @@ export class VsCodeMdWorkspace extends Disposable implements IMdWorkspace { // Add opened files (such as untitled files) const openTextDocumentResults = await Promise.all(vscode.workspace.textDocuments - .filter(doc => !foundFiles.has(doc.uri.toString()) && this.isRelevantMarkdownDocument(doc))); + .filter(doc => !foundFiles.has(doc.uri) && this.isRelevantMarkdownDocument(doc))); return coalesce([...onDiskResults, ...openTextDocumentResults]); }