Add multiMapAdd helper

Andy Hanson · Andy Hanson · commit 949c517fdd15 · 2016-08-19T07:15:19.000-07:00
diff --git a/src/compiler/core.ts b/src/compiler/core.ts
@@ -566,6 +566,21 @@ namespace ts {
         return result;
     }
 
+    /**
+     * Adds the value to an array of values associated with the key, and returns the array.
+     * Creates the array if it does not already exist.
+     */
+    export function multiMapAdd<V>(map: Map<V[]>, key: string, value: V): V[] {
+        const values = map[key];
+        if (values) {
+            values.push(value);
+            return values;
+        }
+        else {
+            return map[key] = [value];
+        }
+    }
+
     /**
      * Tests whether a value is an array.
      */
diff --git a/src/compiler/emitter.ts b/src/compiler/emitter.ts
@@ -6847,7 +6847,7 @@ const _super = (function (geti, seti) {
                                 // export { x, y }
                                 for (const specifier of (<ExportDeclaration>node).exportClause.elements) {
                                     const name = (specifier.propertyName || specifier.name).text;
-                                    (exportSpecifiers[name] || (exportSpecifiers[name] = [])).push(specifier);
+                                    multiMapAdd(exportSpecifiers, name, specifier);
                                 }
                             }
                             break;
diff --git a/src/compiler/sys.ts b/src/compiler/sys.ts
@@ -267,7 +267,7 @@ namespace ts {
                 }
 
                 function addFileWatcherCallback(filePath: string, callback: FileWatcherCallback): void {
-                    (fileWatcherCallbacks[filePath] || (fileWatcherCallbacks[filePath] = [])).push(callback);
+                    multiMapAdd(fileWatcherCallbacks, filePath, callback);
                 }
 
                 function addFile(fileName: string, callback: FileWatcherCallback): WatchedFile {
diff --git a/src/harness/fourslash.ts b/src/harness/fourslash.ts
@@ -1635,8 +1635,7 @@ namespace FourSlash {
             const result = ts.createMap<Range[]>();
             for (const range of this.getRanges()) {
                 const text = this.rangeText(range);
-                const ranges = result[text] || (result[text] = []);
-                ranges.push(range);
+                ts.multiMapAdd(result, text, range);
             }
             return result;
         }
diff --git a/src/harness/unittests/tsserverProjectSystem.ts b/src/harness/unittests/tsserverProjectSystem.ts
@@ -198,8 +198,7 @@ namespace ts {
 
         watchDirectory(directoryName: string, callback: DirectoryWatcherCallback, recursive: boolean): DirectoryWatcher {
             const path = this.toPath(directoryName);
-            const callbacks = this.watchedDirectories[path] || (this.watchedDirectories[path] = []);
-            callbacks.push({ cb: callback, recursive });
+            const callbacks = multiMapAdd(this.watchedDirectories, path, { cb: callback, recursive });
             return {
                 referenceCount: 0,
                 directoryName,
@@ -239,8 +238,7 @@ namespace ts {
 
         watchFile(fileName: string, callback: FileWatcherCallback) {
             const path = this.toPath(fileName);
-            const callbacks = this.watchedFiles[path] || (this.watchedFiles[path] = []);
-            callbacks.push(callback);
+            const callbacks = multiMapAdd(this.watchedFiles, path, callback);
             return {
                 close: () => {
                     const i = callbacks.indexOf(callback);
diff --git a/src/services/services.ts b/src/services/services.ts
@@ -984,8 +984,7 @@ namespace ts {
             function addDeclaration(declaration: Declaration) {
                 const name = getDeclarationName(declaration);
                 if (name) {
-                    const declarations = getDeclarations(name);
-                    declarations.push(declaration);
+                    multiMapAdd(result, name, declaration);
                 }
             }
 

Original file line number	Diff line number	Diff line change
`@@ -6847,7 +6847,7 @@ const _super = (function (geti, seti) {`
`6847`	`6847`	`// export { x, y }`
`6848`	`6848`	`for (const specifier of (<ExportDeclaration>node).exportClause.elements) {`
`6849`	`6849`	`const name = (specifier.propertyName \|\| specifier.name).text;`
`6850`		`- (exportSpecifiers[name] \|\| (exportSpecifiers[name] = [])).push(specifier);`
	`6850`	`+ multiMapAdd(exportSpecifiers, name, specifier);`
`6851`	`6851`	`}`
`6852`	`6852`	`}`
`6853`	`6853`	`break;`
Original file line number	Diff line number	Diff line change
`@@ -267,7 +267,7 @@ namespace ts {`
`267`	`267`	`}`
`268`	`268`
`269`	`269`	`function addFileWatcherCallback(filePath: string, callback: FileWatcherCallback): void {`
`270`		`- (fileWatcherCallbacks[filePath] \|\| (fileWatcherCallbacks[filePath] = [])).push(callback);`
	`270`	`+ multiMapAdd(fileWatcherCallbacks, filePath, callback);`
`271`	`271`	`}`
`272`	`272`
`273`	`273`	`function addFile(fileName: string, callback: FileWatcherCallback): WatchedFile {`
Original file line number	Diff line number	Diff line change
`@@ -1635,8 +1635,7 @@ namespace FourSlash {`
`1635`	`1635`	`const result = ts.createMap<Range[]>();`
`1636`	`1636`	`for (const range of this.getRanges()) {`
`1637`	`1637`	`const text = this.rangeText(range);`
`1638`		`- const ranges = result[text] \|\| (result[text] = []);`
`1639`		`- ranges.push(range);`
	`1638`	`+ ts.multiMapAdd(result, text, range);`
`1640`	`1639`	`}`
`1641`	`1640`	`return result;`
`1642`	`1641`	`}`
Original file line number	Diff line number	Diff line change
`@@ -984,8 +984,7 @@ namespace ts {`
`984`	`984`	`function addDeclaration(declaration: Declaration) {`
`985`	`985`	`const name = getDeclarationName(declaration);`
`986`	`986`	`if (name) {`
`987`		`- const declarations = getDeclarations(name);`
`988`		`- declarations.push(declaration);`
	`987`	`+ multiMapAdd(result, name, declaration);`
`989`	`988`	`}`
`990`	`989`	`}`
`991`	`990`