ckaczor/azuredatastudio
1
0
Fork 0
mirror of https://github.com/ckaczor/azuredatastudio.git synced 2026-02-02 17:23:40 -05:00
Code Issues Packages Projects Releases Wiki Activity

Add datavirtualization extension (#21594)

Browse Source 
* initial

* cleanup

* Add typings ref

* fix compile

* remove unused

* add missing

* another unused

* Use newer vscodetestcover

* newer dataprotocol

* format

* cleanup ignores

* fix out path

* fix entry point

* more cleanup

* Move into src folder

* Handle service client log messages

* remove unused
This commit is contained in:
Charles Gagnon
2023-01-17 09:57:21 -08:00
committed by GitHub
parent 9184c414de
commit ec838947b0
103 changed files with 12432 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

189
extensions/datavirtualization/src/services/features.ts Normal file
View File

@@ -0,0 +1,189 @@
/*---------------------------------------------------------------------------------------------
* Copyright (c) Microsoft Corporation. All rights reserved.
* Licensed under the Source EULA. See License.txt in the project root for license information.
*--------------------------------------------------------------------------------------------*/
import { SqlOpsDataClient, SqlOpsFeature } from 'dataprotocol-client';
import { ClientCapabilities, StaticFeature, RPCMessageType, ServerCapabilities } from 'vscode-languageclient';
import * as UUID from 'vscode-languageclient/lib/utils/uuid';
import { Disposable } from 'vscode';
import * as azdata from 'azdata';
import { Telemetry } from './telemetry';
import * as serviceUtils from './serviceUtils';
import {
TelemetryNotification,
DataSourceWizardCreateSessionRequest, DataSourceWizardConfigInfoResponse,
DataSourceWizardService, DisposeWizardSessionRequest, VirtualizeDataInput,
ValidateVirtualizeDataInputResponse, GetDatabaseInfoRequestParams, GetDatabaseInfoResponse,
ProcessVirtualizeDataInputResponse, GenerateScriptResponse, ValidateVirtualizeDataInputRequest,
GetDatabaseInfoRequest, ProcessVirtualizeDataInputRequest, GenerateScriptRequest, GetSourceDatabasesResponse,
GetSourceDatabasesRequest, GetSourceTablesResponse, GetSourceTablesRequestParams, GetSourceTablesRequest,
GetSourceColumnDefinitionsRequestParams, GetSourceColumnDefinitionsResponse, GetSourceColumnDefinitionsRequest,
ProseDiscoveryParams, ProseDiscoveryResponse, ProseDiscoveryRequest, DataSourceBrowsingParams, ExecutionResult, GetSourceViewListRequest, SchemaViews
} from './contracts';
import { managerInstance, ApiType } from './serviceApiManager';
export class TelemetryFeature implements StaticFeature {
constructor(private _client: SqlOpsDataClient) { }
fillClientCapabilities(capabilities: ClientCapabilities): void {
serviceUtils.ensure(capabilities, 'telemetry')!.telemetry = true;
}
initialize(): void {
this._client.onNotification(TelemetryNotification.type, e => {
Telemetry.sendTelemetryEvent(e.params.eventName, e.params.properties, e.params.measures);
});
}
}
export class DataSourceWizardFeature extends SqlOpsFeature<undefined> {
private static readonly messagesTypes: RPCMessageType[] = [
DataSourceWizardCreateSessionRequest.type
];
constructor(client: SqlOpsDataClient) {
super(client, DataSourceWizardFeature.messagesTypes);
}
public fillClientCapabilities(capabilities: ClientCapabilities): void {
// ensure(ensure(capabilities, 'connection')!, 'objectExplorer')!.dynamicRegistration = true;
}
public initialize(capabilities: ServerCapabilities): void {
this.register(this.messages, {
id: UUID.generateUuid(),
registerOptions: undefined
});
}
protected registerProvider(options: undefined): Disposable {
const client = this._client;
let createDataSourceWizardSession = (requestParams: azdata.connection.ConnectionProfile): Thenable<DataSourceWizardConfigInfoResponse> => {
return client.sendRequest(DataSourceWizardCreateSessionRequest.type, requestParams).then(
r => r,
e => {
client.logFailedRequest(DataSourceWizardCreateSessionRequest.type, e);
return Promise.reject(e);
}
);
};
let disposeWizardSession = (sessionId: string): Thenable<boolean> => {
return client.sendRequest(DisposeWizardSessionRequest.type, sessionId).then(
r => r,
e => {
client.logFailedRequest(DisposeWizardSessionRequest.type, e);
return Promise.reject(e);
}
);
};
let validateVirtualizeDataInput = (requestParams: VirtualizeDataInput): Thenable<ValidateVirtualizeDataInputResponse> => {
return client.sendRequest(ValidateVirtualizeDataInputRequest.type, requestParams).then(
r => r,
e => {
client.logFailedRequest(ValidateVirtualizeDataInputRequest.type, e);
return Promise.reject(e);
}
);
};
let getDatabaseInfo = (requestParams: GetDatabaseInfoRequestParams): Thenable<GetDatabaseInfoResponse> => {
return client.sendRequest(GetDatabaseInfoRequest.type, requestParams).then(
r => r,
e => {
client.logFailedRequest(GetDatabaseInfoRequest.type, e);
return Promise.reject(e);
}
);
};
let processVirtualizeDataInput = (requestParams: VirtualizeDataInput): Thenable<ProcessVirtualizeDataInputResponse> => {
return client.sendRequest(ProcessVirtualizeDataInputRequest.type, requestParams).then(
r => r,
e => {
client.logFailedRequest(ProcessVirtualizeDataInputRequest.type, e);
return Promise.reject(e);
}
);
};
let generateScript = (requestParams: VirtualizeDataInput): Thenable<GenerateScriptResponse> => {
return client.sendRequest(GenerateScriptRequest.type, requestParams).then(
r => r,
e => {
client.logFailedRequest(GenerateScriptRequest.type, e);
return Promise.reject(e);
}
);
};
let getSourceDatabases = (requestParams: VirtualizeDataInput): Thenable<GetSourceDatabasesResponse> => {
return client.sendRequest(GetSourceDatabasesRequest.type, requestParams).then(
r => r,
e => {
client.logFailedRequest(GetSourceDatabasesRequest.type, e);
return Promise.reject(e);
}
);
};
let getSourceTables = (requestParams: GetSourceTablesRequestParams): Thenable<GetSourceTablesResponse> => {
return client.sendRequest(GetSourceTablesRequest.type, requestParams).then(
r => r,
e => {
client.logFailedRequest(GetSourceTablesRequest.type, e);
return Promise.reject(e);
}
);
};
let getSourceViewList = (requestParams: DataSourceBrowsingParams<string>): Thenable<ExecutionResult<SchemaViews[]>> => {
return client.sendRequest(GetSourceViewListRequest.type, requestParams).then(
r => r,
e => {
client.logFailedRequest(GetSourceViewListRequest.type, e);
return Promise.reject(e);
}
);
};
let getSourceColumnDefinitions = (requestParams: GetSourceColumnDefinitionsRequestParams): Thenable<GetSourceColumnDefinitionsResponse> => {
return client.sendRequest(GetSourceColumnDefinitionsRequest.type, requestParams).then(
r => r,
e => {
client.logFailedRequest(GetSourceColumnDefinitionsRequest.type, e);
return Promise.reject(e);
}
);
};
let sendProseDiscoveryRequest = (requestParams: ProseDiscoveryParams): Thenable<ProseDiscoveryResponse> => {
return client.sendRequest(ProseDiscoveryRequest.type, requestParams).then(
r => r,
e => {
client.logFailedRequest(ProseDiscoveryRequest.type, e);
return Promise.reject(e);
}
);
};
return managerInstance.registerApi<DataSourceWizardService>(ApiType.DataSourceWizard, {
providerId: client.providerId,
createDataSourceWizardSession,
disposeWizardSession,
validateVirtualizeDataInput,
getDatabaseInfo,
processVirtualizeDataInput,
generateScript,
getSourceDatabases,
getSourceTables,
getSourceViewList,
getSourceColumnDefinitions,
sendProseDiscoveryRequest
});
}
}