[feature] support backend: mongodb #25

Author: 高鸣飞

.gitignore

@@ -1,3 +1,4 @@
 node_modules
 .*
-package-lock.json
+package-lock.json
+yarn.lock

README.md

@@ -15,6 +15,7 @@
 - [Usage](#usage)
   - [database list](#database-list)
   - [feature list](#feature-list)
+  - [api support](#api-support)
   - [interface](#interface)
     - [overall](#overall)
     - [action whitelist: operator_info.actions](#action-whitelist-operator_infoactions)
@@ -32,13 +33,26 @@ database driver with extended features.
 
 ## database list
 * mysql
+* mongodb
 * redis
 * define custom function to use any database or service
 
 ## feature list
 * changelog/oplog
 * auto release timeout connection
 
+## api support
+
+|  Method  | MySQL  | MongoDB | Redis |
+| :------- |:------:| :------:| :----:|
+| select   |   √    |    √    |   √   |
+| selectEx |   √    |    ×    |   ×   |
+| insert   |   √    |    √    |   √   |
+| update   |   √    |    √    |   √   |
+| delete   |   √    |    √    |   √   |
+| find     |   ×    |    √    |   ×   |
+| aggregate|   ×    |    √    |   ×   |
+
 ## interface
 
 
@@ -179,7 +193,7 @@ const InterfaceTwo = {
 ### method
 ```javascript
 // callback
-conn.selectEx(
+conn.selectEx(    // mysql only
   /* table */ 'table0',
   /* custom sql */ 'select ... join ...where field1 = ? and field2 = ?',
   /* where */ [1,2]
@@ -219,6 +233,23 @@ conn.delete(
   function(err, rows, info) {
   })
 
+conn.find(        // mongodb only
+  /* table */ 'table5',
+  /* opt   */ {fields: ['field1'], skip: 10, sort: 'field1'},
+  /* where */ {field1: 10},
+  function(err, rows, info) {
+  })
+
+conn.aggregate(   // mongodb only
+  /* table */ 'table5',
+  /* opt   */ {skip: 10, sort: 'name'},
+  /* pipe  */ [
+                {$match: {field1: {$regex: /[a-z]*/}}},
+                {$sort:  {field1: -1}},
+              ]
+  function(err, rows, info) {
+  })
+
 // Promise/async/await
 let result
 try {

lib/$/index.js

@@ -8,6 +8,8 @@ ALL_METHODS: [
   'queryReadonly',
   'release',
   'selectEx',
+  'find',
+  'aggregate',
 ],
 
 util: require('util'),

lib/Connection.js

@@ -72,7 +72,7 @@ $.ALL_METHODS.forEach(function(method) {
       callback(new Error('connection timeout.'))
     }, this._opts.connection_timeout)
 
-    const start_time = +new Date()
+    const start_time = + new Date()
     const that = this
     const args = arguments
     arguments[arguments.length - 1] = function() {

lib/define/mongodb.js

@@ -0,0 +1,163 @@
+module.exports.Context = class {
+
+constructor(config) {
+  if (!config.config.url) {
+    throw new Error('url is required for mongodb connection')
+  }
+  this._config = config.config
+}
+
+getConnection() {
+  return new MongoDB(this._config)
+}
+
+}
+
+class MongoDB {
+
+constructor(config) {
+  this._config = config
+  this._readonly = config.readonly
+  this._released = false
+  this._conn = monk(config.url)
+  log.trace('mongodb connection established')
+}
+
+__destroy() {
+  if (this._conn._state === 'closed') {
+    return
+  }
+  this._conn.close()
+}
+
+/**
+ * MongoDB.insert()
+ * 
+ * @param {string} table 
+ * @param {object|Array} sets 
+ * @param {function} callback 
+ */
+insert(table, sets, callback) {
+  if (sets.constructor !== Object && sets.constructor !== Array) {
+    return callback(new Error('invalid sets'))
+  }
+  if (this._readonly) {
+    return errNotAllowd(callback)
+  }
+  this._conn.collection(table).insert(sets, (err, result) => {
+    if (result.constructor === Object) {
+      result = [result]
+    }
+    return callback(err, {
+      affected_rows: result.length,
+      docs: result,
+    }, null)
+  })
+}
+
+update(table, sets, where, callback) {
+  if (sets.constructor !== Object && sets.constructor !== Array) {
+    return callback(new Error('invalid sets'))
+  }
+  if (this._readonly) {
+    return errNotAllowd(callback)
+  }
+  this._conn.collection(table)
+    .update(where, {$set: sets}, {multi: true}, (err, result) => {
+      return callback(err, {
+        affected_rows: result.n,
+        changed_rows: result.nModified,
+      }, null)
+  })
+}
+
+/**
+ * Mongodb.remove()
+ * 
+ * @param {string} table 
+ * @param {object} where 
+ * @param {function} callback
+ */
+delete(table, where, callback) {
+  if (this._readonly) {
+    return errNotAllowd(callback)
+  }
+  this._conn.collection(table).remove(where, (err, result) => {
+    return callback(err, {
+      affected_rows: result.deletedCount,
+    }, null)
+  })
+}
+
+/**
+ * Mongodb.find()
+ * 
+ * @param {string} table 
+ * @param {string|array} options 
+ * @param {string|object_id|object} where 
+ * @param {function} callback 
+ */
+select(table, fields, where, callback) {
+  let _fields = fields === '*' ? {} : fields
+  this._conn.collection(table).find(where, _fields, (err, docs) => {
+    return callback(err, docs, null)
+  })
+}
+
+/**
+ * Mongodb.find()
+ * support sort/fileds/limit/skip/rawCursor in options
+ * 
+ * @param {string} table 
+ * @param {object} options 
+ * @param {string|object_id|object} where 
+ * @param {function} callback 
+ */
+find(table, options, where, callback) {
+  this._conn.collection(table).find(where, options, (err, docs) => {
+    return callback(err, docs, null)
+  })
+}
+
+/**
+ * Mongodb.close()
+ */
+release() {
+  this._released = true
+  if (!this._conn._state === 'open') return
+  this._conn.close()
+  this._conn = null
+}
+
+/**
+ * Mongodb.aggregate()
+ * 
+ * @param {string} table 
+ * @param {object} options 
+ * @param {array} pipeline 
+ * @param {function} callback 
+ */
+aggregate(table, options, pipeline, callback) {
+  this._conn.collection(table).aggregate(pipeline, options, (err, docs) => {
+    return callback(err, docs, null)
+  })
+}
+
+}
+
+const util = require('util')
+const debug = require('debug')('sqlx')
+const monk = require('monk')
+function __log() {
+  debug(util.format.apply(null, arguments))
+}
+const log = {
+  trace: __log,
+  error: __log,
+  debug: __log,
+  info : __log,
+}
+
+function errNotAllowd(callback) {
+  return callback(new Error('not allowed'))
+}

package.json

@@ -1,6 +1,6 @@
 {
   "name": "sqlx",
-  "version": "4.0.0",
+  "version": "4.1.0",
   "description": "Database driver with extended features like mysql changelog/oplog, connection auto release.",
   "main": "index.js",
   "directories": {
@@ -22,7 +22,7 @@
     "release"
   ],
   "author": "yinrong",
-  "license": "ISC",
+  "license": "MIT",
   "bugs": {
     "url": "https://github.com/yinrong/node-sqlx/issues"
   },
@@ -31,6 +31,7 @@
     "async": "^2.0.1",
     "debug": "^2.2.0",
     "mongo-sql": "^2.7.5",
+    "monk": "^6.0.5",
     "mysql": "^2.11.1",
     "redis": "^2.8.0"
   },