fix: flush events when exiting cli-repl

Author: gagik

packages/cli-repl/src/cli-repl.spec.ts

@@ -1695,6 +1695,7 @@ describe('CliRepl', function () {
             cliRepl,
             await testServer.connectionString()
           );
+
           expect(
             requests
               .flatMap((req) =>

packages/cli-repl/src/cli-repl.ts

@@ -1194,6 +1194,7 @@ export class CliRepl implements MongoshIOProvider {
         }
       );
       await this.logWriter?.flush();
+
       markTime(TimingCategories.Logging, 'flushed log writer');
       this.bus.emit('mongosh:closed');
     })());
@@ -1206,11 +1207,14 @@ export class CliRepl implements MongoshIOProvider {
    * @param code The user-provided exit code, if any.
    */
   async exit(code?: number): Promise<never> {
+    this.loggingAndTelemetry?.flush();
+
     await this.close();
     await this.onExit(code);
     // onExit never returns. If it does, that's a bug.
     const error = new MongoshInternalError('onExit() unexpectedly returned');
     this.bus.emit('mongosh:error', error, 'fatal');
+
     throw error;
   }
 

packages/logging/src/logging-and-telemetry.ts

@@ -113,6 +113,9 @@ export class LoggingAndTelemetry implements MongoshLoggingAndTelemetry {
   /** @internal */
   public setupTelemetryPromise: Promise<void> = Promise.resolve();
 
+  // eslint-disable-next-line @typescript-eslint/no-empty-function
+  private resolveDeviceId: (value: string) => void = () => {};
+
   constructor({
     bus,
     analytics,
@@ -132,7 +135,6 @@ export class LoggingAndTelemetry implements MongoshLoggingAndTelemetry {
     if (this.isSetup) {
       throw new Error('Setup can only be called once.');
     }
-    this.isBufferingBusEvents = true;
     this.isBufferingTelemetryEvents = true;
 
     this.setupTelemetryPromise = this.setupTelemetry();
@@ -141,10 +143,25 @@ export class LoggingAndTelemetry implements MongoshLoggingAndTelemetry {
     this.isSetup = true;
   }
 
+  public flush(): void {
+    // Run any telemetry events even if device ID hasn't been resolved yet
+    this.runAndClearPendingTelemetryEvents();
+
+    // Run any other pending events with the set or dummy log for telemetry purposes.
+    this.runAndClearPendingBusEvents();
+
+    this.resolveDeviceId('unknown');
+  }
+
   private async setupTelemetry(): Promise<void> {
     if (!this.deviceId) {
       try {
-        this.deviceId ??= await getDeviceId();
+        this.deviceId ??= await Promise.race([
+          getDeviceId(),
+          new Promise<string>((resolve) => {
+            this.resolveDeviceId = resolve;
+          }),
+        ]);
       } catch (error) {
         this.bus.emit('mongosh:error', error as Error, 'telemetry');
         this.deviceId = 'unknown';
@@ -173,8 +190,8 @@ export class LoggingAndTelemetry implements MongoshLoggingAndTelemetry {
 
   public detachLogger() {
     this.log = LoggingAndTelemetry.dummyLogger;
-    // Still run any remaining pending events with the dummy log for telemetry purposes.
-    this.runAndClearPendingBusEvents();
+
+    this.flush();
   }
 
   private runAndClearPendingBusEvents() {

packages/logging/src/types.ts

@@ -6,6 +6,8 @@ import type { MultiSet } from './helpers';
 export interface MongoshLoggingAndTelemetry {
   attachLogger(logger: MongoLogWriter): void;
   detachLogger(): void;
+  /** Flush any remaining log or telemetry events. */
+  flush(): void;
 }
 
 export type MongoshLoggingAndTelemetryArguments = {