feat: expose db driver to reduce dependency in consumer

tubero64 · Jan 4, 2023 · 7f3e511 · 7f3e511
1 parent 538c823
commit 7f3e511
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/packages/database/src/database.js b/packages/database/src/database.js
@@ -119,6 +119,10 @@ class Database extends Clonable {
     return this.collections[name];
   }
 
+  createId(key) {
+    return this.getAdapter().createId(key);
+  }
+
   find(name, condition, limit) {
     return this.getAdapter(name).find(name, condition, limit);
   }

diff --git a/packages/mongodb-adapter/src/mongodb-adapter.js b/packages/mongodb-adapter/src/mongodb-adapter.js
@@ -21,9 +21,10 @@
  * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  */
 
-const { MongoClient, ObjectId } = require('mongodb');
+const mongodb = require('mongodb');
 const { Clonable } = require('@nlpjs/core');
 
+const { MongoClient, ObjectId } = mongodb;
 const idField = '_id';
 
 class MongodbAdapter extends Clonable {
@@ -52,6 +53,7 @@ class MongodbAdapter extends Clonable {
       useUnifiedTopology: true,
     });
     this.registerDefault();
+    this.driver = mongodb;
   }
 
   registerDefault() {
@@ -134,6 +136,10 @@ class MongodbAdapter extends Clonable {
     });
   }
 
+  createId(key) {
+    return new ObjectId(key);
+  }
+
   async find(name, condition, limit, offset, sort) {
     return this.executeInCollection(name, (collection, cb) => {
       const options = {};