Cache the MultiTablePrettyfier to make subsequent document, selection and command formatting faster

Author: darkriszty

src/extension/extension.ts

@@ -1,11 +1,20 @@
 'use strict';
 import * as vscode from 'vscode';
-import { getSupportLanguageIds, getDocumentRangePrettyfier, getDocumentPrettyfier, getDocumentPrettyfierCommand } from './prettyfierFactory';
+import { getSupportLanguageIds, getDocumentRangePrettyfier, getDocumentPrettyfier, getDocumentPrettyfierCommand, invalidateCache } from './prettyfierFactory';
 
 // This method is called when the extension is activated.
 // The extension is activated the very first time the command is executed.
 export function activate(context: vscode.ExtensionContext): void {
 
+    // Invalidate cache when configuration changes
+    context.subscriptions.push(
+        vscode.workspace.onDidChangeConfiguration(event => {
+            if (event.affectsConfiguration("markdownTablePrettify")) {
+                invalidateCache();
+            }
+        })
+    );
+
     const supportedLanguageIds = getSupportLanguageIds();
     for (let language of supportedLanguageIds) {
         context.subscriptions.push(

src/extension/prettyfierFactory.ts

@@ -24,6 +24,12 @@ import { SingleTablePrettyfier } from '../prettyfiers/singleTablePrettyfier';
 import { TableStringWriter } from "../writers/tableStringWriter";
 import { ValuePaddingProvider } from '../writers/valuePaddingProvider';
 
+let cachedMultiTablePrettyfier: MultiTablePrettyfier | null = null;
+
+export function invalidateCache() {
+    cachedMultiTablePrettyfier = null;
+}
+
 export function getSupportLanguageIds() {
     return [ "markdown", ...getConfigurationValue<Array<string>>("extendedLanguages", []) ];
 }
@@ -41,15 +47,21 @@ export function getDocumentPrettyfierCommand(): TableDocumentPrettyfierCommand {
 }
 
 function getMultiTablePrettyfier(): MultiTablePrettyfier {
+    if (cachedMultiTablePrettyfier) {
+        return cachedMultiTablePrettyfier;
+    }
+
     const loggers = getLoggers();
     const sizeLimitCheker = getSizeLimitChecker(loggers);
     const columnPadding: number = getConfigurationValue<number>("columnPadding", 0);
 
-    return new MultiTablePrettyfier(
+    cachedMultiTablePrettyfier = new MultiTablePrettyfier(
         new TableFinder(new TableValidator(new SelectionInterpreter(true))),
         getSingleTablePrettyfier(loggers, sizeLimitCheker, columnPadding),
         sizeLimitCheker
     );
+
+    return cachedMultiTablePrettyfier;
 }
 
 function getSingleTablePrettyfier(loggers: ILogger[], sizeLimitCheker: ConfigSizeLimitChecker, columnPadding: number): SingleTablePrettyfier {