wingbot-mongodb 4.0.0 → 4.1.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "wingbot-mongodb",
-  "version": "4.0.0",
+  "version": "4.1.1",
   "description": "MongoDB storage for wingbot.ai",
   "main": "src/main.js",
   "scripts": {

package/src/BaseStorage.js

@@ -9,7 +9,7 @@ const crypto = require('crypto');
 const defaultLogger = require('./defaultLogger');
 
 /** @typedef {import('mongodb').Db} Db */
-/** @typedef {import('mongodb').Collection} Collection */
+/** @typedef {import('mongodb').Document} Document */
 /** @typedef {import('mongodb').CreateIndexesOptions} CreateIndexesOptions */
 
 /**
@@ -49,6 +49,9 @@ function signReplacer (k, v) {
     return v;
 }
 
+/**
+ * @template T={} {Document}
+ */
 class BaseStorage {
 
     static netFailuresIntervalMs = 600000; // 10 minutes
@@ -108,7 +111,7 @@ class BaseStorage {
         this._log = log;
 
         /**
-         * @type {Collection|Promise<Collection>}
+         * @type {import('mongodb').Collection<T>|Promise<import('mongodb').Collection<T>>}
          */
         this._collection = null;
 
@@ -189,7 +192,7 @@ class BaseStorage {
     /**
      * Insert defalt document to DB
      *
-     * @param  {...{ _id: string|number|ObjectId }} objects
+     * @param  {...T} objects
      */
     addFixtureDoc (...objects) {
         this._fixtures.push(...objects);
@@ -241,17 +244,24 @@ class BaseStorage {
         return getNestedObjects(obj, nested, attr);
     }
 
+    /**
+     *
+     * @param {string} name
+     * @returns {Promise<import('mongodb').Collection<T>>}
+     */
     async _getOrCreateCollection (name) {
         const db = typeof this._mongoDb === 'function'
             ? await this._mongoDb()
             : this._mongoDb;
 
+        /** @type {import('mongodb').Collection<T>} */
         let collection;
 
         if (this._isCosmo) {
             // @ts-ignore
             const collections = await db.collections();
 
+            // @ts-ignore
             collection = collections
                 .find((c) => c.collectionName === name);
 
@@ -290,7 +300,7 @@ class BaseStorage {
      *
      * @protected
      * @param {boolean} [forRead]
-     * @returns {Promise<Collection>}
+     * @returns {Promise<import('mongodb').Collection<T>>}
      */
     async _getCollection (forRead = false) {
         if (this._collection === null) {
@@ -321,7 +331,7 @@ class BaseStorage {
     /**
      *
      * @param {object[]} indexes
-     * @param {Collection} collection
+     * @param {import('mongodb').Collection<T>} collection
      * @returns {Promise}
      */
     async _ensureIndexes (indexes, collection) {
@@ -480,6 +490,25 @@ class BaseStorage {
 
         return h.digest('base64');
     }
+
+    /**
+     * Drop collection
+     *
+     * @returns {Promise}
+     */
+    async drop () {
+        const db = typeof this._mongoDb === 'function'
+            ? await this._mongoDb()
+            : this._mongoDb;
+
+        await db.dropCollection(this._collectionName)
+            .catch(() => {});
+
+        this._collection = null;
+        this._indexing = null;
+        this._shouldIndexBeforeRead = null;
+        this._shouldWaitForIndex = null;
+    }
 }
 
 module.exports = BaseStorage;