Initial commit
This commit is contained in:
378
node_modules/knex/lib/dialects/mssql/schema/mssql-tablecompiler.js
generated
vendored
Executable file
378
node_modules/knex/lib/dialects/mssql/schema/mssql-tablecompiler.js
generated
vendored
Executable file
@@ -0,0 +1,378 @@
|
||||
/* eslint max-len:0 */
|
||||
|
||||
// MSSQL Table Builder & Compiler
|
||||
// -------
|
||||
const TableCompiler = require('../../../schema/tablecompiler');
|
||||
const helpers = require('../../../util/helpers');
|
||||
const { isObject } = require('../../../util/is');
|
||||
|
||||
// Table Compiler
|
||||
// ------
|
||||
|
||||
class TableCompiler_MSSQL extends TableCompiler {
|
||||
constructor(client, tableBuilder) {
|
||||
super(client, tableBuilder);
|
||||
}
|
||||
|
||||
createQuery(columns, ifNot, like) {
|
||||
let createStatement = ifNot
|
||||
? `if object_id('${this.tableName()}', 'U') is null `
|
||||
: '';
|
||||
|
||||
if (like) {
|
||||
// This query copy only columns and not all indexes and keys like other databases.
|
||||
createStatement += `SELECT * INTO ${this.tableName()} FROM ${this.tableNameLike()} WHERE 0=1`;
|
||||
} else {
|
||||
createStatement +=
|
||||
'CREATE TABLE ' +
|
||||
this.tableName() +
|
||||
(this._formatting ? ' (\n ' : ' (') +
|
||||
columns.sql.join(this._formatting ? ',\n ' : ', ') +
|
||||
this._addChecks() +
|
||||
')';
|
||||
}
|
||||
|
||||
this.pushQuery(createStatement);
|
||||
|
||||
if (this.single.comment) {
|
||||
this.comment(this.single.comment);
|
||||
}
|
||||
if (like) {
|
||||
this.addColumns(columns, this.addColumnsPrefix);
|
||||
}
|
||||
}
|
||||
|
||||
comment(/** @type {string} */ comment) {
|
||||
if (!comment) {
|
||||
return;
|
||||
}
|
||||
|
||||
// XXX: This is a byte limit, not character, so we cannot definitively say they'll exceed the limit without server collation info.
|
||||
// When I checked in SQL Server 2019, the ctext column in sys.syscomments is defined as a varbinary(8000), so it doesn't even have its own defined collation.
|
||||
if (comment.length > 7500 / 2) {
|
||||
this.client.logger.warn(
|
||||
'Your comment might be longer than the max comment length for MSSQL of 7,500 bytes.'
|
||||
);
|
||||
}
|
||||
|
||||
// See: https://docs.microsoft.com/en-us/sql/relational-databases/system-stored-procedures/sp-addextendedproperty-transact-sql?view=sql-server-ver15#f-adding-an-extended-property-to-a-table
|
||||
const value = this.formatter.escapingStringDelimiters(comment);
|
||||
const level0name = this.formatter.escapingStringDelimiters(
|
||||
this.schemaNameRaw || 'dbo'
|
||||
);
|
||||
const level1name = this.formatter.escapingStringDelimiters(
|
||||
this.tableNameRaw
|
||||
);
|
||||
const args = `N'MS_Description', N'${value}', N'Schema', N'${level0name}', N'Table', N'${level1name}'`;
|
||||
const isAlreadyDefined = `EXISTS(SELECT * FROM sys.fn_listextendedproperty(N'MS_Description', N'Schema', N'${level0name}', N'Table', N'${level1name}', NULL, NULL))`;
|
||||
this.pushQuery(
|
||||
`IF ${isAlreadyDefined}\n EXEC sys.sp_updateextendedproperty ${args}\nELSE\n EXEC sys.sp_addextendedproperty ${args}`
|
||||
);
|
||||
}
|
||||
|
||||
// Compiles column add. Multiple columns need only one ADD clause (not one ADD per column) so core addColumns doesn't work. #1348
|
||||
addColumns(columns, prefix) {
|
||||
prefix = prefix || this.addColumnsPrefix;
|
||||
|
||||
if (columns.sql.length > 0) {
|
||||
this.pushQuery({
|
||||
sql:
|
||||
(this.lowerCase ? 'alter table ' : 'ALTER TABLE ') +
|
||||
this.tableName() +
|
||||
' ' +
|
||||
prefix +
|
||||
columns.sql.join(', '),
|
||||
bindings: columns.bindings,
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
alterColumns(columns, colBuilder) {
|
||||
for (let i = 0, l = colBuilder.length; i < l; i++) {
|
||||
const builder = colBuilder[i];
|
||||
if (builder.modified.defaultTo) {
|
||||
const schema = this.schemaNameRaw || 'dbo';
|
||||
const baseQuery = `
|
||||
DECLARE @constraint varchar(100) = (SELECT default_constraints.name
|
||||
FROM sys.all_columns
|
||||
INNER JOIN sys.tables
|
||||
ON all_columns.object_id = tables.object_id
|
||||
INNER JOIN sys.schemas
|
||||
ON tables.schema_id = schemas.schema_id
|
||||
INNER JOIN sys.default_constraints
|
||||
ON all_columns.default_object_id = default_constraints.object_id
|
||||
WHERE schemas.name = '${schema}'
|
||||
AND tables.name = '${
|
||||
this.tableNameRaw
|
||||
}'
|
||||
AND all_columns.name = '${builder.getColumnName()}')
|
||||
|
||||
IF @constraint IS NOT NULL EXEC('ALTER TABLE ${
|
||||
this.tableNameRaw
|
||||
} DROP CONSTRAINT ' + @constraint)`;
|
||||
this.pushQuery(baseQuery);
|
||||
}
|
||||
}
|
||||
// in SQL server only one column can be altered at a time
|
||||
columns.sql.forEach((sql) => {
|
||||
this.pushQuery({
|
||||
sql:
|
||||
(this.lowerCase ? 'alter table ' : 'ALTER TABLE ') +
|
||||
this.tableName() +
|
||||
' ' +
|
||||
(this.lowerCase
|
||||
? this.alterColumnPrefix.toLowerCase()
|
||||
: this.alterColumnPrefix) +
|
||||
sql,
|
||||
bindings: columns.bindings,
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
// Compiles column drop. Multiple columns need only one DROP clause (not one DROP per column) so core dropColumn doesn't work. #1348
|
||||
dropColumn() {
|
||||
const _this2 = this;
|
||||
const columns = helpers.normalizeArr.apply(null, arguments);
|
||||
const columnsArray = Array.isArray(columns) ? columns : [columns];
|
||||
const drops = columnsArray.map((column) => _this2.formatter.wrap(column));
|
||||
const schema = this.schemaNameRaw || 'dbo';
|
||||
|
||||
for (const column of columns) {
|
||||
const baseQuery = `
|
||||
DECLARE @constraint varchar(100) = (SELECT default_constraints.name
|
||||
FROM sys.all_columns
|
||||
INNER JOIN sys.tables
|
||||
ON all_columns.object_id = tables.object_id
|
||||
INNER JOIN sys.schemas
|
||||
ON tables.schema_id = schemas.schema_id
|
||||
INNER JOIN sys.default_constraints
|
||||
ON all_columns.default_object_id = default_constraints.object_id
|
||||
WHERE schemas.name = '${schema}'
|
||||
AND tables.name = '${this.tableNameRaw}'
|
||||
AND all_columns.name = '${column}')
|
||||
|
||||
IF @constraint IS NOT NULL EXEC('ALTER TABLE ${this.tableNameRaw} DROP CONSTRAINT ' + @constraint)`;
|
||||
this.pushQuery(baseQuery);
|
||||
}
|
||||
this.pushQuery(
|
||||
(this.lowerCase ? 'alter table ' : 'ALTER TABLE ') +
|
||||
this.tableName() +
|
||||
' ' +
|
||||
this.dropColumnPrefix +
|
||||
drops.join(', ')
|
||||
);
|
||||
}
|
||||
|
||||
changeType() {}
|
||||
|
||||
// Renames a column on the table.
|
||||
renameColumn(from, to) {
|
||||
this.pushQuery(
|
||||
`exec sp_rename ${this.client.parameter(
|
||||
this.tableName() + '.' + from,
|
||||
this.tableBuilder,
|
||||
this.bindingsHolder
|
||||
)}, ${this.client.parameter(
|
||||
to,
|
||||
this.tableBuilder,
|
||||
this.bindingsHolder
|
||||
)}, 'COLUMN'`
|
||||
);
|
||||
}
|
||||
|
||||
dropFKRefs(runner, refs) {
|
||||
const formatter = this.client.formatter(this.tableBuilder);
|
||||
return Promise.all(
|
||||
refs.map(function (ref) {
|
||||
const constraintName = formatter.wrap(ref.CONSTRAINT_NAME);
|
||||
const tableName = formatter.wrap(ref.TABLE_NAME);
|
||||
return runner.query({
|
||||
sql: `ALTER TABLE ${tableName} DROP CONSTRAINT ${constraintName}`,
|
||||
});
|
||||
})
|
||||
);
|
||||
}
|
||||
|
||||
createFKRefs(runner, refs) {
|
||||
const formatter = this.client.formatter(this.tableBuilder);
|
||||
|
||||
return Promise.all(
|
||||
refs.map(function (ref) {
|
||||
const tableName = formatter.wrap(ref.TABLE_NAME);
|
||||
const keyName = formatter.wrap(ref.CONSTRAINT_NAME);
|
||||
const column = formatter.columnize(ref.COLUMN_NAME);
|
||||
const references = formatter.columnize(ref.REFERENCED_COLUMN_NAME);
|
||||
const inTable = formatter.wrap(ref.REFERENCED_TABLE_NAME);
|
||||
const onUpdate = ` ON UPDATE ${ref.UPDATE_RULE}`;
|
||||
const onDelete = ` ON DELETE ${ref.DELETE_RULE}`;
|
||||
|
||||
return runner.query({
|
||||
sql:
|
||||
`ALTER TABLE ${tableName} ADD CONSTRAINT ${keyName}` +
|
||||
' FOREIGN KEY (' +
|
||||
column +
|
||||
') REFERENCES ' +
|
||||
inTable +
|
||||
' (' +
|
||||
references +
|
||||
')' +
|
||||
onUpdate +
|
||||
onDelete,
|
||||
});
|
||||
})
|
||||
);
|
||||
}
|
||||
|
||||
index(columns, indexName, options) {
|
||||
indexName = indexName
|
||||
? this.formatter.wrap(indexName)
|
||||
: this._indexCommand('index', this.tableNameRaw, columns);
|
||||
|
||||
let predicate;
|
||||
if (isObject(options)) {
|
||||
({ predicate } = options);
|
||||
}
|
||||
const predicateQuery = predicate
|
||||
? ' ' + this.client.queryCompiler(predicate).where()
|
||||
: '';
|
||||
this.pushQuery(
|
||||
`CREATE INDEX ${indexName} ON ${this.tableName()} (${this.formatter.columnize(
|
||||
columns
|
||||
)})${predicateQuery}`
|
||||
);
|
||||
}
|
||||
|
||||
/**
|
||||
* Create a primary key.
|
||||
*
|
||||
* @param {undefined | string | string[]} columns
|
||||
* @param {string | {constraintName: string, deferrable?: 'not deferrable'|'deferred'|'immediate' }} constraintName
|
||||
*/
|
||||
primary(columns, constraintName) {
|
||||
let deferrable;
|
||||
if (isObject(constraintName)) {
|
||||
({ constraintName, deferrable } = constraintName);
|
||||
}
|
||||
if (deferrable && deferrable !== 'not deferrable') {
|
||||
this.client.logger.warn(
|
||||
`mssql: primary key constraint [${constraintName}] will not be deferrable ${deferrable} because mssql does not support deferred constraints.`
|
||||
);
|
||||
}
|
||||
constraintName = constraintName
|
||||
? this.formatter.wrap(constraintName)
|
||||
: this.formatter.wrap(`${this.tableNameRaw}_pkey`);
|
||||
if (!this.forCreate) {
|
||||
this.pushQuery(
|
||||
`ALTER TABLE ${this.tableName()} ADD CONSTRAINT ${constraintName} PRIMARY KEY (${this.formatter.columnize(
|
||||
columns
|
||||
)})`
|
||||
);
|
||||
} else {
|
||||
this.pushQuery(
|
||||
`CONSTRAINT ${constraintName} PRIMARY KEY (${this.formatter.columnize(
|
||||
columns
|
||||
)})`
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Create a unique index.
|
||||
*
|
||||
* @param {string | string[]} columns
|
||||
* @param {string | {indexName: undefined | string, deferrable?: 'not deferrable'|'deferred'|'immediate', useConstraint?: true|false, predicate?: QueryBuilder }} indexName
|
||||
*/
|
||||
unique(columns, indexName) {
|
||||
/** @type {string | undefined} */
|
||||
let deferrable;
|
||||
let useConstraint = false;
|
||||
let predicate;
|
||||
if (isObject(indexName)) {
|
||||
({ indexName, deferrable, useConstraint, predicate } = indexName);
|
||||
}
|
||||
if (deferrable && deferrable !== 'not deferrable') {
|
||||
this.client.logger.warn(
|
||||
`mssql: unique index [${indexName}] will not be deferrable ${deferrable} because mssql does not support deferred constraints.`
|
||||
);
|
||||
}
|
||||
if (useConstraint && predicate) {
|
||||
throw new Error('mssql cannot create constraint with predicate');
|
||||
}
|
||||
indexName = indexName
|
||||
? this.formatter.wrap(indexName)
|
||||
: this._indexCommand('unique', this.tableNameRaw, columns);
|
||||
|
||||
if (!Array.isArray(columns)) {
|
||||
columns = [columns];
|
||||
}
|
||||
|
||||
if (useConstraint) {
|
||||
// mssql supports unique indexes and unique constraints.
|
||||
// unique indexes cannot be used with foreign key relationships hence unique constraints are used instead.
|
||||
this.pushQuery(
|
||||
`ALTER TABLE ${this.tableName()} ADD CONSTRAINT ${indexName} UNIQUE (${this.formatter.columnize(
|
||||
columns
|
||||
)})`
|
||||
);
|
||||
} else {
|
||||
// default to making unique index that allows null https://stackoverflow.com/a/767702/360060
|
||||
// to be more or less compatible with other DBs (if any of the columns is NULL then "duplicates" are allowed)
|
||||
const predicateQuery = predicate
|
||||
? ' ' + this.client.queryCompiler(predicate).where()
|
||||
: ' WHERE ' +
|
||||
columns
|
||||
.map((column) => this.formatter.columnize(column) + ' IS NOT NULL')
|
||||
.join(' AND ');
|
||||
this.pushQuery(
|
||||
`CREATE UNIQUE INDEX ${indexName} ON ${this.tableName()} (${this.formatter.columnize(
|
||||
columns
|
||||
)})${predicateQuery}`
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
// Compile a drop index command.
|
||||
dropIndex(columns, indexName) {
|
||||
indexName = indexName
|
||||
? this.formatter.wrap(indexName)
|
||||
: this._indexCommand('index', this.tableNameRaw, columns);
|
||||
this.pushQuery(`DROP INDEX ${indexName} ON ${this.tableName()}`);
|
||||
}
|
||||
|
||||
// Compile a drop foreign key command.
|
||||
dropForeign(columns, indexName) {
|
||||
indexName = indexName
|
||||
? this.formatter.wrap(indexName)
|
||||
: this._indexCommand('foreign', this.tableNameRaw, columns);
|
||||
this.pushQuery(
|
||||
`ALTER TABLE ${this.tableName()} DROP CONSTRAINT ${indexName}`
|
||||
);
|
||||
}
|
||||
|
||||
// Compile a drop primary key command.
|
||||
dropPrimary(constraintName) {
|
||||
constraintName = constraintName
|
||||
? this.formatter.wrap(constraintName)
|
||||
: this.formatter.wrap(`${this.tableNameRaw}_pkey`);
|
||||
this.pushQuery(
|
||||
`ALTER TABLE ${this.tableName()} DROP CONSTRAINT ${constraintName}`
|
||||
);
|
||||
}
|
||||
|
||||
// Compile a drop unique key command.
|
||||
dropUnique(column, indexName) {
|
||||
indexName = indexName
|
||||
? this.formatter.wrap(indexName)
|
||||
: this._indexCommand('unique', this.tableNameRaw, column);
|
||||
this.pushQuery(`DROP INDEX ${indexName} ON ${this.tableName()}`);
|
||||
}
|
||||
}
|
||||
|
||||
TableCompiler_MSSQL.prototype.createAlterTableMethods = ['foreign', 'primary'];
|
||||
TableCompiler_MSSQL.prototype.lowerCase = false;
|
||||
|
||||
TableCompiler_MSSQL.prototype.addColumnsPrefix = 'ADD ';
|
||||
TableCompiler_MSSQL.prototype.dropColumnPrefix = 'DROP COLUMN ';
|
||||
TableCompiler_MSSQL.prototype.alterColumnPrefix = 'ALTER COLUMN ';
|
||||
|
||||
module.exports = TableCompiler_MSSQL;
|
||||
Reference in New Issue
Block a user