Add query execution time to status bar (#17942)

Author: cssuh

src/controllers/queryRunner.ts

@@ -247,6 +247,7 @@ export default class QueryRunner {
 			this._uriToQueryPromiseMap.delete(result.ownerUri);
 		}
 		this._statusView.executedQuery(result.ownerUri);
+		this._statusView.setExecutionTime(result.ownerUri, Utils.parseNumAsTimeString(this._totalElapsedMilliseconds));
 		let hasError = this._batchSets.some(batch => batch.hasError === true);
 		this.removeRunningQuery();
 		this.eventEmitter.emit('complete', Utils.parseNumAsTimeString(this._totalElapsedMilliseconds), hasError);

src/views/statusView.ts

@@ -35,6 +35,9 @@ class FileStatusBar {
 	// Item for Row Count
 	public rowCount: vscode.StatusBarItem;
 
+	// Item for execution time
+	public executionTime: vscode.StatusBarItem;
+
 	public currentLanguageServiceStatus: string;
 }
 
@@ -62,6 +65,7 @@ export default class StatusView implements vscode.Disposable {
 				this._statusBars[bar].statusLanguageService.dispose();
 				this._statusBars[bar].sqlCmdMode.dispose();
 				this._statusBars[bar].rowCount.dispose();
+				this._statusBars[bar].executionTime.dispose();
 				clearInterval(this._statusBars[bar].progressTimerId);
 				delete this._statusBars[bar];
 			}
@@ -81,6 +85,7 @@ export default class StatusView implements vscode.Disposable {
 		bar.statusLanguageService = vscode.window.createStatusBarItem(vscode.StatusBarAlignment.Right);
 		bar.sqlCmdMode = vscode.window.createStatusBarItem(vscode.StatusBarAlignment.Right, 90);
 		bar.rowCount = vscode.window.createStatusBarItem(vscode.StatusBarAlignment.Right, 80);
+		bar.executionTime = vscode.window.createStatusBarItem(vscode.StatusBarAlignment.Right, 70);
 		this._statusBars[fileUri] = bar;
 	}
 
@@ -108,6 +113,9 @@ export default class StatusView implements vscode.Disposable {
 			if (bar.rowCount) {
 				bar.rowCount.dispose();
 			}
+			if (bar.executionTime) {
+				bar.executionTime.dispose();
+			}
 
 			delete this._statusBars[fileUri];
 		}
@@ -197,6 +205,12 @@ export default class StatusView implements vscode.Disposable {
 		}, 200);
 	}
 
+	public setExecutionTime(fileUri: string, time: string): void {
+		let bar = this.getStatusBar(fileUri);
+		bar.executionTime.text = time;
+		this.showStatusBarItem(fileUri, bar.executionTime);
+	}
+
 	public cancelingQuery(fileUri: string): void {
 		let bar = this.getStatusBar(fileUri);
 		bar.statusQuery.hide();
@@ -302,6 +316,7 @@ export default class StatusView implements vscode.Disposable {
 			this._lastShownStatusBar.statusLanguageService.hide();
 			this._lastShownStatusBar.sqlCmdMode.hide();
 			this._lastShownStatusBar.rowCount.hide();
+			this._lastShownStatusBar.executionTime.hide();
 		}
 	}
 
@@ -318,6 +333,7 @@ export default class StatusView implements vscode.Disposable {
 				this.showStatusBarItem(fileUri, bar.statusLanguageService);
 				this.showStatusBarItem(fileUri, bar.sqlCmdMode);
 				this.showStatusBarItem(fileUri, bar.rowCount);
+				this.showStatusBarItem(fileUri, bar.executionTime);
 			}
 		}
 	}