fixup: update schema-validation data service usage to use cursor

Author: Anemy

package-lock.json

@@ -66,6 +66,7 @@
     "compass-preferences-model": "^2.35.0",
     "hadron-app-registry": "^9.4.8",
     "lodash": "^4.17.21",
+    "mongodb": "^6.14.1",
     "mongodb-ns": "^2.4.2",
     "mongodb-schema": "^12.5.2",
     "react": "^17.0.2",

packages/compass-data-modeling/package.json

@@ -13,6 +13,7 @@ import { createLoggerLocator } from '@mongodb-js/compass-logging/provider';
 import { telemetryLocator } from '@mongodb-js/compass-telemetry/provider';
 import { SchemaValidationTabTitle } from './plugin-title';
 import { workspacesServiceLocator } from '@mongodb-js/compass-workspaces/provider';
+import type { RequiredDataServiceProps } from './modules';
 
 const CompassSchemaValidationHadronPlugin = registerHadronPlugin(
   {
@@ -23,13 +24,8 @@ const CompassSchemaValidationHadronPlugin = registerHadronPlugin(
     activate: onActivated,
   },
   {
-    dataService: dataServiceLocator as DataServiceLocator<
-      | 'aggregate'
-      | 'collectionInfo'
-      | 'updateCollection'
-      | 'sample'
-      | 'isCancelError'
-    >,
+    dataService:
+      dataServiceLocator as DataServiceLocator<RequiredDataServiceProps>,
     connectionInfoRef: connectionInfoRefLocator,
     instance: mongoDBInstanceLocator,
     preferences: preferencesLocator,

packages/compass-schema-validation/src/index.ts

@@ -67,14 +67,13 @@ export type RootAction =
   | EditModeAction
   | ResetAction;
 
-export type DataService = Pick<
-  OriginalDataService,
+export type RequiredDataServiceProps =
   | 'aggregate'
   | 'collectionInfo'
   | 'updateCollection'
-  | 'sample'
-  | 'isCancelError'
->;
+  | 'sampleCursor'
+  | 'isCancelError';
+export type DataService = Pick<OriginalDataService, RequiredDataServiceProps>;
 
 export type SchemaValidationExtraArgs = {
   dataService: DataService;

packages/compass-schema-validation/src/modules/index.ts

@@ -42,7 +42,12 @@ const fakeDataService = {
       /* never resolves */
     }),
   isCancelError: () => false,
-  sample: () => [{ prop1: 'abc' }],
+  sampleCursor: () =>
+    ({
+      *[Symbol.asyncIterator]() {
+        yield* [{ prop1: 'abc' }];
+      },
+    } as any),
 } as any;
 
 const fakeWorkspaces = {