sworm/mysqlDriver.js

a lightweight write-only ORM for MSSQL, MySQL, PostgreSQL, Oracle, Sqlite 3

var promisify = require('./promisify');
var optionalRequire = require("./optionalRequire");
var debug = require('debug')('sworm:mysql');
var paramRegex = require('./paramRegex')

module.exports = function() {
  var mysql = optionalRequire("mysql");

  return {
    query: function(query, params, options) {
      var self = this;
      var paramList = [];

      if (params) {
        query = query.replace(paramRegex, function(_, paramName) {
          if (!params.hasOwnProperty(paramName)) {
            throw new Error('no such parameter @' + paramName);
          } else {
            paramList.push(params[paramName]);
          }
          return '?';
        });
      }

      return promisify(function(cb) {
        debug(query, paramList);
        return self.connection.query(query, paramList, cb)
      }).then(function (result) {
        if (options.insert || options.statement) {
          return {
            id: params.hasOwnProperty(options.id) ? params[options.id] : result.insertId,
            changes: result.affectedRows + result.changedRows
          }
        } else {
          return result
        }
      })
    },

    insert: function(query, params, options) {
      return this.query(query, params, options)
    },

    connect: function(config) {
      var self = this;

      config.config.multipleStatements = true;
      self.connection = mysql.createConnection(config.config);

      return promisify(function(cb) {
        return self.connection.connect(cb);
      });
    },

    close: function() {
      var self = this;
      return promisify(function (cb) { self.connection.end(cb); });
    },

    insertEmpty: function(meta) {
      return 'insert into ' + meta.table + ' () values ()';
    }
  };
};