mirror of
https://github.com/knex/knex.git
synced 2025-07-10 10:31:43 +00:00
105 lines
3.0 KiB
JavaScript
105 lines
3.0 KiB
JavaScript
![]() |
|
||
|
// MSSQL Query Compiler
|
||
|
// ------
|
||
|
var inherits = require('inherits')
|
||
|
var QueryCompiler = require('../../../query/compiler')
|
||
|
var assign = require('lodash/object/assign');
|
||
|
|
||
|
function QueryCompiler_MSSQL(client, builder) {
|
||
|
QueryCompiler.call(this, client, builder)
|
||
|
}
|
||
|
inherits(QueryCompiler_MSSQL, QueryCompiler)
|
||
|
|
||
|
assign(QueryCompiler_MSSQL.prototype, {
|
||
|
|
||
|
_emptyInsertValue: '() values ()',
|
||
|
|
||
|
// Update method, including joins, wheres, order & limits.
|
||
|
update: function() {
|
||
|
var join = this.join();
|
||
|
var updates = this._prepUpdate(this.single.update);
|
||
|
var where = this.where();
|
||
|
var order = this.order();
|
||
|
var limit = this.limit_();
|
||
|
return 'update ' + (limit ? limit + ' ' : '') + this.tableName +
|
||
|
(join ? ' ' + join : '') +
|
||
|
' set ' + updates.join(', ') +
|
||
|
(where ? ' ' + where : '') +
|
||
|
(order ? ' ' + order : '');
|
||
|
},
|
||
|
|
||
|
// Compiles the columns in the query, specifying if an item was distinct.
|
||
|
columns: function() {
|
||
|
var distinct = false;
|
||
|
if (this.onlyUnions()) return ''
|
||
|
var columns = this.grouped.columns || []
|
||
|
var i = -1, sql = [];
|
||
|
if (columns) {
|
||
|
while (++i < columns.length) {
|
||
|
var stmt = columns[i];
|
||
|
if (stmt.distinct) distinct = true
|
||
|
if (stmt.type === 'aggregate') {
|
||
|
sql.push(this.aggregate(stmt))
|
||
|
}
|
||
|
else if (stmt.value && stmt.value.length > 0) {
|
||
|
sql.push(this.formatter.columnize(stmt.value))
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
if (sql.length === 0) sql = ['*'];
|
||
|
var limit = this.limit_();
|
||
|
return 'select ' + (distinct ? 'distinct ' : '') +
|
||
|
(limit ? limit + ' ' : '') +
|
||
|
sql.join(', ') + (this.tableName ? ' from ' + this.tableName : '');
|
||
|
},
|
||
|
|
||
|
// Compiles a `truncate` query.
|
||
|
truncate: function() {
|
||
|
return 'truncate table ' + this.tableName;
|
||
|
},
|
||
|
|
||
|
forUpdate: function() {
|
||
|
return 'with (READCOMMITTEDLOCK)';
|
||
|
},
|
||
|
|
||
|
forShare: function() {
|
||
|
return 'with (NOLOCK)';
|
||
|
},
|
||
|
|
||
|
// Compiles a `columnInfo` query.
|
||
|
columnInfo: function() {
|
||
|
var column = this.single.columnInfo;
|
||
|
return {
|
||
|
sql: 'select * from information_schema.columns where table_name = ? and table_schema = ?',
|
||
|
bindings: [this.single.table, this.client.database()],
|
||
|
output: function(resp) {
|
||
|
var out = resp.reduce(function(columns, val) {
|
||
|
columns[val.COLUMN_NAME] = {
|
||
|
defaultValue: val.COLUMN_DEFAULT,
|
||
|
type: val.DATA_TYPE,
|
||
|
maxLength: val.CHARACTER_MAXIMUM_LENGTH,
|
||
|
nullable: (val.IS_NULLABLE === 'YES')
|
||
|
};
|
||
|
return columns
|
||
|
}, {})
|
||
|
return column && out[column] || out;
|
||
|
}
|
||
|
};
|
||
|
},
|
||
|
|
||
|
limit_: function() {
|
||
|
var noLimit = !this.single.limit && this.single.limit !== 0;
|
||
|
if (noLimit) return '';
|
||
|
return 'top (' + this.formatter.parameter(this.single.limit) + ')';
|
||
|
},
|
||
|
|
||
|
limit: function() {
|
||
|
return '';
|
||
|
}
|
||
|
|
||
|
})
|
||
|
|
||
|
// Set the QueryBuilder & QueryCompiler on the client object,
|
||
|
// incase anyone wants to modify things to suit their own purposes.
|
||
|
module.exports = QueryCompiler_MSSQL;
|