@toa.io/storages.mongodb 1.0.0-alpha.154 → 1.0.0-alpha.158

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@toa.io/storages.mongodb",
-  "version": "1.0.0-alpha.154",
+  "version": "1.0.0-alpha.158",
   "description": "Toa MongoDB Storage Connector",
   "author": "temich <tema.gurtovoy@gmail.com>",
   "homepage": "https://github.com/toa-io/toa#readme",
@@ -20,12 +20,12 @@
   },
   "dependencies": {
     "@toa.io/conveyor": "1.0.0-alpha.93",
-    "@toa.io/core": "1.0.0-alpha.154",
+    "@toa.io/core": "1.0.0-alpha.156",
     "@toa.io/generic": "1.0.0-alpha.93",
     "@toa.io/pointer": "1.0.0-alpha.143",
     "mongodb": "6.7.0",
     "openspan": "1.0.0-alpha.93",
     "saslprep": "1.0.3"
   },
-  "gitHead": "01832fc47f9906d6a63180d5c2813db86763506a"
+  "gitHead": "958edd392a05ed179e174a10a1c132d4e49daa01"
 }

package/src/storage.js

@@ -47,13 +47,25 @@ class Storage extends Connector {
   }
 
   async find (query) {
-    const { criteria, options } = translate(query)
+    const { criteria, options, sample } = translate(query)
 
     criteria._deleted = null
 
-    this.debug('find', { criteria, options })
+    let cursor
+
+    if (sample === undefined) {
+      this.debug('find', { criteria, options })
 
-    const recordset = await this.#collection.find(criteria, options).toArray()
+      cursor = this.#collection.find(criteria, options)
+    } else {
+      const pipeline = toPipeline(criteria, options, sample)
+
+      this.debug('aggregate', { pipeline })
+
+      cursor = this.#collection.aggregate(pipeline)
+    }
+
+    const recordset = await cursor.toArray()
 
     return recordset.map((item) => from(item))
   }
@@ -255,6 +267,24 @@ class Storage extends Connector {
   }
 }
 
+function toPipeline (criteria, options, sample) {
+  const pipeline = []
+
+  if (criteria !== undefined)
+    pipeline.push({ $match: criteria })
+
+  if (sample !== undefined)
+    pipeline.push({ $sample: { size: sample } })
+
+  if (options?.sort !== undefined)
+    pipeline.push({ $sort: options.sort })
+
+  if (options?.projection !== undefined)
+    pipeline.push({ $project: options.projection })
+
+  return pipeline
+}
+
 const INDEX_TYPES = {
   'asc': 1,
   'desc': -1,

package/src/translate.js

@@ -9,7 +9,8 @@ const parse = { ...require('./translate/criteria'), ...require('./translate/opti
 const translate = (query) => {
   const result = {
     criteria: query?.criteria === undefined ? {} : parse.criteria(query.criteria),
-    options: query?.options === undefined ? {} : parse.options(query.options)
+    options: query?.options === undefined ? {} : parse.options(query.options),
+    sample: query?.options?.sample
   }
 
   if (query?.id !== undefined)