add some options to the sample aggregate

Author: lerouxb

packages/shell-api/src/collection.ts

@@ -48,30 +48,30 @@ import {
   onlyShardedCollectionsInConfigFilter,
   aggregateBackgroundOptionNotSupportedHelp,
 } from './helpers';
-import type {
-  AnyBulkWriteOperation,
-  BulkWriteOptions,
-  CountOptions,
-  CountDocumentsOptions,
-  ChangeStreamOptions,
-  CreateIndexesOptions,
-  DeleteOptions,
-  DistinctOptions,
-  Document,
-  EstimatedDocumentCountOptions,
-  ExplainVerbosityLike,
-  FindOneAndDeleteOptions,
-  FindOneAndReplaceOptions,
-  FindOneAndUpdateOptions,
-  FindOptions,
-  InsertOneOptions,
-  ReplaceOptions,
-  RunCommandOptions,
-  UpdateOptions,
-  DropCollectionOptions,
-  CheckMetadataConsistencyOptions,
-  AggregateOptions,
-  SearchIndexDescription,
+import {
+  type AnyBulkWriteOperation,
+  type BulkWriteOptions,
+  type CountOptions,
+  type CountDocumentsOptions,
+  type ChangeStreamOptions,
+  type CreateIndexesOptions,
+  type DeleteOptions,
+  type DistinctOptions,
+  type Document,
+  type EstimatedDocumentCountOptions,
+  type ExplainVerbosityLike,
+  type FindOneAndDeleteOptions,
+  type FindOneAndReplaceOptions,
+  type FindOneAndUpdateOptions,
+  type FindOptions,
+  type InsertOneOptions,
+  type ReplaceOptions,
+  type RunCommandOptions,
+  type UpdateOptions,
+  type DropCollectionOptions,
+  type CheckMetadataConsistencyOptions,
+  type AggregateOptions,
+  type SearchIndexDescription,
 } from '@mongosh/service-provider-core';
 import type { RunCommandCursor, Database, DatabaseWithSchema } from './index';
 import {
@@ -2389,7 +2389,7 @@ export class Collection<
   ): Promise<RunCommandCursor> {
     this._emitCollectionApiCall('checkMetadataConsistency', { options });
 
-    return this._database._runCursorCommand({
+    return await this._database._runCursorCommand({
       checkMetadataConsistency: this._name,
     });
   }
@@ -2531,7 +2531,11 @@ export class Collection<
 
   async _getSampleDocs(): Promise<Document[]> {
     this._cachedSampleDocs = await (
-      await this.aggregate([{ $sample: { size: 10 } }])
+      await this.aggregate([{ $sample: { size: 10 } }], {
+        allowDiskUse: true,
+        maxTimeMS: 1000,
+        readPreference: 'secondaryPreferred',
+      })
     ).toArray();
     return this._cachedSampleDocs;
   }