ckaczor/azuredatastudio
1
0
Fork 0
mirror of https://github.com/ckaczor/azuredatastudio.git synced 2026-02-13 11:38:36 -05:00
Code Issues Packages Projects Releases Wiki Activity

move server report ext to extensions (#14087)

Browse Source 
* move server report ext to extensions

* limit the size of data

* comments
This commit is contained in:
Alan Ren
2021-01-28 13:23:30 -08:00
committed by GitHub
parent f9ea6430ee
commit c16aee760f
59 changed files with 535 additions and 4885 deletions
Download Patch File Download Diff File Expand all files Collapse all files

70
extensions/server-report/src/controllers/mainController.ts Normal file
View File

@@ -0,0 +1,70 @@
/*---------------------------------------------------------------------------------------------
* Copyright (c) Microsoft Corporation. All rights reserved.
* Licensed under the Source EULA. See License.txt in the project root for license information.
*--------------------------------------------------------------------------------------------*/
'use strict';
import * as azdata from 'azdata';
import * as Utils from '../utils';
import ControllerBase from './controllerBase';
import { promises } from 'fs';
import * as path from 'path';
import * as vscode from 'vscode';
import * as constants from '../constants';
/**
* The main controller class that initializes the extension
*/
export default class MainController extends ControllerBase {
private autoRefreshState: boolean = false;
// PUBLIC METHODS //////////////////////////////////////////////////////
/**
* Deactivates the extension
*/
public deactivate(): void {
Utils.logDebug('Main controller deactivated');
}
public activate(): Promise<boolean> {
azdata.tasks.registerTask('tempdb.startEvent', e => this.onExecute(e, 'startEvent.sql'));
azdata.tasks.registerTask('tempdb.stopEvent', e => this.onExecute(e, 'stopEvent.sql'));
azdata.tasks.registerTask('tempdb.contention', () => this.openurl('https://aka.ms/tempdbblog'));
azdata.tasks.registerTask('tempdb.pauseEvent', e => this.stopAutoRefresh(e));
return Promise.resolve(true);
}
private openurl(link: string): void {
vscode.commands.executeCommand('vscode.open', vscode.Uri.parse(link));
}
private async onExecute(connection: azdata.IConnectionProfile, fileName: string): Promise<void> {
// Command to start/stop autorefresh and run the query
let connectionUri = await azdata.connection.getUriForConnection(connection.id);
let connectionProvider = azdata.dataprotocol.getProvider<azdata.ConnectionProvider>(connection.providerName, azdata.DataProviderType.ConnectionProvider);
connectionProvider.changeDatabase(connectionUri, 'tempdb');
let queryProvider = azdata.dataprotocol.getProvider<azdata.QueryProvider>(connection.providerName, azdata.DataProviderType.QueryProvider);
let sqlContent: string = await promises.readFile(path.join(__dirname, '..', 'sql', fileName), { encoding: 'utf8' });
let seResult = await queryProvider.runQueryAndReturn(connectionUri, sqlContent);
if (seResult.rowCount > 0 && seResult.rows[0][0].displayValue === '0') {
vscode.window.showInformationMessage((fileName === 'startEvent.sql') ? constants.XEventsStarted : constants.XEventsStopped);
this.autoRefreshState = (fileName === 'startEvent.sql') ? true : false;
vscode.commands.executeCommand('azdata.widget.setAutoRefreshState', 'type-of-contention', connection.id, this.autoRefreshState);
vscode.commands.executeCommand('azdata.widget.setAutoRefreshState', 'metadata-contention', connection.id, this.autoRefreshState);
vscode.commands.executeCommand('azdata.widget.setAutoRefreshState', 'allocation-contention', connection.id, this.autoRefreshState);
} else if (seResult.rowCount > 0 && seResult.rows[0][0].displayValue === '1') {
vscode.window.showErrorMessage(constants.XEventsNotSupported);
} else {
vscode.window.showErrorMessage(constants.XEventsFailed);
}
}
private stopAutoRefresh(connection: azdata.IConnectionProfile): void {
this.autoRefreshState = !this.autoRefreshState === true;
vscode.commands.executeCommand('azdata.widget.setAutoRefreshState', 'type-of-contention', connection.id, this.autoRefreshState);
vscode.commands.executeCommand('azdata.widget.setAutoRefreshState', 'metadata-contention', connection.id, this.autoRefreshState);
vscode.commands.executeCommand('azdata.widget.setAutoRefreshState', 'allocation-contention', connection.id, this.autoRefreshState);
}
}