feat(NODE-6883): allow rawData option on time series collections (#4642)

Co-authored-by: Bailey Pearson <[email protected]>

Author: durran

src/cmap/wire_protocol/constants.ts

@@ -1,9 +1,11 @@
 export const MIN_SUPPORTED_SERVER_VERSION = '4.2';
-export const MAX_SUPPORTED_SERVER_VERSION = '8.0';
+export const MAX_SUPPORTED_SERVER_VERSION = '8.2';
 export const MIN_SUPPORTED_WIRE_VERSION = 8;
-export const MAX_SUPPORTED_WIRE_VERSION = 25;
+export const MAX_SUPPORTED_WIRE_VERSION = 27;
 export const MIN_SUPPORTED_QE_WIRE_VERSION = 21;
 export const MIN_SUPPORTED_QE_SERVER_VERSION = '7.0';
+export const MIN_SUPPORTED_RAW_DATA_WIRE_VERSION = 27;
+export const MIN_SUPPORTED_RAW_DATA_SERVER_VERSION = '8.2';
 export const OP_REPLY = 1;
 export const OP_UPDATE = 2001;
 export const OP_INSERT = 2002;

src/operations/aggregate.ts

@@ -161,5 +161,6 @@ defineAspects(AggregateOperation, [
   Aspect.READ_OPERATION,
   Aspect.RETRYABLE,
   Aspect.EXPLAINABLE,
-  Aspect.CURSOR_CREATING
+  Aspect.CURSOR_CREATING,
+  Aspect.SUPPORTS_RAW_DATA
 ]);

src/operations/client_bulk_write/client_bulk_write.ts

@@ -67,5 +67,6 @@ defineAspects(ClientBulkWriteOperation, [
   Aspect.SKIP_COLLATION,
   Aspect.CURSOR_CREATING,
   Aspect.RETRYABLE,
-  Aspect.COMMAND_BATCHING
+  Aspect.COMMAND_BATCHING,
+  Aspect.SUPPORTS_RAW_DATA
 ]);

src/operations/command.ts

@@ -1,5 +1,6 @@
 import { type Connection } from '..';
 import type { BSONSerializeOptions, Document } from '../bson';
+import { MIN_SUPPORTED_RAW_DATA_WIRE_VERSION } from '../cmap/wire_protocol/constants';
 import { MongoInvalidArgumentError } from '../error';
 import {
   decorateWithExplain,
@@ -12,7 +13,7 @@ import type { ReadPreference } from '../read_preference';
 import type { ServerCommandOptions } from '../sdam/server';
 import type { ClientSession } from '../sessions';
 import { type TimeoutContext } from '../timeout';
-import { commandSupportsReadConcern, MongoDBNamespace } from '../utils';
+import { commandSupportsReadConcern, maxWireVersion, MongoDBNamespace } from '../utils';
 import { WriteConcern, type WriteConcernOptions } from '../write_concern';
 import type { ReadConcernLike } from './../read_concern';
 import { AbstractOperation, Aspect, type OperationOptions } from './operation';
@@ -63,6 +64,14 @@ export interface CommandOperationOptions
    * This option is deprecated and will be removed in an upcoming major version.
    */
   noResponse?: boolean;
+
+  /**
+   * Used when the command needs to grant access to the underlying namespaces for time series collections.
+   * Only available on server versions 8.2 and above.
+   * @public
+   * @sinceServerVersion 8.2
+   **/
+  rawData?: boolean;
 }
 
 /** @internal */
@@ -153,6 +162,14 @@ export abstract class CommandOperation<T> extends AbstractOperation<T> {
       command.maxTimeMS = this.options.maxTimeMS;
     }
 
+    if (
+      this.options.rawData != null &&
+      this.hasAspect(Aspect.SUPPORTS_RAW_DATA) &&
+      maxWireVersion(connection) >= MIN_SUPPORTED_RAW_DATA_WIRE_VERSION
+    ) {
+      command.rawData = this.options.rawData;
+    }
+
     if (this.hasAspect(Aspect.EXPLAINABLE) && this.explain) {
       return decorateWithExplain(command, this.explain);
     }

src/operations/count.ts

@@ -71,4 +71,4 @@ export class CountOperation extends CommandOperation<number> {
   }
 }
 
-defineAspects(CountOperation, [Aspect.READ_OPERATION, Aspect.RETRYABLE]);
+defineAspects(CountOperation, [Aspect.READ_OPERATION, Aspect.RETRYABLE, Aspect.SUPPORTS_RAW_DATA]);

src/operations/create_collection.ts

@@ -65,7 +65,7 @@ export interface ClusteredCollectionOptions extends Document {
 }
 
 /** @public */
-export interface CreateCollectionOptions extends CommandOperationOptions {
+export interface CreateCollectionOptions extends Omit<CommandOperationOptions, 'rawData'> {
   /** Create a capped collection */
   capped?: boolean;
   /** @deprecated Create an index on the _id field of the document. This option is deprecated in MongoDB 3.2+ and will be removed once no longer supported by the server. */

src/operations/delete.ts

@@ -163,15 +163,21 @@ export function makeDeleteStatement(
   return op;
 }
 
-defineAspects(DeleteOperation, [Aspect.RETRYABLE, Aspect.WRITE_OPERATION]);
+defineAspects(DeleteOperation, [
+  Aspect.RETRYABLE,
+  Aspect.WRITE_OPERATION,
+  Aspect.SUPPORTS_RAW_DATA
+]);
 defineAspects(DeleteOneOperation, [
   Aspect.RETRYABLE,
   Aspect.WRITE_OPERATION,
   Aspect.EXPLAINABLE,
-  Aspect.SKIP_COLLATION
+  Aspect.SKIP_COLLATION,
+  Aspect.SUPPORTS_RAW_DATA
 ]);
 defineAspects(DeleteManyOperation, [
   Aspect.WRITE_OPERATION,
   Aspect.EXPLAINABLE,
-  Aspect.SKIP_COLLATION
+  Aspect.SKIP_COLLATION,
+  Aspect.SUPPORTS_RAW_DATA
 ]);

src/operations/distinct.ts

@@ -84,4 +84,9 @@ export class DistinctOperation extends CommandOperation<any[] | Document> {
   }
 }
 
-defineAspects(DistinctOperation, [Aspect.READ_OPERATION, Aspect.RETRYABLE, Aspect.EXPLAINABLE]);
+defineAspects(DistinctOperation, [
+  Aspect.READ_OPERATION,
+  Aspect.RETRYABLE,
+  Aspect.EXPLAINABLE,
+  Aspect.SUPPORTS_RAW_DATA
+]);

src/operations/drop.ts

@@ -11,7 +11,7 @@ import { executeOperation } from './execute_operation';
 import { Aspect, defineAspects } from './operation';
 
 /** @public */
-export interface DropCollectionOptions extends CommandOperationOptions {
+export interface DropCollectionOptions extends Omit<CommandOperationOptions, 'rawData'> {
   /** @experimental */
   encryptedFields?: Document;
 }

src/operations/estimated_document_count.ts

@@ -56,5 +56,6 @@ export class EstimatedDocumentCountOperation extends CommandOperation<number> {
 defineAspects(EstimatedDocumentCountOperation, [
   Aspect.READ_OPERATION,
   Aspect.RETRYABLE,
-  Aspect.CURSOR_CREATING
+  Aspect.CURSOR_CREATING,
+  Aspect.SUPPORTS_RAW_DATA
 ]);