hack

Author: baileympearson

src/operations/execute_operation.ts

@@ -37,7 +37,7 @@ import {
   supportsRetryableWrites
 } from '../utils';
 import { AggregateOperation } from './aggregate';
-import { AbstractOperation, Aspect, RetryContext } from './operation';
+import { AbstractOperation, Aspect } from './operation';
 
 const MMAPv1_RETRY_WRITES_ERROR_CODE = MONGODB_ERROR_CODES.IllegalOperation;
 const MMAPv1_RETRY_WRITES_ERROR_MESSAGE =
@@ -254,17 +254,17 @@ async function executeOperationWithRetries<
     2 // backoff rate
   );
 
-  const retryContext =
-    operation.retryContext ??
-    new RetryContext(willRetry ? (timeoutContext.csotEnabled() ? Infinity : 2) : 1);
+  let maxAttempts =
+    (operation.maxAttempts ?? willRetry) ? (timeoutContext.csotEnabled() ? Infinity : 2) : 1;
+
   for (
     let attempt = 0;
-    attempt < retryContext.maxAttempts;
+    attempt < maxAttempts;
     attempt++,
-      retryContext.maxAttempts =
+      maxAttempts =
         willRetry && previousOperationError?.hasErrorLabel(MongoErrorLabel.SystemOverloadedError)
           ? 6
-          : retryContext.maxAttempts
+          : maxAttempts
   ) {
     if (previousOperationError) {
       if (hasWriteAspect && previousOperationError.code === MMAPv1_RETRY_WRITES_ERROR_CODE) {

src/operations/operation.ts

@@ -45,10 +45,6 @@ export interface OperationOptions extends BSONSerializeOptions {
   timeoutMS?: number;
 }
 
-export class RetryContext {
-  constructor(public maxAttempts: number) {}
-}
-
 /**
  * This class acts as a parent class for any operation and is responsible for setting this.options,
  * as well as setting and getting a session.
@@ -70,7 +66,7 @@ export abstract class AbstractOperation<TResult = any> {
   /** Specifies the time an operation will run until it throws a timeout error. */
   timeoutMS?: number;
 
-  retryContext?: RetryContext;
+  maxAttempts?: number;
 
   private _session: ClientSession | undefined;
 

src/sessions.ts

@@ -25,7 +25,7 @@ import {
 import type { MongoClient, MongoOptions } from './mongo_client';
 import { TypedEventEmitter } from './mongo_types';
 import { executeOperation } from './operations/execute_operation';
-import { RetryContext } from './operations/operation';
+import { RetryAttemptContext } from './operations/operation';
 import { RunCommandOperation } from './operations/run_command';
 import { ReadConcernLevel } from './read_concern';
 import { ReadPreference } from './read_preference';
@@ -494,14 +494,14 @@ export class ClientSession
       command.recoveryToken = this.transaction.recoveryToken;
     }
 
-    const retryContext = new RetryContext(5);
+    const retryContext = new RetryAttemptContext(5);
 
     const operation = new RunCommandOperation(new MongoDBNamespace('admin'), command, {
       session: this,
       readPreference: ReadPreference.primary,
       bypassPinningCheck: true
     });
-    operation.retryContext = retryContext;
+    operation.attempts = retryContext;
 
     const timeoutContext =
       this.timeoutContext ??
@@ -531,7 +531,7 @@ export class ClientSession
             readPreference: ReadPreference.primary,
             bypassPinningCheck: true
           });
-          op.retryContext = retryContext;
+          op.attempts = retryContext;
           await executeOperation(this.client, op, timeoutContext);
           return;
         } catch (retryCommitError) {