From 8114498fb5291b4640b68de3f4e100a4010d4aeb Mon Sep 17 00:00:00 2001
From: Leila Lali <llali@microsoft.com>
Date: Thu, 1 Mar 2018 08:56:00 -0800
Subject: [PATCH] writing telemetry into a file when running the app for perf
 tests (#802)

---
 .../telemetry/fileTelemetryService.ts         | 43 +++++++++++++++++++
 src/vs/workbench/electron-browser/shell.ts    | 10 ++++-
 2 files changed, 51 insertions(+), 2 deletions(-)
 create mode 100644 src/sql/platform/telemetry/fileTelemetryService.ts

diff --git a/src/sql/platform/telemetry/fileTelemetryService.ts b/src/sql/platform/telemetry/fileTelemetryService.ts
new file mode 100644
index 0000000000..1957d268b7
--- /dev/null
+++ b/src/sql/platform/telemetry/fileTelemetryService.ts
@@ -0,0 +1,43 @@
+/*---------------------------------------------------------------------------------------------
+ *  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 { TPromise } from 'vs/base/common/winjs.base';
+import { ITelemetryService, ITelemetryInfo, ITelemetryData } from 'vs/platform/telemetry/common/telemetry';
+const fs = require('fs');
+
+/**
+ * Write telemetry into a file for test purposes
+ */
+export class FileTelemetryService implements ITelemetryService {
+	_serviceBrand: undefined;
+	private _isFirst = true;
+
+	constructor(private _outputFile: string) {
+	}
+
+	publicLog(eventName: string, data?: ITelemetryData) {
+		let telemetryData = JSON.stringify(Object.assign({eventName: eventName, data: data}));
+		if (this._outputFile) {
+			if (this._isFirst) {
+				fs.open(this._outputFile,  fs.O_WRONLY | fs.O_CREAT, (err, fr) => {
+					fs.writeFileSync(this._outputFile, telemetryData + '\n');
+					this._isFirst = false;
+				});
+			} else {
+				fs.appendFileSync(this._outputFile, telemetryData + '\n');
+			}
+		}
+		return TPromise.wrap<void>(null);
+	}
+	isOptedIn: true;
+	getTelemetryInfo(): TPromise<ITelemetryInfo> {
+		return TPromise.wrap({
+			instanceId: 'someValue.instanceId',
+			sessionId: 'someValue.sessionId',
+			machineId: 'someValue.machineId'
+		});
+	}
+};
\ No newline at end of file
diff --git a/src/vs/workbench/electron-browser/shell.ts b/src/vs/workbench/electron-browser/shell.ts
index 72f6f3f3fa..dd1cbb066d 100644
--- a/src/vs/workbench/electron-browser/shell.ts
+++ b/src/vs/workbench/electron-browser/shell.ts
@@ -89,6 +89,9 @@ import { HashService } from 'vs/workbench/services/hash/node/hashService';
 import { IHashService } from 'vs/workbench/services/hash/common/hashService';
 import { ILogService } from 'vs/platform/log/common/log';
 
+// {{SQL CARBON EDIT}}
+import { FileTelemetryService } from 'sql/platform/telemetry/fileTelemetryService';
+
 /**
  * Services that we require for the Shell
  */
@@ -315,9 +318,12 @@ export class WorkbenchShell {
 		// Experiments
 		this.experimentService = instantiationService.createInstance(ExperimentService);
 		serviceCollection.set(IExperimentService, this.experimentService);
-
+		// {{SQL CARBON EDIT}}
+		if (this.environmentService.args['perf-test']) {
+			let telemetryOutput = this.environmentService.args['telemetry-output'];
+			this.telemetryService = new FileTelemetryService(telemetryOutput);
 		// Telemetry
-		if (this.environmentService.isBuilt && !this.environmentService.isExtensionDevelopment && !this.environmentService.args['disable-telemetry'] && !!product.enableTelemetry) {
+		} else if (this.environmentService.isBuilt && !this.environmentService.isExtensionDevelopment && !this.environmentService.args['disable-telemetry'] && !!product.enableTelemetry) {
 			const channel = getDelayedChannel<ITelemetryAppenderChannel>(sharedProcess.then(c => c.getChannel('telemetryAppender')));
 			const commit = product.commit;
 			const version = pkg.version;