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

src/sql/parts/profiler/service/profilerService.ts

@@ -0,0 +1,125 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the Source EULA. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import { IConnectionManagementService } from 'sql/parts/connection/common/connectionManagement';
+import {
+	ProfilerSessionID, IProfilerSession, IProfilerService, IProfilerSessionTemplate,
+	PROFILER_SETTINGS, IProfilerSettings
+} from './interfaces';
+import { ProfilerInput } from 'sql/parts/profiler/editor/profilerInput';
+import { ProfilerColumnEditorDialog } from 'sql/parts/profiler/dialog/profilerColumnEditorDialog';
+
+import * as data from 'data';
+
+import { TPromise } from 'vs/base/common/winjs.base';
+import { IConfigurationService } from 'vs/platform/configuration/common/configuration';
+import { IInstantiationService } from 'vs/platform/instantiation/common/instantiation';
+
+class TwoWayMap<T, K> {
+	private forwardMap: Map<T, K>;
+	private reverseMap: Map<K, T>;
+
+	constructor() {
+		this.forwardMap = new Map<T, K>();
+		this.reverseMap = new Map<K, T>();
+	}
+
+	get(input: T): K {
+		return this.forwardMap.get(input);
+	}
+
+	reverseGet(input: K): T {
+		return this.reverseMap.get(input);
+	}
+
+	set(input: T, input2: K): TwoWayMap<T, K> {
+		this.forwardMap.set(input, input2);
+		this.reverseMap.set(input2, input);
+		return this;
+	}
+}
+
+export class ProfilerService implements IProfilerService {
+	public _serviceBrand: any;
+	private _providers = new Map<string, data.IProfilerProvider>();
+	private _idMap = new TwoWayMap<ProfilerSessionID, string>();
+	private _sessionMap = new Map<ProfilerSessionID, IProfilerSession>();
+	private _dialog: ProfilerColumnEditorDialog;
+
+	constructor(
+		@IConnectionManagementService private _connectionService: IConnectionManagementService,
+		@IConfigurationService public _configurationService: IConfigurationService,
+		@IInstantiationService private _instantiationService: IInstantiationService
+	) { }
+
+	public registerProvider(providerId: string, provider: data.IProfilerProvider): void {
+		this._providers.set(providerId, provider);
+	}
+
+	public registerSession(uri: string, session: IProfilerSession): ProfilerSessionID {
+		this._sessionMap.set(uri, session);
+		this._idMap.set(uri, uri);
+		return uri;
+	}
+
+	public onMoreRows(params: data.IProfilerMoreRowsNotificationParams): void {
+		this._sessionMap.get(this._idMap.reverseGet(params.uri)).onMoreRows(params.rowCount, params.data);
+	}
+
+	public connectSession(id: ProfilerSessionID): Thenable<boolean> {
+		return this._runAction(id, provider => provider.connectSession(this._idMap.get(id)));
+	}
+
+	public disconnectSession(id: ProfilerSessionID): Thenable<boolean> {
+		return this._runAction(id, provider => provider.disconnectSession(this._idMap.get(id)));
+	}
+
+	public startSession(id: ProfilerSessionID): Thenable<boolean> {
+		return this._runAction(id, provider => provider.startSession(this._idMap.get(id)));
+	}
+
+	public pauseSession(id: ProfilerSessionID): Thenable<boolean> {
+		return this._runAction(id, provider => provider.pauseSession(this._idMap.get(id)));
+	}
+
+	public stopSession(id: ProfilerSessionID): Thenable<boolean> {
+		return this._runAction(id, provider => provider.stopSession(this._idMap.get(id)));
+	}
+
+	private _runAction<T>(id: ProfilerSessionID, action: (handler: data.IProfilerProvider) => Thenable<T>): Thenable<T> {
+		// let providerId = this._connectionService.getProviderIdFromUri(this._idMap.get(id));
+		let providerId = 'default';
+
+		if (!providerId) {
+			return TPromise.wrapError(new Error('Connection is required in order to interact with queries'));
+		}
+		let handler = this._providers.get(providerId);
+		if (handler) {
+			return action(handler);
+		} else {
+			return TPromise.wrapError(new Error('No Handler Registered'));
+		}
+	}
+
+	public getSessionTemplates(provider?: string): Array<IProfilerSessionTemplate> {
+		let config = <IProfilerSettings>this._configurationService.getConfiguration(PROFILER_SETTINGS);
+
+		if (provider) {
+			return config.sessionTemplates;
+		} else {
+			return config.sessionTemplates;
+		}
+	}
+
+	public launchColumnEditor(input?: ProfilerInput): Thenable<void> {
+		if (!this._dialog) {
+			this._dialog = this._instantiationService.createInstance(ProfilerColumnEditorDialog);
+			this._dialog.render();
+		}
+
+		this._dialog.open(input);
+		return TPromise.as(null);
+	}
+}