421 lines
13 KiB
JavaScript
421 lines
13 KiB
JavaScript
// Copyright IBM Corp. 2013,2019. All Rights Reserved.
|
|
// Node module: loopback-connector-mysql
|
|
// This file is licensed under the MIT License.
|
|
// License text available at https://opensource.org/licenses/MIT
|
|
|
|
'use strict';
|
|
const g = require('strong-globalize')();
|
|
|
|
module.exports = mixinDiscovery;
|
|
|
|
/*!
|
|
* @param {MySQL} MySQL connector class
|
|
* @param {Object} mysql mysql driver
|
|
*/
|
|
function mixinDiscovery(MySQL, mysql) {
|
|
const async = require('async');
|
|
|
|
function paginateSQL(sql, orderBy, options) {
|
|
options = options || {};
|
|
let limitClause = '';
|
|
if (options.offset || options.skip || options.limit) {
|
|
// Offset starts from 0
|
|
let offset = Number(options.offset || options.skip || 0);
|
|
if (isNaN(offset)) {
|
|
offset = 0;
|
|
}
|
|
limitClause = ' LIMIT ' + offset;
|
|
if (options.limit) {
|
|
let limit = Number(options.limit);
|
|
if (isNaN(limit)) {
|
|
limit = 0;
|
|
}
|
|
limitClause = limitClause + ',' + limit;
|
|
}
|
|
}
|
|
if (!orderBy) {
|
|
sql += ' ORDER BY ' + orderBy;
|
|
}
|
|
return sql + limitClause;
|
|
}
|
|
|
|
/*!
|
|
* Build sql for listing schemas (databases in MySQL)
|
|
* @params {Object} [options] Options object
|
|
* @returns {String} The SQL statement
|
|
*/
|
|
MySQL.prototype.buildQuerySchemas = function(options) {
|
|
const sql = 'SELECT catalog_name as "catalog",' +
|
|
' schema_name as "schema"' +
|
|
' FROM information_schema.schemata';
|
|
return paginateSQL(sql, 'schema_name', options);
|
|
};
|
|
|
|
/*!
|
|
* Build sql for listing tables
|
|
* @param options {all: for all owners, owner: for a given owner}
|
|
* @returns {string} The sql statement
|
|
*/
|
|
MySQL.prototype.buildQueryTables = function(options) {
|
|
let sqlTables = null;
|
|
const schema = options.owner || options.schema;
|
|
|
|
if (options.all && !schema) {
|
|
sqlTables = paginateSQL('SELECT \'table\' AS "type",' +
|
|
' table_name AS "name", table_schema AS "owner"' +
|
|
' FROM information_schema.tables',
|
|
'table_schema, table_name', options);
|
|
} else if (schema) {
|
|
sqlTables = paginateSQL('SELECT \'table\' AS "type",' +
|
|
' table_name AS "name", table_schema AS "owner"' +
|
|
' FROM information_schema.tables' +
|
|
' WHERE table_schema=' + mysql.escape(schema),
|
|
'table_schema, table_name', options);
|
|
} else {
|
|
sqlTables = paginateSQL('SELECT \'table\' AS "type",' +
|
|
' table_name AS "name", ' +
|
|
' table_schema AS "owner" FROM information_schema.tables' +
|
|
' WHERE table_schema=SUBSTRING_INDEX(USER(),\'@\',1)',
|
|
'table_name', options);
|
|
}
|
|
return sqlTables;
|
|
};
|
|
|
|
/*!
|
|
* Build sql for listing views
|
|
* @param options {all: for all owners, owner: for a given owner}
|
|
* @returns {string} The sql statement
|
|
*/
|
|
MySQL.prototype.buildQueryViews = function(options) {
|
|
let sqlViews = null;
|
|
if (options.views) {
|
|
const schema = options.owner || options.schema;
|
|
|
|
if (options.all && !schema) {
|
|
sqlViews = paginateSQL('SELECT \'view\' AS "type",' +
|
|
' table_name AS "name",' +
|
|
' table_schema AS "owner"' +
|
|
' FROM information_schema.views',
|
|
'table_schema, table_name', options);
|
|
} else if (schema) {
|
|
sqlViews = paginateSQL('SELECT \'view\' AS "type",' +
|
|
' table_name AS "name",' +
|
|
' table_schema AS "owner"' +
|
|
' FROM information_schema.views' +
|
|
' WHERE table_schema=' + mysql.escape(schema),
|
|
'table_schema, table_name', options);
|
|
} else {
|
|
sqlViews = paginateSQL('SELECT \'view\' AS "type",' +
|
|
' table_name AS "name",' +
|
|
' table_schema AS "owner"' +
|
|
' FROM information_schema.views',
|
|
'table_name', options);
|
|
}
|
|
}
|
|
return sqlViews;
|
|
};
|
|
|
|
/**
|
|
* Discover model definitions
|
|
*
|
|
* @param {Object} options Options for discovery
|
|
* @param {Function} [cb] The callback function
|
|
*/
|
|
|
|
/*!
|
|
* Normalize the arguments
|
|
* @param table string, required
|
|
* @param options object, optional
|
|
* @param cb function, optional
|
|
*/
|
|
MySQL.prototype.getArgs = function(table, options, cb) {
|
|
if ('string' !== typeof table || !table) {
|
|
throw new Error(g.f('{{table}} is a required string argument: %s', table));
|
|
}
|
|
options = options || {};
|
|
if (!cb && 'function' === typeof options) {
|
|
cb = options;
|
|
options = {};
|
|
}
|
|
if (typeof options !== 'object') {
|
|
throw new Error(g.f('{{options}} must be an {{object}}: %s', options));
|
|
}
|
|
return {
|
|
schema: options.owner || options.schema,
|
|
table: table,
|
|
options: options,
|
|
cb: cb,
|
|
};
|
|
};
|
|
|
|
/*!
|
|
* Build the sql statement to query columns for a given table
|
|
* @param schema
|
|
* @param table
|
|
* @returns {String} The sql statement
|
|
*/
|
|
MySQL.prototype.buildQueryColumns = function(schema, table, options = {}) {
|
|
let sql = null;
|
|
if (schema) {
|
|
sql = paginateSQL('SELECT table_schema AS "owner",' +
|
|
' table_name AS "tableName",' +
|
|
' column_name AS "columnName",' +
|
|
' data_type AS "dataType",' +
|
|
' character_maximum_length AS "dataLength",' +
|
|
' numeric_precision AS "dataPrecision",' +
|
|
' numeric_scale AS "dataScale",' +
|
|
' column_type AS "columnType",' +
|
|
' is_nullable = \'YES\' AS "nullable",' +
|
|
' CASE WHEN extra LIKE \'%auto_increment%\' THEN 1 ELSE 0 END AS "generated"' +
|
|
' FROM information_schema.columns' +
|
|
' WHERE table_schema=' + mysql.escape(schema) +
|
|
(table ? ' AND table_name=' + mysql.escape(table) : ''),
|
|
'table_name, ordinal_position', {});
|
|
} else {
|
|
sql = paginateSQL('SELECT table_schema AS "owner",' +
|
|
' table_name AS "tableName",' +
|
|
' column_name AS "columnName",' +
|
|
' data_type AS "dataType",' +
|
|
' character_maximum_length AS "dataLength",' +
|
|
' numeric_precision AS "dataPrecision",' +
|
|
' numeric_scale AS "dataScale",' +
|
|
' column_type AS "columnType",' +
|
|
' is_nullable = \'YES\' AS "nullable",' +
|
|
' CASE WHEN extra LIKE \'%auto_increment%\' THEN 1 ELSE 0 END AS "generated"' +
|
|
' FROM information_schema.columns' +
|
|
(table ? ' WHERE table_name=' + mysql.escape(table) : ''),
|
|
'table_name, ordinal_position', {});
|
|
}
|
|
if (options.orderBy) {
|
|
sql += ' ORDER BY ' + options.orderBy;
|
|
}
|
|
return sql;
|
|
};
|
|
|
|
/**
|
|
* Discover model properties from a table
|
|
* @param {String} table The table name
|
|
* @param {Object} options The options for discovery
|
|
* @param {Function} [cb] The callback function
|
|
*
|
|
*/
|
|
|
|
/*!
|
|
* Build the sql statement for querying primary keys of a given table
|
|
* @param schema
|
|
* @param table
|
|
* @returns {string}
|
|
*/
|
|
// http://docs.oracle.com/javase/6/docs/api/java/sql/DatabaseMetaData.html
|
|
// #getPrimaryKeys(java.lang.String, java.lang.String, java.lang.String)
|
|
MySQL.prototype.buildQueryPrimaryKeys = function(schema, table) {
|
|
let sql = 'SELECT table_schema AS "owner",' +
|
|
' table_name AS "tableName",' +
|
|
' column_name AS "columnName",' +
|
|
' ordinal_position AS "keySeq",' +
|
|
' constraint_name AS "pkName"' +
|
|
' FROM information_schema.key_column_usage' +
|
|
' WHERE constraint_name=\'PRIMARY\'';
|
|
|
|
if (schema) {
|
|
sql += ' AND table_schema=' + mysql.escape(schema);
|
|
}
|
|
if (table) {
|
|
sql += ' AND table_name=' + mysql.escape(table);
|
|
}
|
|
sql += ' ORDER BY' +
|
|
' table_schema, constraint_name, table_name, ordinal_position';
|
|
return sql;
|
|
};
|
|
|
|
/**
|
|
* Discover primary keys for a given table
|
|
* @param {String} table The table name
|
|
* @param {Object} options The options for discovery
|
|
* @param {Function} [cb] The callback function
|
|
*/
|
|
|
|
/*!
|
|
* Build the sql statement for querying foreign keys of a given table
|
|
* @param schema
|
|
* @param table
|
|
* @returns {string}
|
|
*/
|
|
MySQL.prototype.buildQueryForeignKeys = function(schema, table) {
|
|
let sql =
|
|
'SELECT table_schema AS "fkOwner",' +
|
|
' constraint_name AS "fkName",' +
|
|
' table_name AS "fkTableName",' +
|
|
' column_name AS "fkColumnName",' +
|
|
' ordinal_position AS "keySeq",' +
|
|
' referenced_table_schema AS "pkOwner", \'PRIMARY\' AS "pkName",' +
|
|
' referenced_table_name AS "pkTableName",' +
|
|
' referenced_column_name AS "pkColumnName"' +
|
|
' FROM information_schema.key_column_usage' +
|
|
' WHERE constraint_name!=\'PRIMARY\'' +
|
|
' AND POSITION_IN_UNIQUE_CONSTRAINT IS NOT NULL';
|
|
if (schema) {
|
|
sql += ' AND table_schema=' + mysql.escape(schema);
|
|
}
|
|
if (table) {
|
|
sql += ' AND table_name=' + mysql.escape(table);
|
|
}
|
|
return sql;
|
|
};
|
|
|
|
/**
|
|
* Discover foreign keys for a given table
|
|
* @param {String} table The table name
|
|
* @param {Object} options The options for discovery
|
|
* @param {Function} [cb] The callback function
|
|
*/
|
|
|
|
/*!
|
|
* Retrieves a description of the foreign key columns that reference the
|
|
* given table's primary key columns (the foreign keys exported by a table).
|
|
* They are ordered by fkTableOwner, fkTableName, and keySeq.
|
|
* @param schema
|
|
* @param table
|
|
* @returns {string}
|
|
*/
|
|
MySQL.prototype.buildQueryExportedForeignKeys = function(schema, table) {
|
|
let sql = 'SELECT a.constraint_name AS "fkName",' +
|
|
' a.table_schema AS "fkOwner",' +
|
|
' a.table_name AS "fkTableName",' +
|
|
' a.column_name AS "fkColumnName",' +
|
|
' a.ordinal_position AS "keySeq",' +
|
|
' NULL AS "pkName",' +
|
|
' a.referenced_table_schema AS "pkOwner",' +
|
|
' a.referenced_table_name AS "pkTableName",' +
|
|
' a.referenced_column_name AS "pkColumnName"' +
|
|
' FROM information_schema.key_column_usage a' +
|
|
' WHERE a.position_in_unique_constraint IS NOT NULL';
|
|
if (schema) {
|
|
sql += ' AND a.referenced_table_schema=' + mysql.escape(schema);
|
|
}
|
|
if (table) {
|
|
sql += ' AND a.referenced_table_name=' + mysql.escape(table);
|
|
}
|
|
sql += ' ORDER BY a.table_schema, a.table_name, a.ordinal_position';
|
|
|
|
return sql;
|
|
};
|
|
|
|
/**
|
|
* Discover foreign keys that reference to the primary key of this table
|
|
* @param {String} table The table name
|
|
* @param {Object} options The options for discovery
|
|
* @param {Function} [cb] The callback function
|
|
*/
|
|
|
|
MySQL.prototype.buildPropertyType = function(columnDefinition, options) {
|
|
const mysqlType = columnDefinition.dataType;
|
|
const columnType = columnDefinition.columnType;
|
|
const dataLength = columnDefinition.dataLength;
|
|
|
|
// when options are received here the treatTINYINT1AsTinyInt comes as a string
|
|
if (typeof options.treatCHAR1AsString === 'string') {
|
|
options.treatCHAR1AsString =
|
|
options.treatCHAR1AsString === 'true';
|
|
}
|
|
|
|
// when options are received here the treatBIT1AsBit comes as a string
|
|
if (typeof options.treatBIT1AsBit === 'string') {
|
|
options.treatBIT1AsBit =
|
|
options.treatBIT1AsBit === 'true';
|
|
}
|
|
|
|
// when options are received here the treatTINYINT1AsTinyInt comes as a string
|
|
if (typeof options.treatTINYINT1AsTinyInt === 'string') {
|
|
options.treatTINYINT1AsTinyInt =
|
|
options.treatTINYINT1AsTinyInt === 'true';
|
|
}
|
|
|
|
const type = mysqlType.toUpperCase();
|
|
switch (type) {
|
|
case 'CHAR':
|
|
if (!options.treatCHAR1AsString && columnType === 'char(1)') {
|
|
// Treat char(1) as boolean ('Y', 'N', 'T', 'F', '0', '1')
|
|
return 'Boolean';
|
|
}
|
|
case 'VARCHAR':
|
|
case 'TINYTEXT':
|
|
case 'MEDIUMTEXT':
|
|
case 'LONGTEXT':
|
|
case 'TEXT':
|
|
case 'SET':
|
|
return 'String';
|
|
case 'TINYBLOB':
|
|
case 'MEDIUMBLOB':
|
|
case 'LONGBLOB':
|
|
case 'BLOB':
|
|
case 'BINARY':
|
|
case 'VARBINARY':
|
|
case 'BIT':
|
|
// treat BIT(1) as boolean as it's 1 or 0
|
|
if (!options.treatBIT1AsBit && columnType === 'bit(1)') {
|
|
return 'Boolean';
|
|
}
|
|
return 'Binary';
|
|
case 'TINYINT':
|
|
// treat TINYINT(1) as boolean as it is aliased as BOOL and BOOLEAN in mysql
|
|
if (!options.treatTINYINT1AsTinyInt && columnType === 'tinyint(1)') {
|
|
return 'Boolean';
|
|
}
|
|
case 'SMALLINT':
|
|
case 'INT':
|
|
case 'MEDIUMINT':
|
|
case 'YEAR':
|
|
case 'FLOAT':
|
|
case 'DOUBLE':
|
|
case 'BIGINT':
|
|
case 'INTEGER':
|
|
case 'DECIMAL':
|
|
case 'NUMERIC':
|
|
return 'Number';
|
|
case 'DATE':
|
|
case 'TIMESTAMP':
|
|
case 'DATETIME':
|
|
return 'Date';
|
|
case 'POINT':
|
|
return 'GeoPoint';
|
|
case 'BOOL':
|
|
case 'BOOLEAN':
|
|
return 'Boolean';
|
|
case 'ENUM':
|
|
return columnType;
|
|
default:
|
|
return 'String';
|
|
}
|
|
};
|
|
|
|
MySQL.prototype.getDefaultSchema = function() {
|
|
if (this.dataSource && this.dataSource.settings &&
|
|
this.dataSource.settings.database) {
|
|
return this.dataSource.settings.database;
|
|
}
|
|
return undefined;
|
|
};
|
|
|
|
// Recommended MySQL 5.7 Boolean scheme. See
|
|
// http://dev.mysql.com/doc/refman/5.7/en/numeric-type-overview.html
|
|
// Currently default is the inverse of the recommendation for backward compatibility.
|
|
MySQL.prototype.setDefaultOptions = function(options) {
|
|
const defaultOptions = {
|
|
treatCHAR1AsString: false,
|
|
treatBIT1AsBit: true,
|
|
treatTINYINT1AsTinyInt: true,
|
|
};
|
|
|
|
for (const opt in defaultOptions) {
|
|
if (defaultOptions.hasOwnProperty(opt) && !options.hasOwnProperty(opt)) {
|
|
options[opt] = defaultOptions[opt];
|
|
}
|
|
}
|
|
};
|
|
|
|
MySQL.prototype.setNullableProperty = function(r) {
|
|
r.nullable = r.nullable ? 'Y' : 'N';
|
|
};
|
|
}
|