SQL Operations Studio Public Preview 1 (0.23) release source code

src/vs/base/common/collections.ts

@@ -0,0 +1,94 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the Source EULA. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+'use strict';
+
+
+/**
+ * An interface for a JavaScript object that
+ * acts a dictionary. The keys are strings.
+ */
+export interface IStringDictionary<V> {
+	[name: string]: V;
+}
+
+/**
+ * An interface for a JavaScript object that
+ * acts a dictionary. The keys are numbers.
+ */
+export interface INumberDictionary<V> {
+	[idx: number]: V;
+}
+
+const hasOwnProperty = Object.prototype.hasOwnProperty;
+
+/**
+ * Returns an array which contains all values that reside
+ * in the given set.
+ */
+export function values<T>(from: IStringDictionary<T> | INumberDictionary<T>): T[] {
+	const result: T[] = [];
+	for (var key in from) {
+		if (hasOwnProperty.call(from, key)) {
+			result.push(from[key]);
+		}
+	}
+	return result;
+}
+
+export function size<T>(from: IStringDictionary<T> | INumberDictionary<T>): number {
+	let count = 0;
+	for (var key in from) {
+		if (hasOwnProperty.call(from, key)) {
+			count += 1;
+		}
+	}
+	return count;
+}
+
+/**
+ * Iterates over each entry in the provided set. The iterator allows
+ * to remove elements and will stop when the callback returns {{false}}.
+ */
+export function forEach<T>(from: IStringDictionary<T> | INumberDictionary<T>, callback: (entry: { key: any; value: T; }, remove: Function) => any): void {
+	for (let key in from) {
+		if (hasOwnProperty.call(from, key)) {
+			const result = callback({ key: key, value: from[key] }, function () {
+				delete from[key];
+			});
+			if (result === false) {
+				return;
+			}
+		}
+	}
+}
+
+/**
+ * Removes an element from the dictionary. Returns {{false}} if the property
+ * does not exists.
+ */
+export function remove<T>(from: IStringDictionary<T> | INumberDictionary<T>, key: string): boolean {
+	if (!hasOwnProperty.call(from, key)) {
+		return false;
+	}
+	delete from[key];
+	return true;
+}
+
+/**
+ * Groups the collection into a dictionary based on the provided
+ * group function.
+ */
+export function groupBy<T>(data: T[], groupFn: (element: T) => string): IStringDictionary<T[]> {
+	const result: IStringDictionary<T[]> = Object.create(null);
+	for (const element of data) {
+		const key = groupFn(element);
+		let target = result[key];
+		if (!target) {
+			target = result[key] = [];
+		}
+		target.push(element);
+	}
+	return result;
+}