You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
54 lines
1.7 KiB
JavaScript
54 lines
1.7 KiB
JavaScript
|
|
// MySQL Schema Compiler
|
|
// -------
|
|
var inherits = require('inherits');
|
|
var SchemaCompiler = require('../../../schema/compiler');
|
|
var assign = require('lodash/object/assign');
|
|
|
|
function SchemaCompiler_MSSQL(client, builder) {
|
|
SchemaCompiler.call(this, client, builder)
|
|
}
|
|
inherits(SchemaCompiler_MSSQL, SchemaCompiler)
|
|
|
|
assign(SchemaCompiler_MSSQL.prototype, {
|
|
|
|
dropTablePrefix: 'DROP TABLE ',
|
|
dropTableIfExists: function(tableName) {
|
|
var name = this.formatter.wrap(prefixedTableName(this.schema, tableName));
|
|
this.pushQuery('if object_id(\'' + name +'\', \'U\') is not null DROP TABLE ' + name);
|
|
},
|
|
|
|
// Rename a table on the schema.
|
|
renameTable: function(tableName, to) {
|
|
this.pushQuery('exec sp_rename ' + this.formatter.parameter(tableName) + ', ' + this.formatter.parameter(to));
|
|
},
|
|
|
|
// Check whether a table exists on the query.
|
|
hasTable: function(tableName) {
|
|
this.pushQuery({
|
|
sql: 'select object_id from sys.tables where object_id = object_id(' + this.formatter.parameter(this.formatter.wrap(tableName)) + ')',
|
|
output: function(resp) {
|
|
return resp.length > 0;
|
|
}
|
|
});
|
|
},
|
|
|
|
// Check whether a column exists on the schema.
|
|
hasColumn: function(tableName, column) {
|
|
this.pushQuery({
|
|
sql: 'select object_id from sys.columns where name = ' + this.formatter.parameter(column) +
|
|
' and object_id = object_id(' + this.formatter.parameter(this.formatter.wrap(tableName)) + ')',
|
|
output: function(resp) {
|
|
return resp.length > 0;
|
|
}
|
|
});
|
|
}
|
|
|
|
})
|
|
|
|
function prefixedTableName(prefix, table) {
|
|
return prefix ? `${prefix}.${table}` : table;
|
|
}
|
|
|
|
module.exports = SchemaCompiler_MSSQL;
|