Mark documents with lean4 in parallel.

leanprover · Mar 25, 2021 · 7bcb876 · 7bcb876
1 parent 234fe1b
commit 7bcb876
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/src/extension.ts b/src/extension.ts
@@ -46,10 +46,8 @@ export async function activate(context: ExtensionContext): Promise<any> {
         return api
     }
 
-    for (const textDocument of workspace.textDocuments) {
-        if (textDocument.languageId === 'lean')
-            await languages.setTextDocumentLanguage(textDocument, 'lean4')
-    }
+    await Promise.all(workspace.textDocuments.map(async (doc) =>
+        doc.languageId === 'lean' && languages.setTextDocumentLanguage(doc, 'lean4')))
 
     const client: LeanClient = new LeanClient()
     context.subscriptions.push(client)