Dispose the reporter (#335)

microsoft · Oct 9, 2018 · b1be4c4 · b1be4c4
1 parent ff733a6
commit b1be4c4
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/extension/src/extension.ts b/extension/src/extension.ts
@@ -41,6 +41,7 @@ const testConfigManager: TestConfigManager = new TestConfigManager(projectManage
 // your extension is activated the very first time the command is executed
 export async function activate(context: ExtensionContext) {
     TelemetryWrapper.initilizeFromJsonFile(context.asAbsolutePath('./package.json'));
+    context.subscriptions.push(TelemetryWrapper.getReporter());
     await initializeFromJsonFile(context.asAbsolutePath('./package.json'));
     await instrumentOperation('activation', doActivate)(context);
 }
@@ -138,13 +139,12 @@ async function doActivate(_operationId: string, context: ExtensionContext): Prom
     await commands.executeCommand('setContext', 'java.test.activated', true);
 }
 
-// this method is called when your extension is deactivated
-export function deactivate() {
+export async function deactivate() {
     testResourceManager.dispose();
     classPathManager.dispose();
     testStatusBarItem.dispose();
     CommandUtility.clearCommandsCache();
-    disposeTelemetryWrapper();
+    await disposeTelemetryWrapper();
 }
 
 function instrumentAndRegisterCommand(name: string, cb: (...args: any[]) => any): Disposable {