loopback-connector-mysql/lib/mysql.js

1097 lines
32 KiB
JavaScript
Raw Normal View History

2013-10-03 21:50:38 +00:00
/*!
2012-12-14 14:01:44 +00:00
* Module dependencies
*/
var mysql = require('mysql');
2013-11-20 18:54:01 +00:00
var juggler = require('loopback-datasource-juggler');
2013-07-21 05:37:59 +00:00
var EnumFactory = require('./enumFactory').EnumFactory;
2012-12-14 14:01:44 +00:00
2013-10-03 21:50:38 +00:00
/**
* @module loopback-connector-mysql
*
* Initialize the MySQL connector against the given data source
*
* @param {DataSource} dataSource The loopback-datasource-juggler dataSource
* @param {Function} [callback] The callback function
*/
exports.initialize = function initializeDataSource(dataSource, callback) {
2013-10-03 21:50:38 +00:00
if (!mysql) {
return;
}
2012-12-14 14:01:44 +00:00
var s = dataSource.settings;
2013-11-27 01:40:31 +00:00
if (s.collation) {
2013-11-27 01:40:31 +00:00
s.charset = s.collation.substr(0, s.collation.indexOf('_')); // Charset should be first 'chunk' of collation.
} else {
2013-07-22 00:27:54 +00:00
s.collation = 'utf8_general_ci';
s.charset = 'utf8';
}
2013-11-27 01:40:31 +00:00
s.supportBigNumbers = (s.supportBigNumbers || false);
s.timezone = (s.timezone || 'local');
2013-11-27 01:40:31 +00:00
if(isNaN(s.connectionLimit)) {
s.connectionLimit = s.connectionLimit;
} else {
s.connectionLimit = 10;
}
var options = {
2013-08-21 20:37:01 +00:00
host: s.host || s.hostname || 'localhost',
2012-12-14 14:01:44 +00:00
port: s.port || 3306,
2013-08-21 20:37:01 +00:00
user: s.username || s.user,
2012-12-14 14:01:44 +00:00
password: s.password,
timezone: s.timezone,
2012-12-14 14:01:44 +00:00
debug: s.debug,
socketPath: s.socketPath,
charset: s.collation.toUpperCase(), // Correct by docs despite seeming odd.
2013-11-27 02:50:56 +00:00
supportBigNumbers: s.supportBigNumbers,
connectionLimit: s.connectionLimit
};
// Don't configure the DB if the pool can be used for multiple DBs
if(!s.createDatabase) {
options.database = s.database;
}
dataSource.client = mysql.createPool(options);
2012-12-14 14:01:44 +00:00
dataSource.client.on('error', function (err) {
dataSource.emit('error', err);
dataSource.connected = false;
dataSource.connecting = false;
});
2013-11-27 01:40:31 +00:00
if (s.debug) {
console.log('Settings: ', s);
}
dataSource.connector = new MySQL(dataSource.client, s);
dataSource.connector.dataSource = dataSource;
// MySQL specific column types
2013-11-27 01:40:31 +00:00
juggler.ModelBuilder.registerType(function Point() {
});
dataSource.EnumFactory = EnumFactory; // factory for Enums. Note that currently Enums can not be registered.
2013-11-27 01:40:31 +00:00
2013-11-27 02:50:56 +00:00
process.nextTick(function() {
callback && callback();
});
2012-12-14 14:01:44 +00:00
};
2013-07-21 06:38:40 +00:00
exports.MySQL = MySQL;
2012-12-14 14:01:44 +00:00
/**
2013-10-03 21:50:38 +00:00
* @constructor
* Constructor for MySQL connector
* @param {Object} client The node-mysql client object
2012-12-14 14:01:44 +00:00
*/
function MySQL(client, settings) {
2013-01-30 15:15:32 +00:00
this.name = 'mysql';
2012-12-14 14:01:44 +00:00
this._models = {};
this.client = client;
this.settings = settings;
2012-12-14 14:01:44 +00:00
}
2013-11-20 18:54:01 +00:00
require('util').inherits(MySQL, juggler.BaseSQL);
2012-12-14 14:01:44 +00:00
2013-10-03 21:50:38 +00:00
/**
* Execute the sql statement
*
* @param {String} sql The SQL statement
* @param {Function} [callback] The callback after the SQL statement is executed
*/
2012-12-14 14:01:44 +00:00
MySQL.prototype.query = function (sql, callback) {
2013-11-27 01:40:31 +00:00
var self = this;
if (!this.dataSource.connected) {
return this.dataSource.on('connected', function () {
2012-12-14 14:01:44 +00:00
this.query(sql, callback);
}.bind(this));
}
var client = this.client;
var time = Date.now();
var debug = this.settings.debug;
2013-11-27 01:40:31 +00:00
var db = this.settings.database;
2012-12-14 14:01:44 +00:00
var log = this.log;
if (typeof callback !== 'function') throw new Error('callback should be a function');
2013-11-27 01:40:31 +00:00
if (debug) {
console.log('SQL:', sql);
}
function releaseConnectionAndCallback(connection, err, result) {
connection.release();
callback && callback(err, result);
}
function runQuery(connection) {
connection.query(sql, function (err, data) {
if (debug) {
if (err) {
console.error('Error:', err);
}
console.log('Data:', data);
}
if (log) log(sql, time);
releaseConnectionAndCallback(connection, err, data);
});
}
2013-11-27 01:40:31 +00:00
client.getConnection(function (err, connection) {
if (err) {
callback && callback(err);
return;
}
if (self.settings.createDatabase) {
// Call USE db ...
connection.query('USE `' + db + '`', function (err) {
if (err) {
if (err && err.message.match(/(^|: )unknown database/i)) {
var charset = self.settings.charset;
var collation = self.settings.collation;
var q = 'CREATE DATABASE ' + db + ' CHARACTER SET ' + charset + ' COLLATE ' + collation;
connection.query(q, function (err) {
if (!err) {
connection.query('USE `' + db + '`', function (err) {
runQuery(connection);
2013-11-27 01:40:31 +00:00
});
} else {
releaseConnectionAndCallback(connection, err);
}
});
return;
} else {
releaseConnectionAndCallback(connection, err);
return;
}
2012-12-14 14:01:44 +00:00
}
runQuery(connection);
});
} else {
// Bypass USE db
runQuery(connection);
}
2012-12-14 14:01:44 +00:00
});
};
2013-11-27 01:40:31 +00:00
2012-12-14 14:01:44 +00:00
/**
2013-10-03 21:50:38 +00:00
* Create the data model in MySQL
*
* @param {String} model The model name
* @param {Object} data The model instance data
* @param {Function} [callback] The callback function
2012-12-14 14:01:44 +00:00
*/
MySQL.prototype.create = function (model, data, callback) {
var fields = this.toFields(model, data);
var sql = 'INSERT INTO ' + this.tableEscaped(model);
if (fields) {
sql += ' SET ' + fields;
} else {
sql += ' VALUES ()';
}
this.query(sql, function (err, info) {
callback(err, info && info.insertId);
});
};
2013-10-03 21:50:38 +00:00
/**
* Update if the model instance exists with the same id or create a new instance
*
* @param {String} model The model name
* @param {Object} data The model instance data
* @param {Function} [callback] The callback function
*/
2012-12-14 14:01:44 +00:00
MySQL.prototype.updateOrCreate = function (model, data, callback) {
var mysql = this;
var fieldsNames = [];
var fieldValues = [];
var combined = [];
var props = this._models[model].properties;
Object.keys(data).forEach(function (key) {
2013-07-21 17:36:26 +00:00
if (props[key] || mysql.id(model, key)) {
2013-10-14 21:32:19 +00:00
var k = mysql.columnEscaped(model, key);
2012-12-14 14:01:44 +00:00
var v;
2013-07-21 17:36:26 +00:00
if (!mysql.id(model, key)) {
2012-12-14 14:01:44 +00:00
v = mysql.toDatabase(props[key], data[key]);
} else {
v = data[key];
}
fieldsNames.push(k);
fieldValues.push(v);
2013-07-21 17:36:26 +00:00
if (!mysql.id(model, key)) combined.push(k + ' = ' + v);
2012-12-14 14:01:44 +00:00
}
});
var sql = 'INSERT INTO ' + this.tableEscaped(model);
sql += ' (' + fieldsNames.join(', ') + ')';
sql += ' VALUES (' + fieldValues.join(', ') + ')';
sql += ' ON DUPLICATE KEY UPDATE ' + combined.join(', ');
this.query(sql, function (err, info) {
if (!err && info && info.insertId) {
data.id = info.insertId;
}
callback(err, data);
});
};
MySQL.prototype.toFields = function (model, data) {
2013-10-14 21:32:19 +00:00
var self = this;
2012-12-14 14:01:44 +00:00
var fields = [];
var props = this._models[model].properties;
Object.keys(data).forEach(function (key) {
if (props[key]) {
2013-03-27 00:38:18 +00:00
var value = this.toDatabase(props[key], data[key]);
if ('undefined' === typeof value) return;
2013-10-14 21:32:19 +00:00
fields.push(self.columnEscaped(model, key) + ' = ' + value);
2012-12-14 14:01:44 +00:00
}
}.bind(this));
return fields.join(',');
};
function dateToMysql(val) {
return val.getUTCFullYear() + '-' +
fillZeros(val.getUTCMonth() + 1) + '-' +
fillZeros(val.getUTCDate()) + ' ' +
fillZeros(val.getUTCHours()) + ':' +
fillZeros(val.getUTCMinutes()) + ':' +
fillZeros(val.getUTCSeconds());
function fillZeros(v) {
return v < 10 ? '0' + v : v;
}
}
2013-10-03 21:50:38 +00:00
/*!
* Convert property name/value to a DB column
* @param prop
* @param val
* @returns {*}
*/
2012-12-14 14:01:44 +00:00
MySQL.prototype.toDatabase = function (prop, val) {
if (val === null) return 'NULL';
2013-03-27 00:38:18 +00:00
if (val === undefined) return;
2012-12-14 14:01:44 +00:00
if (val.constructor.name === 'Object') {
var operator = Object.keys(val)[0]
val = val[operator];
if (operator === 'between') {
return this.toDatabase(prop, val[0]) +
2013-11-27 01:40:31 +00:00
' AND ' +
this.toDatabase(prop, val[1]);
2012-12-14 14:01:44 +00:00
} else if (operator == 'inq' || operator == 'nin') {
if (!(val.propertyIsEnumerable('length')) && typeof val === 'object' && typeof val.length === 'number') { //if value is array
for (var i = 0; i < val.length; i++) {
val[i] = this.client.escape(val[i]);
}
return val.join(',');
} else {
return val;
}
}
}
if (!prop) return val;
2013-04-18 18:58:21 +00:00
if (prop.type.name === 'Number') return Number(val);
2012-12-14 14:01:44 +00:00
if (prop.type.name === 'Date') {
if (!val) return 'NULL';
if (!val.toUTCString) {
val = new Date(val);
}
return '"' + dateToMysql(val) + '"';
}
if (prop.type.name == "Boolean") return val ? 1 : 0;
if (prop.type.name === 'GeoPoint') {
2013-11-27 01:40:31 +00:00
return val ? 'Point(' + val.lat + ',' + val.lng + ')' : 'NULL';
}
if (typeof prop.type === 'function') return this.client.escape(prop.type(val));
2012-12-14 14:01:44 +00:00
return this.client.escape(val.toString());
};
2013-10-03 21:50:38 +00:00
/*!
* Convert the data from database
* @param model
* @param data
* @returns {*}
*/
2012-12-14 14:01:44 +00:00
MySQL.prototype.fromDatabase = function (model, data) {
if (!data) {
return null;
}
2012-12-14 14:01:44 +00:00
var props = this._models[model].properties;
var json = {};
2013-11-27 01:40:31 +00:00
for (var p in props) {
var key = this.column(model, p);
2012-12-14 14:01:44 +00:00
var val = data[key];
2013-04-06 10:36:32 +00:00
if (typeof val === 'undefined' || val === null) {
continue;
2013-04-06 10:36:32 +00:00
}
if (props[p]) {
2013-11-27 01:40:31 +00:00
switch (props[p].type.name) {
2013-04-06 10:36:32 +00:00
case 'Date':
2013-11-27 01:40:31 +00:00
val = new Date(val.toString().replace(/GMT.*$/, 'GMT'));
break;
2013-04-06 10:36:32 +00:00
case 'Boolean':
2013-11-27 01:40:31 +00:00
val = Boolean(val);
break;
2013-10-07 21:54:26 +00:00
case 'GeoPoint':
case 'Point':
2013-11-27 01:40:31 +00:00
val = {
lat: val.x,
lng: val.y
};
break;
2012-12-14 14:01:44 +00:00
}
}
json[p] = val;
}
return json;
2012-12-14 14:01:44 +00:00
};
MySQL.prototype.escapeName = function (name) {
return '`' + name.replace(/\./g, '`.`') + '`';
};
2013-11-27 01:40:31 +00:00
MySQL.prototype.getColumns = function (model, props) {
2013-07-21 05:37:59 +00:00
var cols = this._models[model].properties;
2013-11-27 01:40:31 +00:00
if (!cols) {
2013-07-22 00:27:54 +00:00
return '*';
}
2013-07-21 05:37:59 +00:00
var self = this;
var keys = Object.keys(cols);
if (Array.isArray(props) && props.length > 0) {
// No empty array, including all the fields
keys = props;
} else if ('object' === typeof props && Object.keys(props).length > 0) {
// { field1: boolean, field2: boolean ... }
var included = [];
var excluded = [];
keys.forEach(function (k) {
if (props[k]) {
included.push(k);
} else if ((k in props) && !props[k]) {
excluded.push(k);
}
});
if (included.length > 0) {
keys = included;
} else if (excluded.length > 0) {
excluded.forEach(function (e) {
var index = keys.indexOf(e);
keys.splice(index, 1);
});
}
}
2013-11-27 01:40:31 +00:00
var names = keys.map(function (c) {
2013-07-21 05:37:59 +00:00
return self.columnEscaped(model, c);
});
return names.join(', ');
};
function buildWhere(self, model, conds) {
var props = self._models[model].properties;
var cs = [];
Object.keys(conds).forEach(function (key) {
2013-10-14 21:32:19 +00:00
var keyEscaped = self.columnEscaped(model, key);
var val = self.toDatabase(props[key], conds[key]);
if (conds[key] === null || conds[key] === undefined) {
cs.push(keyEscaped + ' IS NULL');
} else if (conds[key] && conds[key].constructor.name === 'Object') {
var condType = Object.keys(conds[key])[0];
var sqlCond = keyEscaped;
if ((condType === 'inq' || condType === 'nin') && val.length === 0) {
cs.push(condType === 'inq' ? 0 : 1);
return true;
}
switch (condType) {
case 'gt':
sqlCond += ' > ';
break;
case 'gte':
sqlCond += ' >= ';
break;
case 'lt':
sqlCond += ' < ';
break;
case 'lte':
sqlCond += ' <= ';
break;
case 'between':
sqlCond += ' BETWEEN ';
break;
case 'inq':
sqlCond += ' IN ';
break;
case 'nin':
sqlCond += ' NOT IN ';
break;
case 'neq':
sqlCond += ' != ';
break;
}
sqlCond += (condType === 'inq' || condType === 'nin') ? '(' + val + ')' : val;
cs.push(sqlCond);
} else {
cs.push(keyEscaped + ' = ' + val);
}
});
if (cs.length === 0) {
return '';
}
return 'WHERE ' + cs.join(' AND ');
}
2013-10-14 21:32:19 +00:00
function buildOrderBy(self, model, order) {
if (typeof order === 'string') {
order = [order];
}
2013-11-27 01:40:31 +00:00
return 'ORDER BY ' + order.map(function (o) {
2013-08-30 15:51:32 +00:00
var t = o.split(/[\s,]+/);
2013-10-14 21:32:19 +00:00
if (t.length === 1) {
return self.columnEscaped(model, o);
}
return self.columnEscaped(model, t[0]) + ' ' + t[1];
}).join(', ');
}
function buildLimit(limit, offset) {
return 'LIMIT ' + (offset ? (offset + ', ' + limit) : limit);
2013-07-21 05:37:59 +00:00
}
2013-10-03 21:50:38 +00:00
/**
* Find matching model instances by the filter
*
* @param {String} model The model name
* @param {Object} filter The filter
* @param {Function} [callback] The callback function
*/
2012-12-14 14:01:44 +00:00
MySQL.prototype.all = function all(model, filter, callback) {
var self = this;
2013-07-21 05:37:59 +00:00
// Order by id if no order is specified
filter = filter || {};
2013-11-27 01:40:31 +00:00
if (!filter.order) {
2013-07-21 05:37:59 +00:00
var idNames = this.idNames(model);
2013-11-27 01:40:31 +00:00
if (idNames && idNames.length) {
2013-07-21 05:37:59 +00:00
filter.order = idNames.join(' ');
}
}
2012-12-14 14:01:44 +00:00
2013-11-27 01:40:31 +00:00
var sql = 'SELECT ' + this.getColumns(model, filter.fields) + ' FROM ' + this.tableEscaped(model);
2012-12-14 14:01:44 +00:00
if (filter) {
if (filter.where) {
sql += ' ' + buildWhere(self, model, filter.where);
2012-12-14 14:01:44 +00:00
}
if (filter.order) {
2013-10-14 21:32:19 +00:00
sql += ' ' + buildOrderBy(self, model, filter.order);
2012-12-14 14:01:44 +00:00
}
if (filter.limit) {
sql += ' ' + buildLimit(filter.limit, filter.skip || 0);
2012-12-14 14:01:44 +00:00
}
}
this.query(sql, function (err, data) {
if (err) {
return callback(err, []);
}
2012-12-16 17:57:39 +00:00
var objs = data.map(function (obj) {
2012-12-14 14:01:44 +00:00
return self.fromDatabase(model, obj);
2012-12-16 17:57:39 +00:00
});
if (filter && filter.include) {
this._models[model].model.include(objs, filter.include, callback);
} else {
callback(null, objs);
}
2012-12-14 14:01:44 +00:00
}.bind(this));
return sql;
};
2012-12-14 14:01:44 +00:00
/**
* Delete instances for the given model
*
* @param {String} model The model name
* @param {Object} [where] The filter for where
* @param {Function} [callback] The callback function
*
*/
MySQL.prototype.destroyAll = function destroyAll(model, where, callback) {
2013-11-27 01:40:31 +00:00
if (!callback && 'function' === typeof where) {
callback = where;
where = undefined;
2012-12-14 14:01:44 +00:00
}
this.query('DELETE FROM '
+ this.tableEscaped(model) + ' ' + buildWhere(this, model, where || {}), function (err, data) {
callback && callback(err, data);
}.bind(this));
2012-12-14 14:01:44 +00:00
};
2013-10-03 21:50:38 +00:00
/**
* Perform autoupdate for the given models
* @param {String[]} [models] A model name or an array of model names. If not present, apply to all models
* @param {Function} [cb] The callback function
*/
MySQL.prototype.autoupdate = function (models, cb) {
2012-12-14 14:01:44 +00:00
var self = this;
var wait = 0;
if ((!cb) && ('function' === typeof models)) {
cb = models;
models = undefined;
}
// First argument is a model name
if ('string' === typeof models) {
models = [models];
}
models = models || Object.keys(this._models);
models.forEach(function (model) {
if (model in self._models) {
wait++;
self.query('SHOW FIELDS FROM ' + self.tableEscaped(model), function (err, fields) {
self.query('SHOW INDEXES FROM ' + self.tableEscaped(model), function (err, indexes) {
if (!err && fields.length) {
self.alterTable(model, fields, indexes, done);
} else {
self.createTable(model, done);
}
});
2012-12-14 14:01:44 +00:00
});
}
2012-12-14 14:01:44 +00:00
});
function done(err) {
if (err) {
console.error(err);
2012-12-14 14:01:44 +00:00
}
if (--wait === 0 && cb) {
cb();
}
}
};
2013-10-08 20:44:58 +00:00
/*!
* Create table
* @param model
* @param cb
*/
MySQL.prototype.createTable = function (model, cb) {
var metadata = this._models[model].settings[this.name];
var engine = metadata && metadata.engine;
var sql = 'CREATE TABLE ' + this.tableEscaped(model) +
' (\n ' + this.propertiesSQL(model) + '\n)';
2013-11-27 01:40:31 +00:00
if (engine) {
2013-10-08 20:44:58 +00:00
sql += 'ENGINE=' + engine + '\n';
}
this.query(sql, cb);
};
2013-10-03 21:50:38 +00:00
/**
* Check if the models exist
* @param {String[]} [models] A model name or an array of model names. If not present, apply to all models
* @param {Function} [cb] The callback function
*/
2012-12-14 14:01:44 +00:00
MySQL.prototype.isActual = function (cb) {
var ok = false;
var self = this;
var wait = 0;
Object.keys(this._models).forEach(function (model) {
wait += 1;
self.query('SHOW FIELDS FROM ' + model, function (err, fields) {
self.query('SHOW INDEXES FROM ' + model, function (err, indexes) {
self.alterTable(model, fields, indexes, done, true);
});
});
});
function done(err, needAlter) {
if (err) {
console.log(err);
}
ok = ok || needAlter;
if (--wait === 0 && cb) {
cb(null, !ok);
}
}
};
MySQL.prototype.alterTable = function (model, actualFields, actualIndexes, done, checkOnly) {
var self = this;
var m = this._models[model];
var propNames = Object.keys(m.properties).filter(function (name) {
return !!m.properties[name];
});
var indexNames = m.settings.indexes ? Object.keys(m.settings.indexes).filter(function (name) {
return !!m.settings.indexes[name];
}) : [];
var sql = [];
var ai = {};
if (actualIndexes) {
actualIndexes.forEach(function (i) {
var name = i.Key_name;
if (!ai[name]) {
ai[name] = {
info: i,
columns: []
};
}
ai[name].columns[i.Seq_in_index - 1] = i.Column_name;
});
}
var aiNames = Object.keys(ai);
// change/add new fields
propNames.forEach(function (propName) {
2013-07-22 00:27:54 +00:00
if (m.properties[propName] && self.id(model, propName)) return;
2012-12-14 14:01:44 +00:00
var found;
2013-04-18 13:57:56 +00:00
if (actualFields) {
2013-04-18 11:19:59 +00:00
actualFields.forEach(function (f) {
if (f.Field === propName) {
found = f;
}
});
}
2012-12-14 14:01:44 +00:00
if (found) {
actualize(propName, found);
} else {
sql.push('ADD COLUMN `' + propName + '` ' + self.propertySettingsSQL(model, propName));
}
});
// drop columns
2013-04-18 11:19:59 +00:00
if (actualFields) {
actualFields.forEach(function (f) {
var notFound = !~propNames.indexOf(f.Field);
2013-07-22 00:27:54 +00:00
if (m.properties[f.Field] && self.id(model, f.Field)) return;
2013-04-18 11:19:59 +00:00
if (notFound || !m.properties[f.Field]) {
sql.push('DROP COLUMN `' + f.Field + '`');
}
});
}
2012-12-14 14:01:44 +00:00
// remove indexes
aiNames.forEach(function (indexName) {
2013-11-27 01:40:31 +00:00
if (indexName === 'PRIMARY' || (m.properties[indexName] && self.id(model, indexName))) return;
2012-12-14 14:01:44 +00:00
if (indexNames.indexOf(indexName) === -1 && !m.properties[indexName] || m.properties[indexName] && !m.properties[indexName].index) {
sql.push('DROP INDEX `' + indexName + '`');
} else {
// first: check single (only type and kind)
if (m.properties[indexName] && !m.properties[indexName].index) {
// TODO
return;
}
// second: check multiple indexes
var orderMatched = true;
if (indexNames.indexOf(indexName) !== -1) {
m.settings.indexes[indexName].columns.split(/,\s*/).forEach(function (columnName, i) {
if (ai[indexName].columns[i] !== columnName) orderMatched = false;
});
}
if (!orderMatched) {
sql.push('DROP INDEX `' + indexName + '`');
delete ai[indexName];
}
}
});
// add single-column indexes
propNames.forEach(function (propName) {
var i = m.properties[propName].index;
if (!i) {
return;
}
var found = ai[propName] && ai[propName].info;
if (!found) {
var type = '';
var kind = '';
if (i.type) {
type = 'USING ' + i.type;
}
if (i.kind) {
// kind = i.kind;
}
if (kind && type) {
sql.push('ADD ' + kind + ' INDEX `' + propName + '` (`' + propName + '`) ' + type);
} else {
sql.push('ADD ' + kind + ' INDEX `' + propName + '` ' + type + ' (`' + propName + '`) ');
}
}
});
// add multi-column indexes
indexNames.forEach(function (indexName) {
var i = m.settings.indexes[indexName];
var found = ai[indexName] && ai[indexName].info;
if (!found) {
var type = '';
var kind = '';
if (i.type) {
type = 'USING ' + i.type;
2012-12-14 14:01:44 +00:00
}
if (i.kind) {
kind = i.kind;
}
if (kind && type) {
sql.push('ADD ' + kind + ' INDEX `' + indexName + '` (' + i.columns + ') ' + type);
} else {
sql.push('ADD ' + kind + ' INDEX ' + type + ' `' + indexName + '` (' + i.columns + ')');
}
}
});
if (sql.length) {
var query = 'ALTER TABLE ' + self.tableEscaped(model) + ' ' + sql.join(',\n');
if (checkOnly) {
done(null, true, {statements: sql, query: query});
} else {
this.query(query, done);
}
} else {
done();
}
function actualize(propName, oldSettings) {
var newSettings = m.properties[propName];
if (newSettings && changed(newSettings, oldSettings)) {
sql.push('CHANGE COLUMN `' + propName + '` `' + propName + '` ' + self.propertySettingsSQL(model, propName));
}
}
function changed(newSettings, oldSettings) {
if (oldSettings.Null === 'YES') { // Used to allow null and does not now.
2013-11-27 01:40:31 +00:00
if (newSettings.allowNull === false) return true;
if (newSettings.null === false) return true;
}
if (oldSettings.Null === 'NO') { // Did not allow null and now does.
2013-11-27 01:40:31 +00:00
if (newSettings.allowNull === true) return true;
if (newSettings.null === true) return true;
if (newSettings.null === undefined && newSettings.allowNull === undefined) return true;
}
2013-11-27 01:40:31 +00:00
if (oldSettings.Type.toUpperCase() !== datatype(newSettings).toUpperCase()) return true;
2012-12-14 14:01:44 +00:00
return false;
}
};
MySQL.prototype.propertiesSQL = function (model) {
var self = this;
2013-10-14 21:32:19 +00:00
var pks = this.idNames(model).map(function (i) {
return self.columnEscaped(model, i);
});
var sql = [];
if (pks.length === 1) {
var idName = this.idName(model);
sql.push(self.columnEscaped(model, idName) + ' INT(11) NOT NULL AUTO_INCREMENT PRIMARY KEY');
}
Object.keys(this._models[model].properties).forEach(function (prop) {
if (self.id(model, prop) && pks.length === 1) {
return;
}
var colName = self.columnEscaped(model, prop);
sql.push(colName + ' ' + self.propertySettingsSQL(model, prop));
});
if (pks.length > 1) {
sql.push('PRIMARY KEY(' + pks.join(',') + ')');
}
/*
2013-11-27 01:40:31 +00:00
var sql = ['`id` INT(11) NOT NULL AUTO_INCREMENT PRIMARY KEY'];
Object.keys(this._models[model].properties).forEach(function (prop) {
if (self.id(model, prop)) return;
sql.push('`' + prop + '` ' + self.propertySettingsSQL(model, prop));
});
*/
2013-10-14 21:32:19 +00:00
// Declared in model index property indexes.
Object.keys(this._models[model].properties).forEach(function (prop) {
var i = self._models[model].properties[prop].index;
if (i) {
sql.push(self.singleIndexSettingsSQL(model, prop));
}
});
// Settings might not have an indexes property.
var dxs = this._models[model].settings.indexes;
2013-11-27 01:40:31 +00:00
if (dxs) {
Object.keys(this._models[model].settings.indexes).forEach(function (prop) {
sql.push(self.indexSettingsSQL(model, prop));
});
}
2012-12-14 14:01:44 +00:00
return sql.join(',\n ');
};
MySQL.prototype.singleIndexSettingsSQL = function (model, prop) {
// Recycled from alterTable single indexes above, more or less.
var i = this._models[model].properties[prop].index;
var type = '';
var kind = '';
if (i.type) {
type = 'USING ' + i.type;
}
if (i.kind) {
kind = i.kind;
}
2013-10-14 21:32:19 +00:00
var columnName = this.columnEscaped(model, prop);
if (kind && type) {
2013-10-14 21:32:19 +00:00
return (kind + ' INDEX ' + columnName + ' (' + columnName + ') ' + type);
} else {
2013-10-14 21:32:19 +00:00
return (kind + ' INDEX ' + columnName + ' ' + type + ' (' + columnName + ') ');
}
2013-02-14 16:52:00 +00:00
};
2012-12-14 14:01:44 +00:00
MySQL.prototype.indexSettingsSQL = function (model, prop) {
// Recycled from alterTable multi-column indexes above, more or less.
var i = this._models[model].settings.indexes[prop];
var type = '';
var kind = '';
if (i.type) {
type = 'USING ' + i.type;
}
if (i.kind) {
kind = i.kind;
}
2013-10-14 21:32:19 +00:00
var columnName = this.columnEscaped(model, prop);
if (kind && type) {
2013-10-14 21:32:19 +00:00
return (kind + ' INDEX ' + columnName + ' (' + i.columns + ') ' + type);
} else {
2013-10-14 21:32:19 +00:00
return (kind + ' INDEX ' + type + ' ' + columnName + ' (' + i.columns + ')');
}
2012-12-14 14:01:44 +00:00
};
MySQL.prototype.propertySettingsSQL = function (model, prop) {
var p = this._models[model].properties[prop];
2013-11-27 01:40:31 +00:00
var line = this.columnDataType(model, prop) + ' ' +
(p.nullable === false || p.allowNull === false || p['null'] === false ? 'NOT NULL' : 'NULL');
return line;
2012-12-14 14:01:44 +00:00
};
2013-10-08 20:44:58 +00:00
MySQL.prototype.columnDataType = function (model, property) {
var columnMetadata = this.columnMetadata(model, property);
var colType = columnMetadata && columnMetadata.dataType;
2013-11-27 01:40:31 +00:00
if (colType) {
2013-10-08 20:44:58 +00:00
colType = colType.toUpperCase();
}
var prop = this._models[model].properties[property];
2013-11-27 01:40:31 +00:00
if (!prop) {
2013-10-08 20:44:58 +00:00
return null;
}
var colLength = columnMetadata && columnMetadata.dataLength || prop.length;
if (colType) {
return colType + (colLength ? '(' + colLength + ')' : '');
}
return datatype(prop);
};
2012-12-14 14:01:44 +00:00
function datatype(p) {
var dt = '';
switch (p.type.name) {
default:
case 'String':
case 'JSON':
dt = columnType(p, 'VARCHAR');
dt = stringOptionsByType(p, dt);
2013-11-27 01:40:31 +00:00
break;
2012-12-14 14:01:44 +00:00
case 'Text':
dt = columnType(p, 'TEXT');
dt = stringOptionsByType(p, dt);
2013-11-27 01:40:31 +00:00
break;
2012-12-14 14:01:44 +00:00
case 'Number':
dt = columnType(p, 'INT');
dt = numericOptionsByType(p, dt);
2013-11-27 01:40:31 +00:00
break;
2012-12-14 14:01:44 +00:00
case 'Date':
dt = columnType(p, 'DATETIME'); // Currently doesn't need options.
2013-11-27 01:40:31 +00:00
break;
2012-12-14 14:01:44 +00:00
case 'Boolean':
dt = 'TINYINT(1)';
2013-11-27 01:40:31 +00:00
break;
case 'Point':
case 'GeoPoint':
dt = 'POINT';
2013-11-27 01:40:31 +00:00
break;
case 'Enum':
dt = 'ENUM(' + p.type._string + ')';
dt = stringOptions(p, dt); // Enum columns can have charset/collation.
2013-11-27 01:40:31 +00:00
break;
2012-12-14 14:01:44 +00:00
}
return dt;
}
function columnType(p, defaultType) {
var dt = defaultType;
2013-11-27 01:40:31 +00:00
if (p.dataType) {
dt = String(p.dataType);
}
return dt;
}
function stringOptionsByType(p, dt) {
switch (dt.toLowerCase()) {
default:
case 'varchar':
case 'char':
dt += '(' + (p.limit || 255) + ')';
2013-11-27 01:40:31 +00:00
break;
case 'text':
case 'tinytext':
case 'mediumtext':
case 'longtext':
2013-11-27 01:40:31 +00:00
break;
}
dt = stringOptions(p, dt);
return dt;
}
2013-11-27 01:40:31 +00:00
function stringOptions(p, dt) {
if (p.charset) {
dt += " CHARACTER SET " + p.charset;
}
2013-11-27 01:40:31 +00:00
if (p.collation) {
dt += " COLLATE " + p.collation;
}
return dt;
}
function numericOptionsByType(p, dt) {
switch (dt.toLowerCase()) {
default:
case 'tinyint':
case 'smallint':
case 'mediumint':
case 'int':
case 'integer':
case 'bigint':
dt = integerOptions(p, dt);
2013-11-27 01:40:31 +00:00
break;
case 'decimal':
case 'numeric':
dt = fixedPointOptions(p, dt);
2013-11-27 01:40:31 +00:00
break;
case 'float':
case 'double':
dt = floatingPointOptions(p, dt);
2013-11-27 01:40:31 +00:00
break;
}
dt = unsigned(p, dt);
return dt;
}
function floatingPointOptions(p, dt) {
var precision = 16;
var scale = 8;
2013-11-27 01:40:31 +00:00
if (p.precision) {
precision = Number(p.precision);
}
2013-11-27 01:40:31 +00:00
if (p.scale) {
scale = Number(p.scale);
}
if (p.precision && p.scale) {
dt += '(' + precision + ',' + scale + ')';
2013-11-27 01:40:31 +00:00
} else if (p.precision) {
dt += '(' + precision + ')';
2013-11-27 01:40:31 +00:00
}
return dt;
}
/* @TODO: Change fixed point to use an arbitrary precision arithmetic library. */
/* Currently fixed point will lose precision because it's turned to non-fixed in */
/* JS. Also, defaulting column to (9,2) and not allowing non-specified 'DECIMAL' */
/* declaration which would default to DECIMAL(10,0). Instead defaulting to (9,2). */
function fixedPointOptions(p, dt) {
var precision = 9;
var scale = 2;
2013-11-27 01:40:31 +00:00
if (p.precision) {
precision = Number(p.precision);
}
2013-11-27 01:40:31 +00:00
if (p.scale) {
scale = Number(p.scale);
}
dt += '(' + precision + ',' + scale + ')';
return dt;
}
function integerOptions(p, dt) {
var tmp = 0;
if (p.display || p.limit) {
tmp = Number(p.display || p.limit);
}
2013-11-27 01:40:31 +00:00
if (tmp > 0) {
dt += '(' + tmp + ')';
2013-11-27 01:40:31 +00:00
} else if (p.unsigned) {
switch (dt.toLowerCase()) {
default:
case 'int':
dt += '(10)';
2013-11-27 01:40:31 +00:00
break;
case 'mediumint':
dt += '(8)';
2013-11-27 01:40:31 +00:00
break;
case 'smallint':
dt += '(5)';
2013-11-27 01:40:31 +00:00
break;
case 'tinyint':
dt += '(3)';
2013-11-27 01:40:31 +00:00
break;
case 'bigint':
dt += '(20)';
2013-11-27 01:40:31 +00:00
break;
}
} else {
switch (dt.toLowerCase()) {
default:
case 'int':
dt += '(11)';
2013-11-27 01:40:31 +00:00
break;
case 'mediumint':
dt += '(9)';
2013-11-27 01:40:31 +00:00
break;
case 'smallint':
dt += '(6)';
2013-11-27 01:40:31 +00:00
break;
case 'tinyint':
dt += '(4)';
2013-11-27 01:40:31 +00:00
break;
case 'bigint':
dt += '(20)';
2013-11-27 01:40:31 +00:00
break;
}
}
return dt;
}
2013-11-27 01:40:31 +00:00
function unsigned(p, dt) {
if (p.unsigned) {
dt += ' UNSIGNED';
}
return dt;
}
2013-07-21 06:38:40 +00:00
2013-11-27 01:40:31 +00:00
/**
* Disconnect from MySQL
2013-11-27 01:40:31 +00:00
*/
MySQL.prototype.disconnect = function () {
if(this.debug) {
console.log('disconnect');
}
if(this.client) {
this.client.end();
}
};
2013-07-21 17:36:26 +00:00
require('./discovery')(MySQL);