chore: adjust timeouts and correct promise chain

Author: nbbeeken

src/cmap/wire_protocol/on_data.ts

@@ -105,7 +105,7 @@ export function onData(
   function errorHandler(err: Error) {
     const promise = unconsumedPromises.shift();
     const timeoutError = TimeoutError.is(err)
-      ? new MongoOperationTimeoutError('Timed out during socket read')
+      ? new MongoOperationTimeoutError(`Timed out during socket read (${err.duration}ms)`)
       : undefined;
 
     if (promise != null) promise.reject(timeoutError ?? err);

src/sessions.ts

@@ -281,9 +281,9 @@ export class ClientSession
     try {
       if (this.inTransaction()) {
         if (typeof options?.timeoutMS === 'number') {
-          await this.abortTransaction({ timeoutMS: options.timeoutMS });
+          await endTransaction(this, 'abortTransaction', { timeoutMS: options.timeoutMS });
         } else {
-          await this.abortTransaction();
+          await endTransaction(this, 'abortTransaction');
         }
       }
       if (!this.hasEnded) {

src/timeout.ts

@@ -5,12 +5,14 @@ import { csotMin, noop } from './utils';
 
 /** @internal */
 export class TimeoutError extends Error {
+  duration: number;
   override get name(): 'TimeoutError' {
     return 'TimeoutError';
   }
 
-  constructor(message: string, options?: { cause?: Error }) {
+  constructor(message: string, options: { cause?: Error; duration: number }) {
     super(message, options);
+    this.duration = options.duration;
   }
 
   static is(error: unknown): error is TimeoutError {
@@ -52,7 +54,12 @@ export class Timeout extends Promise<never> {
   }
 
   /** Create a new timeout that expires in `duration` ms */
-  private constructor(executor: Executor = () => null, duration: number, unref = true) {
+  private constructor(
+    executor: Executor = () => null,
+    duration: number,
+    unref = true,
+    rejection: Error | null = null
+  ) {
     let reject!: Reject;
     if (duration < 0) {
       throw new MongoInvalidArgumentError('Cannot create a Timeout with a negative duration');
@@ -71,13 +78,15 @@ export class Timeout extends Promise<never> {
       this.id = setTimeout(() => {
         this.ended = Math.trunc(performance.now());
         this.timedOut = true;
-        reject(new TimeoutError(`Expired after ${duration}ms`));
+        reject(new TimeoutError(`Expired after ${duration}ms`, { duration }));
       }, this.duration);
       if (typeof this.id.unref === 'function' && unref) {
         // Ensure we do not keep the Node.js event loop running
         this.id.unref();
       }
     }
+
+    if (rejection != null) reject(rejection);
   }
 
   /**
@@ -90,7 +99,7 @@ export class Timeout extends Promise<never> {
   }
 
   throwIfExpired(): void {
-    if (this.timedOut) throw new TimeoutError('Timed out');
+    if (this.timedOut) throw new TimeoutError('Timed out', { duration: this.duration });
   }
 
   public static expires(durationMS: number, unref?: boolean): Timeout {
@@ -108,6 +117,10 @@ export class Timeout extends Promise<never> {
       typeof timeout.then === 'function'
     );
   }
+
+  static override reject(rejection?: Error): Timeout {
+    return new Timeout(undefined, 0, true, rejection);
+  }
 }
 
 /** @internal */
@@ -218,8 +231,8 @@ export class CSOTTimeoutContext extends TimeoutContext {
     if (typeof this._serverSelectionTimeout !== 'object' || this._serverSelectionTimeout?.cleared) {
       const { remainingTimeMS, serverSelectionTimeoutMS } = this;
       if (remainingTimeMS <= 0)
-        throw new MongoOperationTimeoutError(
-          `Timed out in server selection after ${this.timeoutMS}ms`
+        return Timeout.reject(
+          new MongoOperationTimeoutError(`Timed out in server selection after ${this.timeoutMS}ms`)
         );
       const usingServerSelectionTimeoutMS =
         serverSelectionTimeoutMS !== 0 &&
@@ -247,8 +260,10 @@ export class CSOTTimeoutContext extends TimeoutContext {
         // null or Timeout
         this._connectionCheckoutTimeout = this._serverSelectionTimeout;
       } else {
-        throw new MongoRuntimeError(
-          'Unreachable. If you are seeing this error, please file a ticket on the NODE driver project on Jira'
+        return Timeout.reject(
+          new MongoRuntimeError(
+            'Unreachable. If you are seeing this error, please file a ticket on the NODE driver project on Jira'
+          )
         );
       }
     }
@@ -259,14 +274,14 @@ export class CSOTTimeoutContext extends TimeoutContext {
     const { remainingTimeMS } = this;
     if (!Number.isFinite(remainingTimeMS)) return null;
     if (remainingTimeMS > 0) return Timeout.expires(remainingTimeMS);
-    throw new MongoOperationTimeoutError('Timed out before socket write');
+    return Timeout.reject(new MongoOperationTimeoutError('Timed out before socket write'));
   }
 
   get timeoutForSocketRead(): Timeout | null {
     const { remainingTimeMS } = this;
     if (!Number.isFinite(remainingTimeMS)) return null;
     if (remainingTimeMS > 0) return Timeout.expires(remainingTimeMS);
-    throw new MongoOperationTimeoutError('Timed out before socket read');
+    return Timeout.reject(new MongoOperationTimeoutError('Timed out before socket read'));
   }
 }
 

src/utils.ts

@@ -528,12 +528,6 @@ export function resolveOptions<T extends CommandOperationOptions>(
     result.readPreference = readPreference;
   }
 
-  if (session?.explicit && session.timeoutMS != null && options?.timeoutMS != null) {
-    throw new MongoInvalidArgumentError(
-      'Do not specify timeoutMS on operation if already specified on an explicit session'
-    );
-  }
-
   const timeoutMS = options?.timeoutMS;
 
   result.timeoutMS = timeoutMS ?? parent?.timeoutMS;

test/integration/client-side-operations-timeout/client_side_operations_timeout.prose.test.ts

@@ -633,7 +633,11 @@ describe('CSOT spec prose tests', function () {
       const failpoint: FailPoint = {
         configureFailPoint: 'failCommand',
         mode: { times: 1 },
-        data: { failCommands: ['abortTransaction'], blockConnection: true, blockTimeMS: 15 }
+        data: {
+          failCommands: ['abortTransaction'],
+          blockConnection: true,
+          blockTimeMS: 60
+        }
       };
 
       beforeEach(async function () {
@@ -653,13 +657,18 @@ describe('CSOT spec prose tests', function () {
 
       let client: MongoClient;
 
-      afterEach(async () => {
+      afterEach(async function () {
+        if (semver.satisfies(this.configuration.version, '>=4.4')) {
+          const internalClient = this.configuration.newClient();
+          await internalClient.db('admin').command({ ...failpoint, mode: 'off' });
+          await internalClient.close();
+        }
         await client?.close();
       });
 
       describe('when timeoutMS is provided to the client', () => {
         it('throws a timeout error from endSession', async function () {
-          client = this.configuration.newClient({ timeoutMS: 10 });
+          client = this.configuration.newClient({ timeoutMS: 50, monitorCommands: true });
           const coll = client.db('db').collection('coll');
           const session = client.startSession();
           session.startTransaction();
@@ -673,7 +682,7 @@ describe('CSOT spec prose tests', function () {
         it('throws a timeout error from endSession', async function () {
           client = this.configuration.newClient();
           const coll = client.db('db').collection('coll');
-          const session = client.startSession({ defaultTimeoutMS: 10 });
+          const session = client.startSession({ defaultTimeoutMS: 50 });
           session.startTransaction();
           await coll.insertOne({ x: 1 }, { session });
           const error = await session.endSession().catch(error => error);
@@ -688,7 +697,7 @@ describe('CSOT spec prose tests', function () {
           const session = client.startSession();
           session.startTransaction();
           await coll.insertOne({ x: 1 }, { session });
-          const error = await session.endSession({ timeoutMS: 10 }).catch(error => error);
+          const error = await session.endSession({ timeoutMS: 50 }).catch(error => error);
           expect(error).to.be.instanceOf(MongoOperationTimeoutError);
         });
       });
@@ -736,7 +745,7 @@ describe('CSOT spec prose tests', function () {
           data: {
             failCommands: ['insert', 'abortTransaction'],
             blockConnection: true,
-            blockTimeMS: 15
+            blockTimeMS: 60
           }
         };
 
@@ -757,15 +766,20 @@ describe('CSOT spec prose tests', function () {
 
         let client: MongoClient;
 
-        afterEach(async () => {
+        afterEach(async function () {
+          if (semver.satisfies(this.configuration.version, '>=4.4')) {
+            const internalClient = this.configuration.newClient();
+            await internalClient.db('admin').command({ ...failpoint, mode: 'off' });
+            await internalClient.close();
+          }
           await client?.close();
         });
 
         it('timeoutMS is refreshed for abortTransaction', async function () {
           const commandsFailed = [];
 
           client = this.configuration
-            .newClient({ timeoutMS: 10, monitorCommands: true })
+            .newClient({ timeoutMS: 50, monitorCommands: true })
             .on('commandFailed', e => commandsFailed.push(e));
 
           const coll = client.db('db').collection('coll');