ckaczor/azuredatastudio
1
0
Fork 0
mirror of https://github.com/ckaczor/azuredatastudio.git synced 2026-01-17 09:35:37 -05:00
Code Issues Packages Projects Releases Wiki Activity

Add connection event listeners (#6540)

Browse Source
This commit is contained in:
Karl Burtram
2019-07-31 11:52:58 -07:00
committed by GitHub
parent 2c79d49487
commit bc4b527de0
5 changed files with 78 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
src/sql/workbench/api/browser/mainThreadConnectionManagement.ts
View File

@@ -7,7 +7,7 @@ import { SqlExtHostContext, SqlMainContext, ExtHostConnectionManagementShape, Ma
import * as azdata from 'azdata';
import { IExtHostContext } from 'vs/workbench/api/common/extHost.protocol';
import { extHostNamedCustomer } from 'vs/workbench/api/common/extHostCustomers';
import { IConnectionManagementService, ConnectionType } from 'sql/platform/connection/common/connectionManagement';
import { IConnectionManagementService, ConnectionType, IConnectionParams } from 'sql/platform/connection/common/connectionManagement';
import { IObjectExplorerService } from 'sql/workbench/services/objectExplorer/common/objectExplorerService';
import { IEditorService } from 'vs/workbench/services/editor/common/editorService';
import * as TaskUtilities from 'sql/workbench/browser/taskUtilities';
@@ -44,6 +44,45 @@ export class MainThreadConnectionManagement implements MainThreadConnectionManag
this._toDispose = dispose(this._toDispose);
}
public $registerConnectionEventListener(handle: number, providerId: string): void {
let stripProfile = (inputProfile: azdata.IConnectionProfile) => {
if (!inputProfile) {
return inputProfile;
}
let outputProfile: azdata.IConnectionProfile = {
connectionName: inputProfile.connectionName,
serverName: inputProfile.serverName,
databaseName: inputProfile.databaseName,
userName: inputProfile.userName,
password: inputProfile.password,
authenticationType: inputProfile.authenticationType,
savePassword: inputProfile.savePassword,
groupFullName: inputProfile.groupFullName,
groupId: inputProfile.groupId,
providerName: inputProfile.providerName,
saveProfile: inputProfile.saveProfile,
id: inputProfile.id,
azureTenantId: inputProfile.azureTenantId,
options: inputProfile.options
};
return outputProfile;
};
this._connectionManagementService.onConnect((params: IConnectionParams) => {
this._proxy.$onConnectionEvent(handle, 'onConnect', params.connectionUri, stripProfile(params.connectionProfile));
});
this._connectionManagementService.onConnectionChanged((params: IConnectionParams) => {
this._proxy.$onConnectionEvent(handle, 'onConnectionChanged', params.connectionUri, stripProfile(params.connectionProfile));
});
this._connectionManagementService.onDisconnect((params: IConnectionParams) => {
this._proxy.$onConnectionEvent(handle, 'onDisconnect', params.connectionUri, stripProfile(params.connectionProfile));
});
}
public $getConnections(activeConnectionsOnly?: boolean): Thenable<azdata.connection.ConnectionProfile[]> {
return Promise.resolve(this._connectionManagementService.getConnections(activeConnectionsOnly).map(profile => this.convertToConnectionProfile(profile)));
}

15
src/sql/workbench/api/common/extHostConnectionManagement.ts
View File

@@ -10,6 +10,8 @@ import * as azdata from 'azdata';
export class ExtHostConnectionManagement extends ExtHostConnectionManagementShape {
private _proxy: MainThreadConnectionManagementShape;
private _nextListenerHandle: number = 0;
private _connectionListeners = new Map<number, azdata.connection.ConnectionEventListener>();
constructor(
mainContext: IMainContext
@@ -22,6 +24,19 @@ export class ExtHostConnectionManagement extends ExtHostConnectionManagementShap
return this._proxy.$getCurrentConnectionProfile();
}
public $onConnectionEvent(handle: number, type: azdata.connection.ConnectionEventType, ownerUri: string, profile: azdata.IConnectionProfile): void {
let listener = this._connectionListeners[handle];
if (listener) {
listener.onConnectionEvent(type, ownerUri, profile);
}
}
public $registerConnectionEventListener(providerId: string, listener: azdata.connection.ConnectionEventListener): void {
this._connectionListeners[this._nextListenerHandle] = listener;
this._proxy.$registerConnectionEventListener(this._nextListenerHandle, providerId);
this._nextListenerHandle++;
}
public $getConnections(activeConnectionsOnly?: boolean): Thenable<azdata.connection.ConnectionProfile[]> {
return this._proxy.$getConnections(activeConnectionsOnly);
}

3
src/sql/workbench/api/common/sqlExtHost.protocol.ts
View File

@@ -38,7 +38,7 @@ export abstract class ExtHostAccountManagementShape {
}
export abstract class ExtHostConnectionManagementShape {
$onConnectionOpened(handleId: string, connection: azdata.connection.Connection): void { throw ni; }
$onConnectionEvent(handle: number, type: azdata.connection.ConnectionEventType, ownerUri: string, profile: azdata.IConnectionProfile): void { throw ni(); }
}
export abstract class ExtHostDataProtocolShape {
@@ -606,6 +606,7 @@ export interface MainThreadDataProtocolShape extends IDisposable {
}
export interface MainThreadConnectionManagementShape extends IDisposable {
$registerConnectionEventListener(handle: number, providerId: string): void;
$getConnections(activeConnectionsOnly?: boolean): Thenable<azdata.connection.ConnectionProfile[]>;
$getActiveConnections(): Thenable<azdata.connection.Connection[]>;
$getCurrentConnection(): Thenable<azdata.connection.Connection>;

3
src/sql/workbench/api/node/sqlExtHost.api.impl.ts
View File

@@ -101,6 +101,9 @@ export function createApiFactory(
getConnections(activeConnectionsOnly?: boolean): Thenable<azdata.connection.ConnectionProfile[]> {
return extHostConnectionManagement.$getConnections(activeConnectionsOnly);
},
registerConnectionEventListener(listener: azdata.connection.ConnectionEventListener): void {
return extHostConnectionManagement.$registerConnectionEventListener(mssqlProviderName, listener);
},
// "sqlops" back-compat APIs
getActiveConnections(): Thenable<azdata.connection.Connection[]> {