You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
2090 lines
86 KiB
2090 lines
86 KiB
"use strict"; |
|
var __defProp = Object.defineProperty; |
|
var __defProps = Object.defineProperties; |
|
var __getOwnPropDescs = Object.getOwnPropertyDescriptors; |
|
var __getOwnPropSymbols = Object.getOwnPropertySymbols; |
|
var __hasOwnProp = Object.prototype.hasOwnProperty; |
|
var __propIsEnum = Object.prototype.propertyIsEnumerable; |
|
var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value; |
|
var __spreadValues = (a, b) => { |
|
for (var prop in b || (b = {})) |
|
if (__hasOwnProp.call(b, prop)) |
|
__defNormalProp(a, prop, b[prop]); |
|
if (__getOwnPropSymbols) |
|
for (var prop of __getOwnPropSymbols(b)) { |
|
if (__propIsEnum.call(b, prop)) |
|
__defNormalProp(a, prop, b[prop]); |
|
} |
|
return a; |
|
}; |
|
var __spreadProps = (a, b) => __defProps(a, __getOwnPropDescs(b)); |
|
const util = require("util"); |
|
const _ = require("lodash"); |
|
const uuidv4 = require("uuid").v4; |
|
const Utils = require("../../utils"); |
|
const deprecations = require("../../utils/deprecations"); |
|
const SqlString = require("../../sql-string"); |
|
const DataTypes = require("../../data-types"); |
|
const Model = require("../../model"); |
|
const Association = require("../../associations/base"); |
|
const BelongsTo = require("../../associations/belongs-to"); |
|
const BelongsToMany = require("../../associations/belongs-to-many"); |
|
const HasMany = require("../../associations/has-many"); |
|
const Op = require("../../operators"); |
|
const sequelizeError = require("../../errors"); |
|
const IndexHints = require("../../index-hints"); |
|
class QueryGenerator { |
|
constructor(options) { |
|
if (!options.sequelize) |
|
throw new Error("QueryGenerator initialized without options.sequelize"); |
|
if (!options._dialect) |
|
throw new Error("QueryGenerator initialized without options._dialect"); |
|
this.sequelize = options.sequelize; |
|
this.options = options.sequelize.options; |
|
this.dialect = options._dialect.name; |
|
this._dialect = options._dialect; |
|
this._initQuoteIdentifier(); |
|
} |
|
extractTableDetails(tableName, options) { |
|
options = options || {}; |
|
tableName = tableName || {}; |
|
return { |
|
schema: tableName.schema || options.schema || this.options.schema || "public", |
|
tableName: _.isPlainObject(tableName) ? tableName.tableName : tableName, |
|
delimiter: tableName.delimiter || options.delimiter || "." |
|
}; |
|
} |
|
addSchema(param) { |
|
if (!param._schema) |
|
return param.tableName || param; |
|
const self = this; |
|
return { |
|
tableName: param.tableName || param, |
|
table: param.tableName || param, |
|
name: param.name || param, |
|
schema: param._schema, |
|
delimiter: param._schemaDelimiter || ".", |
|
toString() { |
|
return self.quoteTable(this); |
|
} |
|
}; |
|
} |
|
dropSchema(tableName, options) { |
|
return this.dropTableQuery(tableName, options); |
|
} |
|
describeTableQuery(tableName, schema, schemaDelimiter) { |
|
const table = this.quoteTable(this.addSchema({ |
|
tableName, |
|
_schema: schema, |
|
_schemaDelimiter: schemaDelimiter |
|
})); |
|
return `DESCRIBE ${table};`; |
|
} |
|
dropTableQuery(tableName) { |
|
return `DROP TABLE IF EXISTS ${this.quoteTable(tableName)};`; |
|
} |
|
renameTableQuery(before, after) { |
|
return `ALTER TABLE ${this.quoteTable(before)} RENAME TO ${this.quoteTable(after)};`; |
|
} |
|
populateInsertQueryReturnIntoBinds() { |
|
} |
|
insertQuery(table, valueHash, modelAttributes, options) { |
|
options = options || {}; |
|
_.defaults(options, this.options); |
|
const modelAttributeMap = {}; |
|
const bind = options.bind || []; |
|
const fields = []; |
|
const returningModelAttributes = []; |
|
const returnTypes = []; |
|
const values = []; |
|
const quotedTable = this.quoteTable(table); |
|
const bindParam = options.bindParam === void 0 ? this.bindParam(bind) : options.bindParam; |
|
const returnAttributes = []; |
|
let query; |
|
let valueQuery = ""; |
|
let emptyQuery = ""; |
|
let outputFragment = ""; |
|
let returningFragment = ""; |
|
let identityWrapperRequired = false; |
|
let tmpTable = ""; |
|
if (modelAttributes) { |
|
_.each(modelAttributes, (attribute, key) => { |
|
modelAttributeMap[key] = attribute; |
|
if (attribute.field) { |
|
modelAttributeMap[attribute.field] = attribute; |
|
} |
|
}); |
|
} |
|
if (this._dialect.supports["DEFAULT VALUES"]) { |
|
emptyQuery += " DEFAULT VALUES"; |
|
} else if (this._dialect.supports["VALUES ()"]) { |
|
emptyQuery += " VALUES ()"; |
|
} |
|
if ((this._dialect.supports.returnValues || this._dialect.supports.returnIntoValues) && options.returning) { |
|
const returnValues = this.generateReturnValues(modelAttributes, options); |
|
returningModelAttributes.push(...returnValues.returnFields); |
|
if (this._dialect.supports.returnIntoValues) { |
|
returnTypes.push(...returnValues.returnTypes); |
|
} |
|
returningFragment = returnValues.returningFragment; |
|
tmpTable = returnValues.tmpTable || ""; |
|
outputFragment = returnValues.outputFragment || ""; |
|
} |
|
if (_.get(this, ["sequelize", "options", "dialectOptions", "prependSearchPath"]) || options.searchPath) { |
|
options.bindParam = false; |
|
} |
|
if (this._dialect.supports.EXCEPTION && options.exception) { |
|
options.bindParam = false; |
|
} |
|
valueHash = Utils.removeNullValuesFromHash(valueHash, this.options.omitNull); |
|
for (const key in valueHash) { |
|
if (Object.prototype.hasOwnProperty.call(valueHash, key)) { |
|
const value = valueHash[key]; |
|
fields.push(this.quoteIdentifier(key)); |
|
if (modelAttributeMap && modelAttributeMap[key] && modelAttributeMap[key].autoIncrement === true && value == null) { |
|
if (!this._dialect.supports.autoIncrement.defaultValue) { |
|
fields.splice(-1, 1); |
|
} else if (this._dialect.supports.DEFAULT) { |
|
values.push("DEFAULT"); |
|
} else { |
|
values.push(this.escape(null)); |
|
} |
|
} else { |
|
if (modelAttributeMap && modelAttributeMap[key] && modelAttributeMap[key].autoIncrement === true) { |
|
identityWrapperRequired = true; |
|
} |
|
if (value instanceof Utils.SequelizeMethod || options.bindParam === false) { |
|
values.push(this.escape(value, modelAttributeMap && modelAttributeMap[key] || void 0, { context: "INSERT" })); |
|
} else { |
|
values.push(this.format(value, modelAttributeMap && modelAttributeMap[key] || void 0, { context: "INSERT" }, bindParam)); |
|
} |
|
} |
|
} |
|
} |
|
let onDuplicateKeyUpdate = ""; |
|
if (this._dialect.supports.inserts.updateOnDuplicate && options.updateOnDuplicate) { |
|
if (this._dialect.supports.inserts.updateOnDuplicate == " ON CONFLICT DO UPDATE SET") { |
|
const conflictKeys = options.upsertKeys.map((attr) => this.quoteIdentifier(attr)); |
|
const updateKeys = options.updateOnDuplicate.map((attr) => `${this.quoteIdentifier(attr)}=EXCLUDED.${this.quoteIdentifier(attr)}`); |
|
onDuplicateKeyUpdate = ` ON CONFLICT (${conflictKeys.join(",")})`; |
|
onDuplicateKeyUpdate += _.isEmpty(updateKeys) ? " DO NOTHING " : ` DO UPDATE SET ${updateKeys.join(",")}`; |
|
} else { |
|
const valueKeys = options.updateOnDuplicate.map((attr) => `${this.quoteIdentifier(attr)}=VALUES(${this.quoteIdentifier(attr)})`); |
|
if (_.isEmpty(valueKeys) && options.upsertKeys) { |
|
valueKeys.push(...options.upsertKeys.map((attr) => `${this.quoteIdentifier(attr)}=${this.quoteIdentifier(attr)}`)); |
|
} |
|
if (_.isEmpty(valueKeys)) { |
|
throw new Error("No update values found for ON DUPLICATE KEY UPDATE clause, and no identifier fields could be found to use instead."); |
|
} |
|
onDuplicateKeyUpdate += `${this._dialect.supports.inserts.updateOnDuplicate} ${valueKeys.join(",")}`; |
|
} |
|
} |
|
const replacements = { |
|
ignoreDuplicates: options.ignoreDuplicates ? this._dialect.supports.inserts.ignoreDuplicates : "", |
|
onConflictDoNothing: options.ignoreDuplicates ? this._dialect.supports.inserts.onConflictDoNothing : "", |
|
attributes: fields.join(","), |
|
output: outputFragment, |
|
values: values.join(","), |
|
tmpTable |
|
}; |
|
valueQuery = `${tmpTable}INSERT${replacements.ignoreDuplicates} INTO ${quotedTable} (${replacements.attributes})${replacements.output} VALUES (${replacements.values})${onDuplicateKeyUpdate}${replacements.onConflictDoNothing}${valueQuery}`; |
|
emptyQuery = `${tmpTable}INSERT${replacements.ignoreDuplicates} INTO ${quotedTable}${replacements.output}${onDuplicateKeyUpdate}${replacements.onConflictDoNothing}${emptyQuery}`; |
|
if (this._dialect.supports.EXCEPTION && options.exception) { |
|
const dropFunction = "DROP FUNCTION IF EXISTS pg_temp.testfunc()"; |
|
if (returningModelAttributes.length === 0) { |
|
returningModelAttributes.push("*"); |
|
} |
|
const delimiter = `$func_${uuidv4().replace(/-/g, "")}$`; |
|
const selectQuery = `SELECT (testfunc.response).${returningModelAttributes.join(", (testfunc.response).")}, testfunc.sequelize_caught_exception FROM pg_temp.testfunc();`; |
|
options.exception = "WHEN unique_violation THEN GET STACKED DIAGNOSTICS sequelize_caught_exception = PG_EXCEPTION_DETAIL;"; |
|
valueQuery = `CREATE OR REPLACE FUNCTION pg_temp.testfunc(OUT response ${quotedTable}, OUT sequelize_caught_exception text) RETURNS RECORD AS ${delimiter} BEGIN ${valueQuery} RETURNING * INTO response; EXCEPTION ${options.exception} END ${delimiter} LANGUAGE plpgsql; ${selectQuery} ${dropFunction}`; |
|
} else { |
|
valueQuery += returningFragment; |
|
emptyQuery += returningFragment; |
|
} |
|
if (this._dialect.supports.returnIntoValues && options.returning) { |
|
this.populateInsertQueryReturnIntoBinds(returningModelAttributes, returnTypes, bind.length, returnAttributes, options); |
|
} |
|
query = `${replacements.attributes.length ? valueQuery : emptyQuery}${returnAttributes.join(",")};`; |
|
if (this._dialect.supports.finalTable) { |
|
query = `SELECT * FROM FINAL TABLE(${replacements.attributes.length ? valueQuery : emptyQuery});`; |
|
} |
|
if (identityWrapperRequired && this._dialect.supports.autoIncrement.identityInsert) { |
|
query = `SET IDENTITY_INSERT ${quotedTable} ON; ${query} SET IDENTITY_INSERT ${quotedTable} OFF;`; |
|
} |
|
const result = { query }; |
|
if (options.bindParam !== false) { |
|
result.bind = bind; |
|
} |
|
return result; |
|
} |
|
bulkInsertQuery(tableName, fieldValueHashes, options, fieldMappedAttributes) { |
|
options = options || {}; |
|
fieldMappedAttributes = fieldMappedAttributes || {}; |
|
const tuples = []; |
|
const serials = {}; |
|
const allAttributes = []; |
|
let onDuplicateKeyUpdate = ""; |
|
for (const fieldValueHash of fieldValueHashes) { |
|
_.forOwn(fieldValueHash, (value, key) => { |
|
if (!allAttributes.includes(key)) { |
|
allAttributes.push(key); |
|
} |
|
if (fieldMappedAttributes[key] && fieldMappedAttributes[key].autoIncrement === true) { |
|
serials[key] = true; |
|
} |
|
}); |
|
} |
|
for (const fieldValueHash of fieldValueHashes) { |
|
const values = allAttributes.map((key) => { |
|
if (this._dialect.supports.bulkDefault && serials[key] === true) { |
|
return fieldValueHash[key] != null ? fieldValueHash[key] : "DEFAULT"; |
|
} |
|
return this.escape(fieldValueHash[key], fieldMappedAttributes[key], { context: "INSERT" }); |
|
}); |
|
tuples.push(`(${values.join(",")})`); |
|
} |
|
if (this._dialect.supports.inserts.updateOnDuplicate && options.updateOnDuplicate) { |
|
if (this._dialect.supports.inserts.updateOnDuplicate == " ON CONFLICT DO UPDATE SET") { |
|
const conflictKeys = options.upsertKeys.map((attr) => this.quoteIdentifier(attr)); |
|
const updateKeys = options.updateOnDuplicate.map((attr) => `${this.quoteIdentifier(attr)}=EXCLUDED.${this.quoteIdentifier(attr)}`); |
|
onDuplicateKeyUpdate = ` ON CONFLICT (${conflictKeys.join(",")}) DO UPDATE SET ${updateKeys.join(",")}`; |
|
} else { |
|
const valueKeys = options.updateOnDuplicate.map((attr) => `${this.quoteIdentifier(attr)}=VALUES(${this.quoteIdentifier(attr)})`); |
|
onDuplicateKeyUpdate = `${this._dialect.supports.inserts.updateOnDuplicate} ${valueKeys.join(",")}`; |
|
} |
|
} |
|
const ignoreDuplicates = options.ignoreDuplicates ? this._dialect.supports.inserts.ignoreDuplicates : ""; |
|
const attributes = allAttributes.map((attr) => this.quoteIdentifier(attr)).join(","); |
|
const onConflictDoNothing = options.ignoreDuplicates ? this._dialect.supports.inserts.onConflictDoNothing : ""; |
|
let returning = ""; |
|
if (this._dialect.supports.returnValues && options.returning) { |
|
const returnValues = this.generateReturnValues(fieldMappedAttributes, options); |
|
returning += returnValues.returningFragment; |
|
} |
|
return Utils.joinSQLFragments([ |
|
"INSERT", |
|
ignoreDuplicates, |
|
"INTO", |
|
this.quoteTable(tableName), |
|
`(${attributes})`, |
|
"VALUES", |
|
tuples.join(","), |
|
onDuplicateKeyUpdate, |
|
onConflictDoNothing, |
|
returning, |
|
";" |
|
]); |
|
} |
|
updateQuery(tableName, attrValueHash, where, options, attributes) { |
|
options = options || {}; |
|
_.defaults(options, this.options); |
|
attrValueHash = Utils.removeNullValuesFromHash(attrValueHash, options.omitNull, options); |
|
const values = []; |
|
const bind = []; |
|
const modelAttributeMap = {}; |
|
let outputFragment = ""; |
|
let tmpTable = ""; |
|
let suffix = ""; |
|
if (_.get(this, ["sequelize", "options", "dialectOptions", "prependSearchPath"]) || options.searchPath) { |
|
options.bindParam = false; |
|
} |
|
const bindParam = options.bindParam === void 0 ? this.bindParam(bind) : options.bindParam; |
|
if (this._dialect.supports["LIMIT ON UPDATE"] && options.limit) { |
|
if (!["mssql", "db2", "oracle"].includes(this.dialect)) { |
|
suffix = ` LIMIT ${this.escape(options.limit)} `; |
|
} else if (this.dialect === "oracle") { |
|
if (where && (where.length && where.length > 0 || Object.keys(where).length > 0)) { |
|
suffix += " AND "; |
|
} else { |
|
suffix += " WHERE "; |
|
} |
|
suffix += `rownum <= ${this.escape(options.limit)} `; |
|
} |
|
} |
|
if (this._dialect.supports.returnValues && options.returning) { |
|
const returnValues = this.generateReturnValues(attributes, options); |
|
suffix += returnValues.returningFragment; |
|
tmpTable = returnValues.tmpTable || ""; |
|
outputFragment = returnValues.outputFragment || ""; |
|
if (!this._dialect.supports.returnValues.output && options.returning) { |
|
options.mapToModel = true; |
|
} |
|
} |
|
if (attributes) { |
|
_.each(attributes, (attribute, key) => { |
|
modelAttributeMap[key] = attribute; |
|
if (attribute.field) { |
|
modelAttributeMap[attribute.field] = attribute; |
|
} |
|
}); |
|
} |
|
for (const key in attrValueHash) { |
|
if (modelAttributeMap && modelAttributeMap[key] && modelAttributeMap[key].autoIncrement === true && !this._dialect.supports.autoIncrement.update) { |
|
continue; |
|
} |
|
const value = attrValueHash[key]; |
|
if (value instanceof Utils.SequelizeMethod || options.bindParam === false) { |
|
values.push(`${this.quoteIdentifier(key)}=${this.escape(value, modelAttributeMap && modelAttributeMap[key] || void 0, { context: "UPDATE" })}`); |
|
} else { |
|
values.push(`${this.quoteIdentifier(key)}=${this.format(value, modelAttributeMap && modelAttributeMap[key] || void 0, { context: "UPDATE" }, bindParam)}`); |
|
} |
|
} |
|
const whereOptions = __spreadProps(__spreadValues({}, options), { bindParam }); |
|
if (values.length === 0) { |
|
return ""; |
|
} |
|
const query = `${tmpTable}UPDATE ${this.quoteTable(tableName)} SET ${values.join(",")}${outputFragment} ${this.whereQuery(where, whereOptions)}${suffix}`.trim(); |
|
const result = { query }; |
|
if (options.bindParam !== false) { |
|
result.bind = bind; |
|
} |
|
return result; |
|
} |
|
arithmeticQuery(operator, tableName, where, incrementAmountsByField, extraAttributesToBeUpdated, options) { |
|
options = options || {}; |
|
_.defaults(options, { returning: true }); |
|
extraAttributesToBeUpdated = Utils.removeNullValuesFromHash(extraAttributesToBeUpdated, this.options.omitNull); |
|
let outputFragment = ""; |
|
let returningFragment = ""; |
|
if (this._dialect.supports.returnValues && options.returning) { |
|
const returnValues = this.generateReturnValues(null, options); |
|
outputFragment = returnValues.outputFragment; |
|
returningFragment = returnValues.returningFragment; |
|
} |
|
const updateSetSqlFragments = []; |
|
for (const field in incrementAmountsByField) { |
|
const incrementAmount = incrementAmountsByField[field]; |
|
const quotedField = this.quoteIdentifier(field); |
|
const escapedAmount = this.escape(incrementAmount); |
|
updateSetSqlFragments.push(`${quotedField}=${quotedField}${operator} ${escapedAmount}`); |
|
} |
|
for (const field in extraAttributesToBeUpdated) { |
|
const newValue = extraAttributesToBeUpdated[field]; |
|
const quotedField = this.quoteIdentifier(field); |
|
const escapedValue = this.escape(newValue); |
|
updateSetSqlFragments.push(`${quotedField}=${escapedValue}`); |
|
} |
|
return Utils.joinSQLFragments([ |
|
"UPDATE", |
|
this.quoteTable(tableName), |
|
"SET", |
|
updateSetSqlFragments.join(","), |
|
outputFragment, |
|
this.whereQuery(where), |
|
returningFragment |
|
]); |
|
} |
|
addIndexQuery(tableName, attributes, options, rawTablename) { |
|
options = options || {}; |
|
if (!Array.isArray(attributes)) { |
|
options = attributes; |
|
attributes = void 0; |
|
} else { |
|
options.fields = attributes; |
|
} |
|
options.prefix = options.prefix || rawTablename || tableName; |
|
if (options.prefix && typeof options.prefix === "string") { |
|
options.prefix = options.prefix.replace(/\./g, "_"); |
|
options.prefix = options.prefix.replace(/("|')/g, ""); |
|
} |
|
const fieldsSql = options.fields.map((field) => { |
|
if (field instanceof Utils.SequelizeMethod) { |
|
return this.handleSequelizeMethod(field); |
|
} |
|
if (typeof field === "string") { |
|
field = { |
|
name: field |
|
}; |
|
} |
|
let result = ""; |
|
if (field.attribute) { |
|
field.name = field.attribute; |
|
} |
|
if (!field.name) { |
|
throw new Error(`The following index field has no name: ${util.inspect(field)}`); |
|
} |
|
result += this.quoteIdentifier(field.name); |
|
if (this._dialect.supports.index.collate && field.collate) { |
|
result += ` COLLATE ${this.quoteIdentifier(field.collate)}`; |
|
} |
|
if (this._dialect.supports.index.operator) { |
|
const operator = field.operator || options.operator; |
|
if (operator) { |
|
result += ` ${operator}`; |
|
} |
|
} |
|
if (this._dialect.supports.index.length && field.length) { |
|
result += `(${field.length})`; |
|
} |
|
if (field.order) { |
|
result += ` ${field.order}`; |
|
} |
|
return result; |
|
}); |
|
if (!options.name) { |
|
options = Utils.nameIndex(options, options.prefix); |
|
} |
|
options = Model._conformIndex(options); |
|
if (!this._dialect.supports.index.type) { |
|
delete options.type; |
|
} |
|
if (options.where) { |
|
options.where = this.whereQuery(options.where); |
|
} |
|
if (typeof tableName === "string") { |
|
tableName = this.quoteIdentifiers(tableName); |
|
} else { |
|
tableName = this.quoteTable(tableName); |
|
} |
|
const concurrently = this._dialect.supports.index.concurrently && options.concurrently ? "CONCURRENTLY" : void 0; |
|
let ind; |
|
if (this._dialect.supports.indexViaAlter) { |
|
ind = [ |
|
"ALTER TABLE", |
|
tableName, |
|
concurrently, |
|
"ADD" |
|
]; |
|
} else { |
|
ind = ["CREATE"]; |
|
} |
|
ind = ind.concat(options.unique ? "UNIQUE" : "", options.type, "INDEX", !this._dialect.supports.indexViaAlter ? concurrently : void 0, this.quoteIdentifiers(options.name), this._dialect.supports.index.using === 1 && options.using ? `USING ${options.using}` : "", !this._dialect.supports.indexViaAlter ? `ON ${tableName}` : void 0, this._dialect.supports.index.using === 2 && options.using ? `USING ${options.using}` : "", `(${fieldsSql.join(", ")})`, this._dialect.supports.index.parser && options.parser ? `WITH PARSER ${options.parser}` : void 0, this._dialect.supports.index.where && options.where ? options.where : void 0); |
|
return _.compact(ind).join(" "); |
|
} |
|
addConstraintQuery(tableName, options) { |
|
if (typeof tableName === "string") { |
|
tableName = this.quoteIdentifiers(tableName); |
|
} else { |
|
tableName = this.quoteTable(tableName); |
|
} |
|
return Utils.joinSQLFragments([ |
|
"ALTER TABLE", |
|
tableName, |
|
"ADD", |
|
this.getConstraintSnippet(tableName, options || {}), |
|
";" |
|
]); |
|
} |
|
getConstraintSnippet(tableName, options) { |
|
let constraintSnippet, constraintName; |
|
const fieldsSql = options.fields.map((field) => { |
|
if (typeof field === "string") { |
|
return this.quoteIdentifier(field); |
|
} |
|
if (field instanceof Utils.SequelizeMethod) { |
|
return this.handleSequelizeMethod(field); |
|
} |
|
if (field.attribute) { |
|
field.name = field.attribute; |
|
} |
|
if (!field.name) { |
|
throw new Error(`The following index field has no name: ${field}`); |
|
} |
|
return this.quoteIdentifier(field.name); |
|
}); |
|
const fieldsSqlQuotedString = fieldsSql.join(", "); |
|
const fieldsSqlString = fieldsSql.join("_"); |
|
switch (options.type.toUpperCase()) { |
|
case "UNIQUE": |
|
constraintName = this.quoteIdentifier(options.name || `${tableName}_${fieldsSqlString}_uk`); |
|
constraintSnippet = `CONSTRAINT ${constraintName} UNIQUE (${fieldsSqlQuotedString})`; |
|
break; |
|
case "CHECK": |
|
options.where = this.whereItemsQuery(options.where); |
|
constraintName = this.quoteIdentifier(options.name || `${tableName}_${fieldsSqlString}_ck`); |
|
constraintSnippet = `CONSTRAINT ${constraintName} CHECK (${options.where})`; |
|
break; |
|
case "DEFAULT": |
|
if (options.defaultValue === void 0) { |
|
throw new Error("Default value must be specified for DEFAULT CONSTRAINT"); |
|
} |
|
if (this._dialect.name !== "mssql") { |
|
throw new Error("Default constraints are supported only for MSSQL dialect."); |
|
} |
|
constraintName = this.quoteIdentifier(options.name || `${tableName}_${fieldsSqlString}_df`); |
|
constraintSnippet = `CONSTRAINT ${constraintName} DEFAULT (${this.escape(options.defaultValue)}) FOR ${fieldsSql[0]}`; |
|
break; |
|
case "PRIMARY KEY": |
|
constraintName = this.quoteIdentifier(options.name || `${tableName}_${fieldsSqlString}_pk`); |
|
constraintSnippet = `CONSTRAINT ${constraintName} PRIMARY KEY (${fieldsSqlQuotedString})`; |
|
break; |
|
case "FOREIGN KEY": |
|
const references = options.references; |
|
if (!references || !references.table || !(references.field || references.fields)) { |
|
throw new Error("references object with table and field must be specified"); |
|
} |
|
constraintName = this.quoteIdentifier(options.name || `${tableName}_${fieldsSqlString}_${references.table}_fk`); |
|
const quotedReferences = typeof references.field !== "undefined" ? this.quoteIdentifier(references.field) : references.fields.map((f) => this.quoteIdentifier(f)).join(", "); |
|
const referencesSnippet = `${this.quoteTable(references.table)} (${quotedReferences})`; |
|
constraintSnippet = `CONSTRAINT ${constraintName} `; |
|
constraintSnippet += `FOREIGN KEY (${fieldsSqlQuotedString}) REFERENCES ${referencesSnippet}`; |
|
if (options.onUpdate) { |
|
constraintSnippet += ` ON UPDATE ${options.onUpdate.toUpperCase()}`; |
|
} |
|
if (options.onDelete) { |
|
constraintSnippet += ` ON DELETE ${options.onDelete.toUpperCase()}`; |
|
} |
|
break; |
|
default: |
|
throw new Error(`${options.type} is invalid.`); |
|
} |
|
if (options.deferrable && ["UNIQUE", "PRIMARY KEY", "FOREIGN KEY"].includes(options.type.toUpperCase())) { |
|
constraintSnippet += ` ${this.deferConstraintsQuery(options)}`; |
|
} |
|
return constraintSnippet; |
|
} |
|
removeConstraintQuery(tableName, constraintName) { |
|
if (typeof tableName === "string") { |
|
tableName = this.quoteIdentifiers(tableName); |
|
} else { |
|
tableName = this.quoteTable(tableName); |
|
} |
|
return Utils.joinSQLFragments([ |
|
"ALTER TABLE", |
|
tableName, |
|
"DROP CONSTRAINT", |
|
this.quoteIdentifiers(constraintName) |
|
]); |
|
} |
|
quote(collection, parent, connector) { |
|
const validOrderOptions = [ |
|
"ASC", |
|
"DESC", |
|
"ASC NULLS LAST", |
|
"DESC NULLS LAST", |
|
"ASC NULLS FIRST", |
|
"DESC NULLS FIRST", |
|
"NULLS FIRST", |
|
"NULLS LAST" |
|
]; |
|
connector = connector || "."; |
|
if (typeof collection === "string") { |
|
return this.quoteIdentifiers(collection); |
|
} |
|
if (Array.isArray(collection)) { |
|
collection.forEach((item2, index) => { |
|
const previous = collection[index - 1]; |
|
let previousAssociation; |
|
let previousModel; |
|
if (!previous && parent !== void 0) { |
|
previousModel = parent; |
|
} else if (previous && previous instanceof Association) { |
|
previousAssociation = previous; |
|
previousModel = previous.target; |
|
} |
|
if (previousModel && previousModel.prototype instanceof Model) { |
|
let model; |
|
let as; |
|
if (typeof item2 === "function" && item2.prototype instanceof Model) { |
|
model = item2; |
|
} else if (_.isPlainObject(item2) && item2.model && item2.model.prototype instanceof Model) { |
|
model = item2.model; |
|
as = item2.as; |
|
} |
|
if (model) { |
|
if (!as && previousAssociation && previousAssociation instanceof Association && previousAssociation.through && previousAssociation.through.model === model) { |
|
item2 = new Association(previousModel, model, { |
|
as: model.name |
|
}); |
|
} else { |
|
item2 = previousModel.getAssociationForAlias(model, as); |
|
if (!item2) { |
|
item2 = previousModel.getAssociationForAlias(model, model.name); |
|
} |
|
} |
|
if (!(item2 instanceof Association)) { |
|
throw new Error(util.format("Unable to find a valid association for model, '%s'", model.name)); |
|
} |
|
} |
|
} |
|
if (typeof item2 === "string") { |
|
const orderIndex = validOrderOptions.indexOf(item2.toUpperCase()); |
|
if (index > 0 && orderIndex !== -1) { |
|
item2 = this.sequelize.literal(` ${validOrderOptions[orderIndex]}`); |
|
} else if (previousModel && previousModel.prototype instanceof Model) { |
|
if (previousModel.associations !== void 0 && previousModel.associations[item2]) { |
|
item2 = previousModel.associations[item2]; |
|
} else if (previousModel.rawAttributes !== void 0 && previousModel.rawAttributes[item2] && item2 !== previousModel.rawAttributes[item2].field) { |
|
item2 = previousModel.rawAttributes[item2].field; |
|
} else if (item2.includes(".") && previousModel.rawAttributes !== void 0) { |
|
const itemSplit = item2.split("."); |
|
if (previousModel.rawAttributes[itemSplit[0]].type instanceof DataTypes.JSON) { |
|
const identifier = this.quoteIdentifiers(`${previousModel.name}.${previousModel.rawAttributes[itemSplit[0]].field}`); |
|
const path = itemSplit.slice(1); |
|
item2 = this.jsonPathExtractionQuery(identifier, path); |
|
item2 = this.sequelize.literal(item2); |
|
} |
|
} |
|
} |
|
} |
|
collection[index] = item2; |
|
}, this); |
|
const collectionLength = collection.length; |
|
const tableNames = []; |
|
let item; |
|
let i = 0; |
|
for (i = 0; i < collectionLength - 1; i++) { |
|
item = collection[i]; |
|
if (typeof item === "string" || item._modelAttribute || item instanceof Utils.SequelizeMethod) { |
|
break; |
|
} else if (item instanceof Association) { |
|
tableNames[i] = item.as; |
|
} |
|
} |
|
let sql = ""; |
|
if (i > 0) { |
|
sql += `${this.quoteIdentifier(tableNames.join(connector))}.`; |
|
} else if (typeof collection[0] === "string" && parent) { |
|
sql += `${this.quoteIdentifier(parent.name)}.`; |
|
} |
|
collection.slice(i).forEach((collectionItem) => { |
|
sql += this.quote(collectionItem, parent, connector); |
|
}, this); |
|
return sql; |
|
} |
|
if (collection._modelAttribute) { |
|
return `${this.quoteTable(collection.Model.name)}.${this.quoteIdentifier(collection.fieldName)}`; |
|
} |
|
if (collection instanceof Utils.SequelizeMethod) { |
|
return this.handleSequelizeMethod(collection); |
|
} |
|
if (_.isPlainObject(collection) && collection.raw) { |
|
throw new Error('The `{raw: "..."}` syntax is no longer supported. Use `sequelize.literal` instead.'); |
|
} |
|
throw new Error(`Unknown structure passed to order / group: ${util.inspect(collection)}`); |
|
} |
|
_initQuoteIdentifier() { |
|
this._quoteIdentifier = this.quoteIdentifier; |
|
this.quoteIdentifier = function(identifier, force) { |
|
if (identifier === "*") |
|
return identifier; |
|
return this._quoteIdentifier(identifier, force); |
|
}; |
|
} |
|
quoteIdentifier(identifier, force) { |
|
throw new Error(`quoteIdentifier for Dialect "${this.dialect}" is not implemented`); |
|
} |
|
quoteIdentifiers(identifiers) { |
|
if (identifiers.includes(".")) { |
|
identifiers = identifiers.split("."); |
|
const head = identifiers.slice(0, identifiers.length - 1).join("->"); |
|
const tail = identifiers[identifiers.length - 1]; |
|
return `${this.quoteIdentifier(head)}.${this.quoteIdentifier(tail)}`; |
|
} |
|
return this.quoteIdentifier(identifiers); |
|
} |
|
quoteAttribute(attribute, model) { |
|
if (model && attribute in model.rawAttributes) { |
|
return this.quoteIdentifier(attribute); |
|
} |
|
return this.quoteIdentifiers(attribute); |
|
} |
|
getAliasToken() { |
|
return "AS"; |
|
} |
|
quoteTable(param, alias) { |
|
let table = ""; |
|
if (alias === true) { |
|
alias = param.as || param.name || param; |
|
} |
|
if (_.isObject(param)) { |
|
if (this._dialect.supports.schemas) { |
|
if (param.schema) { |
|
table += `${this.quoteIdentifier(param.schema)}.`; |
|
} |
|
table += this.quoteIdentifier(param.tableName); |
|
} else { |
|
if (param.schema) { |
|
table += param.schema + (param.delimiter || "."); |
|
} |
|
table += param.tableName; |
|
table = this.quoteIdentifier(table); |
|
} |
|
} else { |
|
table = this.quoteIdentifier(param); |
|
} |
|
if (alias) { |
|
table += ` ${this.getAliasToken()} ${this.quoteIdentifier(alias)}`; |
|
} |
|
return table; |
|
} |
|
escape(value, field, options) { |
|
options = options || {}; |
|
if (value !== null && value !== void 0) { |
|
if (value instanceof Utils.SequelizeMethod) { |
|
return this.handleSequelizeMethod(value); |
|
} |
|
if (field && field.type) { |
|
this.validate(value, field, options); |
|
if (field.type.stringify) { |
|
const simpleEscape = (escVal) => SqlString.escape(escVal, this.options.timezone, this.dialect); |
|
value = field.type.stringify(value, { escape: simpleEscape, field, timezone: this.options.timezone, operation: options.operation }); |
|
if (field.type.escape === false) { |
|
return value; |
|
} |
|
} |
|
} |
|
} |
|
return SqlString.escape(value, this.options.timezone, this.dialect); |
|
} |
|
bindParam(bind) { |
|
return (value) => { |
|
bind.push(value); |
|
return `$${bind.length}`; |
|
}; |
|
} |
|
format(value, field, options, bindParam) { |
|
options = options || {}; |
|
if (value !== null && value !== void 0) { |
|
if (value instanceof Utils.SequelizeMethod) { |
|
throw new Error("Cannot pass SequelizeMethod as a bind parameter - use escape instead"); |
|
} |
|
if (field && field.type) { |
|
this.validate(value, field, options); |
|
if (field.type.bindParam) { |
|
return field.type.bindParam(value, { escape: _.identity, field, timezone: this.options.timezone, operation: options.operation, bindParam }); |
|
} |
|
} |
|
} |
|
return bindParam(value); |
|
} |
|
validate(value, field, options) { |
|
if (this.typeValidation && field.type.validate && value) { |
|
try { |
|
if (options.isList && Array.isArray(value)) { |
|
for (const item of value) { |
|
field.type.validate(item, options); |
|
} |
|
} else { |
|
field.type.validate(value, options); |
|
} |
|
} catch (error) { |
|
if (error instanceof sequelizeError.ValidationError) { |
|
error.errors.push(new sequelizeError.ValidationErrorItem(error.message, "Validation error", field.fieldName, value, null, `${field.type.key} validator`)); |
|
} |
|
throw error; |
|
} |
|
} |
|
} |
|
isIdentifierQuoted(identifier) { |
|
return /^\s*(?:([`"'])(?:(?!\1).|\1{2})*\1\.?)+\s*$/i.test(identifier); |
|
} |
|
jsonPathExtractionQuery(column, path, isJson) { |
|
let paths = _.toPath(path); |
|
let pathStr; |
|
const quotedColumn = this.isIdentifierQuoted(column) ? column : this.quoteIdentifier(column); |
|
switch (this.dialect) { |
|
case "mysql": |
|
case "mariadb": |
|
case "sqlite": |
|
if (this.dialect === "mysql") { |
|
paths = paths.map((subPath) => { |
|
return /\D/.test(subPath) ? Utils.addTicks(subPath, '"') : subPath; |
|
}); |
|
} |
|
pathStr = this.escape(["$"].concat(paths).join(".").replace(/\.(\d+)(?:(?=\.)|$)/g, (__, digit) => `[${digit}]`)); |
|
if (this.dialect === "sqlite") { |
|
return `json_extract(${quotedColumn},${pathStr})`; |
|
} |
|
return `json_unquote(json_extract(${quotedColumn},${pathStr}))`; |
|
case "postgres": |
|
const join = isJson ? "#>" : "#>>"; |
|
pathStr = this.escape(`{${paths.join(",")}}`); |
|
return `(${quotedColumn}${join}${pathStr})`; |
|
default: |
|
throw new Error(`Unsupported ${this.dialect} for JSON operations`); |
|
} |
|
} |
|
selectQuery(tableName, options, model) { |
|
options = options || {}; |
|
const limit = options.limit; |
|
const mainQueryItems = []; |
|
const subQueryItems = []; |
|
const subQuery = options.subQuery === void 0 ? limit && options.hasMultiAssociation : options.subQuery; |
|
const attributes = { |
|
main: options.attributes && options.attributes.slice(), |
|
subQuery: null |
|
}; |
|
const mainTable = { |
|
name: tableName, |
|
quotedName: null, |
|
as: null, |
|
model |
|
}; |
|
const topLevelInfo = { |
|
names: mainTable, |
|
options, |
|
subQuery |
|
}; |
|
let mainJoinQueries = []; |
|
let subJoinQueries = []; |
|
let query; |
|
if (this.options.minifyAliases && !options.aliasesMapping) { |
|
options.aliasesMapping = /* @__PURE__ */ new Map(); |
|
options.aliasesByTable = {}; |
|
options.includeAliases = /* @__PURE__ */ new Map(); |
|
} |
|
if (options.tableAs) { |
|
mainTable.as = this.quoteIdentifier(options.tableAs); |
|
} else if (!Array.isArray(mainTable.name) && mainTable.model) { |
|
mainTable.as = this.quoteIdentifier(mainTable.model.name); |
|
} |
|
mainTable.quotedName = !Array.isArray(mainTable.name) ? this.quoteTable(mainTable.name) : tableName.map((t) => { |
|
return Array.isArray(t) ? this.quoteTable(t[0], t[1]) : this.quoteTable(t, true); |
|
}).join(", "); |
|
if (subQuery && attributes.main) { |
|
for (const keyAtt of mainTable.model.primaryKeyAttributes) { |
|
if (!attributes.main.some((attr) => keyAtt === attr || keyAtt === attr[0] || keyAtt === attr[1])) { |
|
attributes.main.push(mainTable.model.rawAttributes[keyAtt].field ? [keyAtt, mainTable.model.rawAttributes[keyAtt].field] : keyAtt); |
|
} |
|
} |
|
} |
|
attributes.main = this.escapeAttributes(attributes.main, options, mainTable.as); |
|
attributes.main = attributes.main || (options.include ? [`${mainTable.as}.*`] : ["*"]); |
|
if (subQuery || options.groupedLimit) { |
|
attributes.subQuery = attributes.main; |
|
attributes.main = [`${mainTable.as || mainTable.quotedName}.*`]; |
|
} |
|
if (options.include) { |
|
for (const include of options.include) { |
|
if (include.separate) { |
|
continue; |
|
} |
|
const joinQueries = this.generateInclude(include, { externalAs: mainTable.as, internalAs: mainTable.as }, topLevelInfo); |
|
subJoinQueries = subJoinQueries.concat(joinQueries.subQuery); |
|
mainJoinQueries = mainJoinQueries.concat(joinQueries.mainQuery); |
|
if (joinQueries.attributes.main.length > 0) { |
|
attributes.main = _.uniq(attributes.main.concat(joinQueries.attributes.main)); |
|
} |
|
if (joinQueries.attributes.subQuery.length > 0) { |
|
attributes.subQuery = _.uniq(attributes.subQuery.concat(joinQueries.attributes.subQuery)); |
|
} |
|
} |
|
} |
|
if (subQuery) { |
|
subQueryItems.push(this.selectFromTableFragment(options, mainTable.model, attributes.subQuery, mainTable.quotedName, mainTable.as)); |
|
subQueryItems.push(subJoinQueries.join("")); |
|
} else { |
|
if (options.groupedLimit) { |
|
if (!mainTable.as) { |
|
mainTable.as = mainTable.quotedName; |
|
} |
|
const where = __spreadValues({}, options.where); |
|
let groupedLimitOrder, whereKey, include, groupedTableName = mainTable.as; |
|
if (typeof options.groupedLimit.on === "string") { |
|
whereKey = options.groupedLimit.on; |
|
} else if (options.groupedLimit.on instanceof HasMany) { |
|
whereKey = options.groupedLimit.on.foreignKeyField; |
|
} |
|
if (options.groupedLimit.on instanceof BelongsToMany) { |
|
groupedTableName = options.groupedLimit.on.manyFromSource.as; |
|
const groupedLimitOptions = Model._validateIncludedElements({ |
|
include: [{ |
|
association: options.groupedLimit.on.manyFromSource, |
|
duplicating: false, |
|
required: true, |
|
where: __spreadValues({ |
|
[Op.placeholder]: true |
|
}, options.groupedLimit.through && options.groupedLimit.through.where) |
|
}], |
|
model |
|
}); |
|
options.hasJoin = true; |
|
options.hasMultiAssociation = true; |
|
options.includeMap = Object.assign(groupedLimitOptions.includeMap, options.includeMap); |
|
options.includeNames = groupedLimitOptions.includeNames.concat(options.includeNames || []); |
|
include = groupedLimitOptions.include; |
|
if (Array.isArray(options.order)) { |
|
options.order.forEach((order, i) => { |
|
if (Array.isArray(order)) { |
|
order = order[0]; |
|
} |
|
let alias = `subquery_order_${i}`; |
|
options.attributes.push([order, alias]); |
|
alias = this.sequelize.literal(this.quote(alias)); |
|
if (Array.isArray(options.order[i])) { |
|
options.order[i][0] = alias; |
|
} else { |
|
options.order[i] = alias; |
|
} |
|
}); |
|
groupedLimitOrder = options.order; |
|
} |
|
} else { |
|
groupedLimitOrder = options.order; |
|
if (!this._dialect.supports.topLevelOrderByRequired) { |
|
delete options.order; |
|
} |
|
where[Op.placeholder] = true; |
|
} |
|
const baseQuery = `SELECT * FROM (${this.selectQuery(tableName, { |
|
attributes: options.attributes, |
|
offset: options.offset, |
|
limit: options.groupedLimit.limit, |
|
order: groupedLimitOrder, |
|
aliasesMapping: options.aliasesMapping, |
|
aliasesByTable: options.aliasesByTable, |
|
where, |
|
include, |
|
model |
|
}, model).replace(/;$/, "")}) ${this.getAliasToken()} sub`; |
|
const placeHolder = this.whereItemQuery(Op.placeholder, true, { model }); |
|
const splicePos = baseQuery.indexOf(placeHolder); |
|
mainQueryItems.push(this.selectFromTableFragment(options, mainTable.model, attributes.main, `(${options.groupedLimit.values.map((value) => { |
|
let groupWhere; |
|
if (whereKey) { |
|
groupWhere = { |
|
[whereKey]: value |
|
}; |
|
} |
|
if (include) { |
|
groupWhere = { |
|
[options.groupedLimit.on.foreignIdentifierField]: value |
|
}; |
|
} |
|
return Utils.spliceStr(baseQuery, splicePos, placeHolder.length, this.getWhereConditions(groupWhere, groupedTableName)); |
|
}).join(this._dialect.supports["UNION ALL"] ? " UNION ALL " : " UNION ")})`, mainTable.as)); |
|
} else { |
|
mainQueryItems.push(this.selectFromTableFragment(options, mainTable.model, attributes.main, mainTable.quotedName, mainTable.as)); |
|
} |
|
mainQueryItems.push(mainJoinQueries.join("")); |
|
} |
|
if (Object.prototype.hasOwnProperty.call(options, "where") && !options.groupedLimit) { |
|
options.where = this.getWhereConditions(options.where, mainTable.as || tableName, model, options); |
|
if (options.where) { |
|
if (subQuery) { |
|
subQueryItems.push(` WHERE ${options.where}`); |
|
} else { |
|
mainQueryItems.push(` WHERE ${options.where}`); |
|
mainQueryItems.forEach((value, key) => { |
|
if (value.startsWith("SELECT")) { |
|
mainQueryItems[key] = this.selectFromTableFragment(options, model, attributes.main, mainTable.quotedName, mainTable.as, options.where); |
|
} |
|
}); |
|
} |
|
} |
|
} |
|
if (options.group) { |
|
options.group = Array.isArray(options.group) ? options.group.map((t) => this.aliasGrouping(t, model, mainTable.as, options)).join(", ") : this.aliasGrouping(options.group, model, mainTable.as, options); |
|
if (subQuery && options.group) { |
|
subQueryItems.push(` GROUP BY ${options.group}`); |
|
} else if (options.group) { |
|
mainQueryItems.push(` GROUP BY ${options.group}`); |
|
} |
|
} |
|
if (Object.prototype.hasOwnProperty.call(options, "having")) { |
|
options.having = this.getWhereConditions(options.having, tableName, model, options, false); |
|
if (options.having) { |
|
if (subQuery) { |
|
subQueryItems.push(` HAVING ${options.having}`); |
|
} else { |
|
mainQueryItems.push(` HAVING ${options.having}`); |
|
} |
|
} |
|
} |
|
if (options.order) { |
|
const orders = this.getQueryOrders(options, model, subQuery); |
|
if (orders.mainQueryOrder.length) { |
|
mainQueryItems.push(` ORDER BY ${orders.mainQueryOrder.join(", ")}`); |
|
} |
|
if (orders.subQueryOrder.length) { |
|
subQueryItems.push(` ORDER BY ${orders.subQueryOrder.join(", ")}`); |
|
} |
|
} |
|
const limitOrder = this.addLimitAndOffset(options, mainTable.model); |
|
if (limitOrder && !options.groupedLimit) { |
|
if (subQuery) { |
|
subQueryItems.push(limitOrder); |
|
} else { |
|
mainQueryItems.push(limitOrder); |
|
} |
|
} |
|
if (subQuery) { |
|
this._throwOnEmptyAttributes(attributes.main, { modelName: model && model.name, as: mainTable.as }); |
|
query = `SELECT ${attributes.main.join(", ")} FROM (${subQueryItems.join("")}) ${this.getAliasToken()} ${mainTable.as}${mainJoinQueries.join("")}${mainQueryItems.join("")}`; |
|
} else { |
|
query = mainQueryItems.join(""); |
|
} |
|
if (options.lock && this._dialect.supports.lock) { |
|
let lock = options.lock; |
|
if (typeof options.lock === "object") { |
|
lock = options.lock.level; |
|
} |
|
if (this._dialect.supports.lockKey && ["KEY SHARE", "NO KEY UPDATE"].includes(lock)) { |
|
query += ` FOR ${lock}`; |
|
} else if (lock === "SHARE") { |
|
query += ` ${this._dialect.supports.forShare}`; |
|
} else { |
|
query += " FOR UPDATE"; |
|
} |
|
if (this._dialect.supports.lockOf && options.lock.of && options.lock.of.prototype instanceof Model) { |
|
query += ` OF ${this.quoteTable(options.lock.of.name)}`; |
|
} |
|
if (this._dialect.supports.skipLocked && options.skipLocked) { |
|
query += " SKIP LOCKED"; |
|
} |
|
} |
|
return `${query};`; |
|
} |
|
aliasGrouping(field, model, tableName, options) { |
|
const src = Array.isArray(field) ? field[0] : field; |
|
return this.quote(this._getAliasForField(tableName, src, options) || src, model); |
|
} |
|
escapeAttributes(attributes, options, mainTableAs) { |
|
return attributes && attributes.map((attr) => { |
|
let addTable = true; |
|
if (attr instanceof Utils.SequelizeMethod) { |
|
return this.handleSequelizeMethod(attr); |
|
} |
|
if (Array.isArray(attr)) { |
|
if (attr.length !== 2) { |
|
throw new Error(`${JSON.stringify(attr)} is not a valid attribute definition. Please use the following format: ['attribute definition', 'alias']`); |
|
} |
|
attr = attr.slice(); |
|
if (attr[0] instanceof Utils.SequelizeMethod) { |
|
attr[0] = this.handleSequelizeMethod(attr[0]); |
|
addTable = false; |
|
} else if (!attr[0].includes("(") && !attr[0].includes(")")) { |
|
attr[0] = this.quoteIdentifier(attr[0]); |
|
} else { |
|
deprecations.noRawAttributes(); |
|
} |
|
let alias = attr[1]; |
|
if (this.options.minifyAliases) { |
|
alias = this._getMinifiedAlias(alias, mainTableAs, options); |
|
} |
|
attr = [attr[0], this.quoteIdentifier(alias)].join(" AS "); |
|
} else { |
|
attr = !attr.includes(Utils.TICK_CHAR) && !attr.includes('"') ? this.quoteAttribute(attr, options.model) : this.escape(attr); |
|
} |
|
if (!_.isEmpty(options.include) && (!attr.includes(".") || options.dotNotation) && addTable) { |
|
attr = `${mainTableAs}.${attr}`; |
|
} |
|
return attr; |
|
}); |
|
} |
|
generateInclude(include, parentTableName, topLevelInfo) { |
|
const joinQueries = { |
|
mainQuery: [], |
|
subQuery: [] |
|
}; |
|
const mainChildIncludes = []; |
|
const subChildIncludes = []; |
|
let requiredMismatch = false; |
|
const includeAs = { |
|
internalAs: include.as, |
|
externalAs: include.as |
|
}; |
|
const attributes = { |
|
main: [], |
|
subQuery: [] |
|
}; |
|
let joinQuery; |
|
topLevelInfo.options.keysEscaped = true; |
|
if (topLevelInfo.names.name !== parentTableName.externalAs && topLevelInfo.names.as !== parentTableName.externalAs) { |
|
includeAs.internalAs = `${parentTableName.internalAs}->${include.as}`; |
|
includeAs.externalAs = `${parentTableName.externalAs}.${include.as}`; |
|
} |
|
if (topLevelInfo.options.includeIgnoreAttributes !== false) { |
|
include.model._expandAttributes(include); |
|
Utils.mapFinderOptions(include, include.model); |
|
const includeAttributes = include.attributes.map((attr) => { |
|
let attrAs = attr; |
|
let verbatim = false; |
|
if (Array.isArray(attr) && attr.length === 2) { |
|
if (attr[0] instanceof Utils.SequelizeMethod && (attr[0] instanceof Utils.Literal || attr[0] instanceof Utils.Cast || attr[0] instanceof Utils.Fn)) { |
|
verbatim = true; |
|
} |
|
attr = attr.map((attr2) => attr2 instanceof Utils.SequelizeMethod ? this.handleSequelizeMethod(attr2) : attr2); |
|
attrAs = attr[1]; |
|
attr = attr[0]; |
|
} |
|
if (attr instanceof Utils.Literal) { |
|
return attr.val; |
|
} |
|
if (attr instanceof Utils.Cast || attr instanceof Utils.Fn) { |
|
throw new Error("Tried to select attributes using Sequelize.cast or Sequelize.fn without specifying an alias for the result, during eager loading. This means the attribute will not be added to the returned instance"); |
|
} |
|
let prefix; |
|
if (verbatim === true) { |
|
prefix = attr; |
|
} else if (/#>>|->>/.test(attr)) { |
|
prefix = `(${this.quoteIdentifier(includeAs.internalAs)}.${attr.replace(/\(|\)/g, "")})`; |
|
} else if (/json_extract\(/.test(attr)) { |
|
prefix = attr.replace(/json_extract\(/i, `json_extract(${this.quoteIdentifier(includeAs.internalAs)}.`); |
|
} else if (/json_value\(/.test(attr)) { |
|
prefix = attr.replace(/json_value\(/i, `json_value(${this.quoteIdentifier(includeAs.internalAs)}.`); |
|
} else { |
|
prefix = `${this.quoteIdentifier(includeAs.internalAs)}.${this.quoteIdentifier(attr)}`; |
|
} |
|
let alias = `${includeAs.externalAs}.${attrAs}`; |
|
if (this.options.minifyAliases) { |
|
alias = this._getMinifiedAlias(alias, includeAs.internalAs, topLevelInfo.options); |
|
} |
|
return Utils.joinSQLFragments([ |
|
prefix, |
|
"AS", |
|
this.quoteIdentifier(alias, true) |
|
]); |
|
}); |
|
if (include.subQuery && topLevelInfo.subQuery) { |
|
for (const attr of includeAttributes) { |
|
attributes.subQuery.push(attr); |
|
} |
|
} else { |
|
for (const attr of includeAttributes) { |
|
attributes.main.push(attr); |
|
} |
|
} |
|
} |
|
if (include.through) { |
|
joinQuery = this.generateThroughJoin(include, includeAs, parentTableName.internalAs, topLevelInfo); |
|
} else { |
|
this._generateSubQueryFilter(include, includeAs, topLevelInfo); |
|
joinQuery = this.generateJoin(include, topLevelInfo); |
|
} |
|
if (joinQuery.attributes.main.length > 0) { |
|
attributes.main = attributes.main.concat(joinQuery.attributes.main); |
|
} |
|
if (joinQuery.attributes.subQuery.length > 0) { |
|
attributes.subQuery = attributes.subQuery.concat(joinQuery.attributes.subQuery); |
|
} |
|
if (include.include) { |
|
for (const childInclude of include.include) { |
|
if (childInclude.separate || childInclude._pseudo) { |
|
continue; |
|
} |
|
const childJoinQueries = this.generateInclude(childInclude, includeAs, topLevelInfo); |
|
if (include.required === false && childInclude.required === true) { |
|
requiredMismatch = true; |
|
} |
|
if (childInclude.subQuery && topLevelInfo.subQuery) { |
|
subChildIncludes.push(childJoinQueries.subQuery); |
|
} |
|
if (childJoinQueries.mainQuery) { |
|
mainChildIncludes.push(childJoinQueries.mainQuery); |
|
} |
|
if (childJoinQueries.attributes.main.length > 0) { |
|
attributes.main = attributes.main.concat(childJoinQueries.attributes.main); |
|
} |
|
if (childJoinQueries.attributes.subQuery.length > 0) { |
|
attributes.subQuery = attributes.subQuery.concat(childJoinQueries.attributes.subQuery); |
|
} |
|
} |
|
} |
|
if (include.subQuery && topLevelInfo.subQuery) { |
|
if (requiredMismatch && subChildIncludes.length > 0) { |
|
joinQueries.subQuery.push(` ${joinQuery.join} ( ${joinQuery.body}${subChildIncludes.join("")} ) ON ${joinQuery.condition}`); |
|
} else { |
|
joinQueries.subQuery.push(` ${joinQuery.join} ${joinQuery.body} ON ${joinQuery.condition}`); |
|
if (subChildIncludes.length > 0) { |
|
joinQueries.subQuery.push(subChildIncludes.join("")); |
|
} |
|
} |
|
joinQueries.mainQuery.push(mainChildIncludes.join("")); |
|
} else { |
|
if (requiredMismatch && mainChildIncludes.length > 0) { |
|
joinQueries.mainQuery.push(` ${joinQuery.join} ( ${joinQuery.body}${mainChildIncludes.join("")} ) ON ${joinQuery.condition}`); |
|
} else { |
|
joinQueries.mainQuery.push(` ${joinQuery.join} ${joinQuery.body} ON ${joinQuery.condition}`); |
|
if (mainChildIncludes.length > 0) { |
|
joinQueries.mainQuery.push(mainChildIncludes.join("")); |
|
} |
|
} |
|
joinQueries.subQuery.push(subChildIncludes.join("")); |
|
} |
|
return { |
|
mainQuery: joinQueries.mainQuery.join(""), |
|
subQuery: joinQueries.subQuery.join(""), |
|
attributes |
|
}; |
|
} |
|
_getMinifiedAlias(alias, tableName, options) { |
|
if (options.aliasesByTable[`${tableName}${alias}`]) { |
|
return options.aliasesByTable[`${tableName}${alias}`]; |
|
} |
|
if (alias.match(/subquery_order_[0-9]/)) { |
|
return alias; |
|
} |
|
const minifiedAlias = `_${options.aliasesMapping.size}`; |
|
options.aliasesMapping.set(minifiedAlias, alias); |
|
options.aliasesByTable[`${tableName}${alias}`] = minifiedAlias; |
|
return minifiedAlias; |
|
} |
|
_getAliasForField(tableName, field, options) { |
|
if (this.options.minifyAliases) { |
|
if (options.aliasesByTable[`${tableName}${field}`]) { |
|
return options.aliasesByTable[`${tableName}${field}`]; |
|
} |
|
} |
|
return null; |
|
} |
|
generateJoin(include, topLevelInfo) { |
|
const association = include.association; |
|
const parent = include.parent; |
|
const parentIsTop = !!parent && !include.parent.association && include.parent.model.name === topLevelInfo.options.model.name; |
|
let $parent; |
|
let joinWhere; |
|
const left = association.source; |
|
const attrLeft = association instanceof BelongsTo ? association.identifier : association.sourceKeyAttribute || left.primaryKeyAttribute; |
|
const fieldLeft = association instanceof BelongsTo ? association.identifierField : left.rawAttributes[association.sourceKeyAttribute || left.primaryKeyAttribute].field; |
|
let asLeft; |
|
const right = include.model; |
|
const tableRight = right.getTableName(); |
|
const fieldRight = association instanceof BelongsTo ? right.rawAttributes[association.targetIdentifier || right.primaryKeyAttribute].field : association.identifierField; |
|
let asRight = include.as; |
|
while (($parent = $parent && $parent.parent || include.parent) && $parent.association) { |
|
if (asLeft) { |
|
asLeft = `${$parent.as}->${asLeft}`; |
|
} else { |
|
asLeft = $parent.as; |
|
} |
|
} |
|
if (!asLeft) |
|
asLeft = parent.as || parent.model.name; |
|
else |
|
asRight = `${asLeft}->${asRight}`; |
|
let joinOn = `${this.quoteTable(asLeft)}.${this.quoteIdentifier(fieldLeft)}`; |
|
const subqueryAttributes = []; |
|
if (topLevelInfo.options.groupedLimit && parentIsTop || topLevelInfo.subQuery && include.parent.subQuery && !include.subQuery) { |
|
if (parentIsTop) { |
|
const tableName = this.quoteTable(parent.as || parent.model.name); |
|
joinOn = this._getAliasForField(tableName, attrLeft, topLevelInfo.options) || `${tableName}.${this.quoteIdentifier(attrLeft)}`; |
|
if (topLevelInfo.subQuery) { |
|
const dbIdentifier = `${tableName}.${this.quoteIdentifier(fieldLeft)}`; |
|
subqueryAttributes.push(dbIdentifier !== joinOn ? `${dbIdentifier} AS ${this.quoteIdentifier(attrLeft)}` : dbIdentifier); |
|
} |
|
} else { |
|
const joinSource = `${asLeft.replace(/->/g, ".")}.${attrLeft}`; |
|
joinOn = this._getAliasForField(asLeft, joinSource, topLevelInfo.options) || this.quoteIdentifier(joinSource); |
|
} |
|
} |
|
joinOn += ` = ${this.quoteIdentifier(asRight)}.${this.quoteIdentifier(fieldRight)}`; |
|
if (include.on) { |
|
joinOn = this.whereItemsQuery(include.on, { |
|
prefix: this.sequelize.literal(this.quoteIdentifier(asRight)), |
|
model: include.model |
|
}); |
|
} |
|
if (include.where) { |
|
joinWhere = this.whereItemsQuery(include.where, { |
|
prefix: this.sequelize.literal(this.quoteIdentifier(asRight)), |
|
model: include.model |
|
}); |
|
if (joinWhere) { |
|
if (include.or) { |
|
joinOn += ` OR ${joinWhere}`; |
|
} else { |
|
joinOn += ` AND ${joinWhere}`; |
|
} |
|
} |
|
} |
|
if (this.options.minifyAliases && asRight.length > 63) { |
|
const alias = `%${topLevelInfo.options.includeAliases.size}`; |
|
topLevelInfo.options.includeAliases.set(alias, asRight); |
|
} |
|
return { |
|
join: include.required ? "INNER JOIN" : include.right && this._dialect.supports["RIGHT JOIN"] ? "RIGHT OUTER JOIN" : "LEFT OUTER JOIN", |
|
body: this.quoteTable(tableRight, asRight), |
|
condition: joinOn, |
|
attributes: { |
|
main: [], |
|
subQuery: subqueryAttributes |
|
} |
|
}; |
|
} |
|
generateReturnValues(modelAttributes, options) { |
|
const returnFields = []; |
|
const returnTypes = []; |
|
let outputFragment = ""; |
|
let returningFragment = ""; |
|
let tmpTable = ""; |
|
if (Array.isArray(options.returning)) { |
|
returnFields.push(...options.returning.map((field) => this.quoteIdentifier(field))); |
|
} else if (modelAttributes) { |
|
_.each(modelAttributes, (attribute) => { |
|
if (!(attribute.type instanceof DataTypes.VIRTUAL)) { |
|
returnFields.push(this.quoteIdentifier(attribute.field)); |
|
returnTypes.push(attribute.type); |
|
} |
|
}); |
|
} |
|
if (_.isEmpty(returnFields)) { |
|
returnFields.push("*"); |
|
} |
|
if (this._dialect.supports.returnValues.returning) { |
|
returningFragment = ` RETURNING ${returnFields.join(",")}`; |
|
} else if (this._dialect.supports.returnIntoValues) { |
|
returningFragment = ` RETURNING ${returnFields.join(",")} INTO `; |
|
} else if (this._dialect.supports.returnValues.output) { |
|
outputFragment = ` OUTPUT ${returnFields.map((field) => `INSERTED.${field}`).join(",")}`; |
|
if (options.hasTrigger && this._dialect.supports.tmpTableTrigger) { |
|
const tmpColumns = returnFields.map((field, i) => `${field} ${returnTypes[i].toSql()}`); |
|
tmpTable = `DECLARE @tmp TABLE (${tmpColumns.join(",")}); `; |
|
outputFragment += " INTO @tmp"; |
|
returningFragment = "; SELECT * FROM @tmp"; |
|
} |
|
} |
|
return { outputFragment, returnFields, returnTypes, returningFragment, tmpTable }; |
|
} |
|
generateThroughJoin(include, includeAs, parentTableName, topLevelInfo) { |
|
const through = include.through; |
|
const throughTable = through.model.getTableName(); |
|
const throughAs = `${includeAs.internalAs}->${through.as}`; |
|
const externalThroughAs = `${includeAs.externalAs}.${through.as}`; |
|
const throughAttributes = through.attributes.map((attr) => { |
|
let alias = `${externalThroughAs}.${Array.isArray(attr) ? attr[1] : attr}`; |
|
if (this.options.minifyAliases) { |
|
alias = this._getMinifiedAlias(alias, throughAs, topLevelInfo.options); |
|
} |
|
return Utils.joinSQLFragments([ |
|
`${this.quoteIdentifier(throughAs)}.${this.quoteIdentifier(Array.isArray(attr) ? attr[0] : attr)}`, |
|
"AS", |
|
this.quoteIdentifier(alias) |
|
]); |
|
}); |
|
const association = include.association; |
|
const parentIsTop = !include.parent.association && include.parent.model.name === topLevelInfo.options.model.name; |
|
const tableSource = parentTableName; |
|
const identSource = association.identifierField; |
|
const tableTarget = includeAs.internalAs; |
|
const identTarget = association.foreignIdentifierField; |
|
const attrTarget = association.targetKeyField; |
|
const joinType = include.required ? "INNER JOIN" : include.right && this._dialect.supports["RIGHT JOIN"] ? "RIGHT OUTER JOIN" : "LEFT OUTER JOIN"; |
|
let joinBody; |
|
let joinCondition; |
|
const attributes = { |
|
main: [], |
|
subQuery: [] |
|
}; |
|
let attrSource = association.sourceKey; |
|
let sourceJoinOn; |
|
let targetJoinOn; |
|
let throughWhere; |
|
let targetWhere; |
|
if (topLevelInfo.options.includeIgnoreAttributes !== false) { |
|
for (const attr of throughAttributes) { |
|
attributes.main.push(attr); |
|
} |
|
} |
|
if (!topLevelInfo.subQuery) { |
|
attrSource = association.sourceKeyField; |
|
} |
|
if (topLevelInfo.subQuery && !include.subQuery && !include.parent.subQuery && include.parent.model !== topLevelInfo.options.mainModel) { |
|
attrSource = association.sourceKeyField; |
|
} |
|
if (topLevelInfo.subQuery && !include.subQuery && include.parent.subQuery && !parentIsTop) { |
|
const joinSource = this._getAliasForField(tableSource, `${tableSource}.${attrSource}`, topLevelInfo.options) || `${tableSource}.${attrSource}`; |
|
sourceJoinOn = `${this.quoteIdentifier(joinSource)} = `; |
|
} else { |
|
const aliasedSource = this._getAliasForField(tableSource, attrSource, topLevelInfo.options) || attrSource; |
|
sourceJoinOn = `${this.quoteTable(tableSource)}.${this.quoteIdentifier(aliasedSource)} = `; |
|
} |
|
sourceJoinOn += `${this.quoteIdentifier(throughAs)}.${this.quoteIdentifier(identSource)}`; |
|
targetJoinOn = `${this.quoteIdentifier(tableTarget)}.${this.quoteIdentifier(attrTarget)} = `; |
|
targetJoinOn += `${this.quoteIdentifier(throughAs)}.${this.quoteIdentifier(identTarget)}`; |
|
if (through.where) { |
|
throughWhere = this.getWhereConditions(through.where, this.sequelize.literal(this.quoteIdentifier(throughAs)), through.model); |
|
} |
|
joinBody = `( ${this.quoteTable(throughTable, throughAs)} INNER JOIN ${this.quoteTable(include.model.getTableName(), includeAs.internalAs)} ON ${targetJoinOn}`; |
|
if (throughWhere) { |
|
joinBody += ` AND ${throughWhere}`; |
|
} |
|
joinBody += ")"; |
|
joinCondition = sourceJoinOn; |
|
if (include.where || include.through.where) { |
|
if (include.where) { |
|
targetWhere = this.getWhereConditions(include.where, this.sequelize.literal(this.quoteIdentifier(includeAs.internalAs)), include.model, topLevelInfo.options); |
|
if (targetWhere) { |
|
joinCondition += ` AND ${targetWhere}`; |
|
} |
|
} |
|
} |
|
this._generateSubQueryFilter(include, includeAs, topLevelInfo); |
|
return { |
|
join: joinType, |
|
body: joinBody, |
|
condition: joinCondition, |
|
attributes |
|
}; |
|
} |
|
_generateSubQueryFilter(include, includeAs, topLevelInfo) { |
|
if (!topLevelInfo.subQuery || !include.subQueryFilter) { |
|
return; |
|
} |
|
if (!topLevelInfo.options.where) { |
|
topLevelInfo.options.where = {}; |
|
} |
|
let parent = include; |
|
let child = include; |
|
let nestedIncludes = this._getRequiredClosure(include).include; |
|
let query; |
|
while (parent = parent.parent) { |
|
if (parent.parent && !parent.required) { |
|
return; |
|
} |
|
if (parent.subQueryFilter) { |
|
return; |
|
} |
|
nestedIncludes = [__spreadProps(__spreadValues({}, child), { include: nestedIncludes, attributes: [] })]; |
|
child = parent; |
|
} |
|
const topInclude = nestedIncludes[0]; |
|
const topParent = topInclude.parent; |
|
const topAssociation = topInclude.association; |
|
topInclude.association = void 0; |
|
if (topInclude.through && Object(topInclude.through.model) === topInclude.through.model) { |
|
query = this.selectQuery(topInclude.through.model.getTableName(), { |
|
attributes: [topInclude.through.model.primaryKeyField], |
|
include: Model._validateIncludedElements({ |
|
model: topInclude.through.model, |
|
include: [{ |
|
association: topAssociation.toTarget, |
|
required: true, |
|
where: topInclude.where, |
|
include: topInclude.include |
|
}] |
|
}).include, |
|
model: topInclude.through.model, |
|
where: { |
|
[Op.and]: [ |
|
this.sequelize.literal([ |
|
`${this.quoteTable(topParent.model.name)}.${this.quoteIdentifier(topParent.model.primaryKeyField)}`, |
|
`${this.quoteIdentifier(topInclude.through.model.name)}.${this.quoteIdentifier(topAssociation.identifierField)}` |
|
].join(" = ")), |
|
topInclude.through.where |
|
] |
|
}, |
|
limit: 1, |
|
includeIgnoreAttributes: false |
|
}, topInclude.through.model); |
|
} else { |
|
const isBelongsTo = topAssociation.associationType === "BelongsTo"; |
|
const sourceField = isBelongsTo ? topAssociation.identifierField : topAssociation.sourceKeyField || topParent.model.primaryKeyField; |
|
const targetField = isBelongsTo ? topAssociation.sourceKeyField || topInclude.model.primaryKeyField : topAssociation.identifierField; |
|
const join = [ |
|
`${this.quoteIdentifier(topInclude.as)}.${this.quoteIdentifier(targetField)}`, |
|
`${this.quoteTable(topParent.as || topParent.model.name)}.${this.quoteIdentifier(sourceField)}` |
|
].join(" = "); |
|
query = this.selectQuery(topInclude.model.getTableName(), { |
|
attributes: [targetField], |
|
include: Model._validateIncludedElements(topInclude).include, |
|
model: topInclude.model, |
|
where: { |
|
[Op.and]: [ |
|
topInclude.where, |
|
{ [Op.join]: this.sequelize.literal(join) } |
|
] |
|
}, |
|
limit: 1, |
|
tableAs: topInclude.as, |
|
includeIgnoreAttributes: false |
|
}, topInclude.model); |
|
} |
|
if (!topLevelInfo.options.where[Op.and]) { |
|
topLevelInfo.options.where[Op.and] = []; |
|
} |
|
topLevelInfo.options.where[`__${includeAs.internalAs}`] = this.sequelize.literal([ |
|
"(", |
|
query.replace(/;$/, ""), |
|
")", |
|
"IS NOT NULL" |
|
].join(" ")); |
|
} |
|
_getRequiredClosure(include) { |
|
const copy = __spreadProps(__spreadValues({}, include), { attributes: [], include: [] }); |
|
if (Array.isArray(include.include)) { |
|
copy.include = include.include.filter((i) => i.required).map((inc) => this._getRequiredClosure(inc)); |
|
} |
|
return copy; |
|
} |
|
getQueryOrders(options, model, subQuery) { |
|
const mainQueryOrder = []; |
|
const subQueryOrder = []; |
|
if (Array.isArray(options.order)) { |
|
for (let order of options.order) { |
|
if (!Array.isArray(order)) { |
|
order = [order]; |
|
} |
|
if (subQuery && Array.isArray(order) && order[0] && !(order[0] instanceof Association) && !(typeof order[0] === "function" && order[0].prototype instanceof Model) && !(typeof order[0].model === "function" && order[0].model.prototype instanceof Model) && !(typeof order[0] === "string" && model && model.associations !== void 0 && model.associations[order[0]])) { |
|
const field = model.rawAttributes[order[0]] ? model.rawAttributes[order[0]].field : order[0]; |
|
const subQueryAlias = this._getAliasForField(this.quoteIdentifier(model.name), field, options); |
|
let parent = null; |
|
let orderToQuote = []; |
|
if (subQueryAlias === null) { |
|
orderToQuote = order; |
|
parent = model; |
|
} else { |
|
orderToQuote = [subQueryAlias, order.length > 1 ? order[1] : "ASC"]; |
|
parent = null; |
|
} |
|
subQueryOrder.push(this.quote(orderToQuote, parent, "->")); |
|
} |
|
if (options.attributes && model) { |
|
const aliasedAttribute = options.attributes.find((attr) => Array.isArray(attr) && attr[1] && (attr[0] === order[0] || attr[1] === order[0])); |
|
if (aliasedAttribute) { |
|
const modelName = this.quoteIdentifier(model.name); |
|
const alias = this._getAliasForField(modelName, aliasedAttribute[1], options); |
|
order[0] = new Utils.Col(alias || aliasedAttribute[1]); |
|
} |
|
} |
|
mainQueryOrder.push(this.quote(order, model, "->")); |
|
} |
|
} else if (options.order instanceof Utils.SequelizeMethod) { |
|
const sql = this.quote(options.order, model, "->"); |
|
if (subQuery) { |
|
subQueryOrder.push(sql); |
|
} |
|
mainQueryOrder.push(sql); |
|
} else { |
|
throw new Error("Order must be type of array or instance of a valid sequelize method."); |
|
} |
|
return { mainQueryOrder, subQueryOrder }; |
|
} |
|
_throwOnEmptyAttributes(attributes, extraInfo = {}) { |
|
if (attributes.length > 0) |
|
return; |
|
const asPart = extraInfo.as && `as ${extraInfo.as}` || ""; |
|
const namePart = extraInfo.modelName && `for model '${extraInfo.modelName}'` || ""; |
|
const message = `Attempted a SELECT query ${namePart} ${asPart} without selecting any columns`; |
|
throw new sequelizeError.QueryError(message.replace(/ +/g, " ")); |
|
} |
|
selectFromTableFragment(options, model, attributes, tables, mainTableAs) { |
|
this._throwOnEmptyAttributes(attributes, { modelName: model && model.name, as: mainTableAs }); |
|
let fragment = `SELECT ${attributes.join(", ")} FROM ${tables}`; |
|
if (mainTableAs) { |
|
fragment += ` ${this.getAliasToken()} ${mainTableAs}`; |
|
} |
|
if (options.indexHints && this._dialect.supports.indexHints) { |
|
for (const hint of options.indexHints) { |
|
if (IndexHints[hint.type]) { |
|
fragment += ` ${IndexHints[hint.type]} INDEX (${hint.values.map((indexName) => this.quoteIdentifiers(indexName)).join(",")})`; |
|
} |
|
} |
|
} |
|
return fragment; |
|
} |
|
addLimitAndOffset(options) { |
|
let fragment = ""; |
|
if (options.offset != null && options.limit == null) { |
|
fragment += " LIMIT " + this.escape(options.offset) + ", " + 1e13; |
|
} else if (options.limit != null) { |
|
if (options.offset != null) { |
|
fragment += " LIMIT " + this.escape(options.offset) + ", " + this.escape(options.limit); |
|
} else { |
|
fragment += " LIMIT " + this.escape(options.limit); |
|
} |
|
} |
|
return fragment; |
|
} |
|
handleSequelizeMethod(smth, tableName, factory, options, prepend) { |
|
let result; |
|
if (Object.prototype.hasOwnProperty.call(this.OperatorMap, smth.comparator)) { |
|
smth.comparator = this.OperatorMap[smth.comparator]; |
|
} |
|
if (smth instanceof Utils.Where) { |
|
let value = smth.logic; |
|
let key; |
|
if (smth.attribute instanceof Utils.SequelizeMethod) { |
|
key = this.getWhereConditions(smth.attribute, tableName, factory, options, prepend); |
|
} else { |
|
key = `${this.quoteTable(smth.attribute.Model.name)}.${this.quoteIdentifier(smth.attribute.field || smth.attribute.fieldName)}`; |
|
} |
|
if (value && value instanceof Utils.SequelizeMethod) { |
|
value = this.getWhereConditions(value, tableName, factory, options, prepend); |
|
if (value === "NULL") { |
|
if (smth.comparator === "=") { |
|
smth.comparator = "IS"; |
|
} |
|
if (smth.comparator === "!=") { |
|
smth.comparator = "IS NOT"; |
|
} |
|
} |
|
return [key, value].join(` ${smth.comparator} `); |
|
} |
|
if (_.isPlainObject(value)) { |
|
return this.whereItemQuery(smth.attribute, value, { |
|
model: factory |
|
}); |
|
} |
|
if ([this.OperatorMap[Op.between], this.OperatorMap[Op.notBetween]].includes(smth.comparator)) { |
|
value = `${this.escape(value[0])} AND ${this.escape(value[1])}`; |
|
} else if (typeof value === "boolean") { |
|
value = this.booleanValue(value); |
|
} else { |
|
value = this.escape(value); |
|
} |
|
if (value === "NULL") { |
|
if (smth.comparator === "=") { |
|
smth.comparator = "IS"; |
|
} |
|
if (smth.comparator === "!=") { |
|
smth.comparator = "IS NOT"; |
|
} |
|
} |
|
return [key, value].join(` ${smth.comparator} `); |
|
} |
|
if (smth instanceof Utils.Literal) { |
|
return smth.val; |
|
} |
|
if (smth instanceof Utils.Cast) { |
|
if (smth.val instanceof Utils.SequelizeMethod) { |
|
result = this.handleSequelizeMethod(smth.val, tableName, factory, options, prepend); |
|
} else if (_.isPlainObject(smth.val)) { |
|
result = this.whereItemsQuery(smth.val); |
|
} else { |
|
result = this.escape(smth.val); |
|
} |
|
return `CAST(${result} AS ${smth.type.toUpperCase()})`; |
|
} |
|
if (smth instanceof Utils.Fn) { |
|
return `${smth.fn}(${smth.args.map((arg) => { |
|
if (arg instanceof Utils.SequelizeMethod) { |
|
return this.handleSequelizeMethod(arg, tableName, factory, options, prepend); |
|
} |
|
if (_.isPlainObject(arg)) { |
|
return this.whereItemsQuery(arg); |
|
} |
|
return this.escape(typeof arg === "string" ? arg.replace(/\$/g, "$$$") : arg); |
|
}).join(", ")})`; |
|
} |
|
if (smth instanceof Utils.Col) { |
|
if (Array.isArray(smth.col) && !factory) { |
|
throw new Error("Cannot call Sequelize.col() with array outside of order / group clause"); |
|
} |
|
if (smth.col.startsWith("*")) { |
|
return "*"; |
|
} |
|
return this.quote(smth.col, factory); |
|
} |
|
return smth.toString(this, factory); |
|
} |
|
whereQuery(where, options) { |
|
const query = this.whereItemsQuery(where, options); |
|
if (query && query.length) { |
|
return `WHERE ${query}`; |
|
} |
|
return ""; |
|
} |
|
whereItemsQuery(where, options, binding) { |
|
if (where === null || where === void 0 || Utils.getComplexSize(where) === 0) { |
|
return ""; |
|
} |
|
if (typeof where === "string") { |
|
throw new Error("Support for `{where: 'raw query'}` has been removed."); |
|
} |
|
const items = []; |
|
binding = binding || "AND"; |
|
if (binding[0] !== " ") |
|
binding = ` ${binding} `; |
|
if (_.isPlainObject(where)) { |
|
Utils.getComplexKeys(where).forEach((prop) => { |
|
const item = where[prop]; |
|
items.push(this.whereItemQuery(prop, item, options)); |
|
}); |
|
} else { |
|
items.push(this.whereItemQuery(void 0, where, options)); |
|
} |
|
return items.length && items.filter((item) => item && item.length).join(binding) || ""; |
|
} |
|
whereItemQuery(key, value, options = {}) { |
|
if (value === void 0) { |
|
throw new Error(`WHERE parameter "${key}" has invalid "undefined" value`); |
|
} |
|
if (typeof key === "string" && key.includes(".") && options.model) { |
|
const keyParts = key.split("."); |
|
if (options.model.rawAttributes[keyParts[0]] && options.model.rawAttributes[keyParts[0]].type instanceof DataTypes.JSON) { |
|
const tmp = {}; |
|
const field2 = options.model.rawAttributes[keyParts[0]]; |
|
_.set(tmp, keyParts.slice(1), value); |
|
return this.whereItemQuery(field2.field || keyParts[0], tmp, __spreadValues({ field: field2 }, options)); |
|
} |
|
} |
|
const field = this._findField(key, options); |
|
const fieldType = field && field.type || options.type; |
|
const isPlainObject = _.isPlainObject(value); |
|
const isArray = !isPlainObject && Array.isArray(value); |
|
key = this.OperatorsAliasMap && this.OperatorsAliasMap[key] || key; |
|
if (isPlainObject) { |
|
value = this._replaceAliases(value); |
|
} |
|
const valueKeys = isPlainObject && Utils.getComplexKeys(value); |
|
if (key === void 0) { |
|
if (typeof value === "string") { |
|
return value; |
|
} |
|
if (isPlainObject && valueKeys.length === 1) { |
|
return this.whereItemQuery(valueKeys[0], value[valueKeys[0]], options); |
|
} |
|
} |
|
if (value === null) { |
|
const opValue2 = options.bindParam ? "NULL" : this.escape(value, field); |
|
return this._joinKeyValue(key, opValue2, this.OperatorMap[Op.is], options.prefix); |
|
} |
|
if (!value) { |
|
const opValue2 = options.bindParam ? this.format(value, field, options, options.bindParam) : this.escape(value, field); |
|
return this._joinKeyValue(key, opValue2, this.OperatorMap[Op.eq], options.prefix); |
|
} |
|
if (value instanceof Utils.SequelizeMethod && !(key !== void 0 && value instanceof Utils.Fn)) { |
|
return this.handleSequelizeMethod(value); |
|
} |
|
if (key === void 0 && isArray) { |
|
if (Utils.canTreatArrayAsAnd(value)) { |
|
key = Op.and; |
|
} else { |
|
throw new Error("Support for literal replacements in the `where` object has been removed."); |
|
} |
|
} |
|
if (key === Op.or || key === Op.and || key === Op.not) { |
|
return this._whereGroupBind(key, value, options); |
|
} |
|
if (value[Op.or]) { |
|
return this._whereBind(this.OperatorMap[Op.or], key, value[Op.or], options); |
|
} |
|
if (value[Op.and]) { |
|
return this._whereBind(this.OperatorMap[Op.and], key, value[Op.and], options); |
|
} |
|
if (isArray && fieldType instanceof DataTypes.ARRAY) { |
|
const opValue2 = options.bindParam ? this.format(value, field, options, options.bindParam) : this.escape(value, field); |
|
return this._joinKeyValue(key, opValue2, this.OperatorMap[Op.eq], options.prefix); |
|
} |
|
if (isPlainObject && fieldType instanceof DataTypes.JSON && options.json !== false) { |
|
return this._whereJSON(key, value, options); |
|
} |
|
if (isPlainObject && valueKeys.length > 1) { |
|
return this._whereBind(this.OperatorMap[Op.and], key, value, options); |
|
} |
|
if (isArray) { |
|
return this._whereParseSingleValueObject(key, field, Op.in, value, options); |
|
} |
|
if (isPlainObject) { |
|
if (this.OperatorMap[valueKeys[0]]) { |
|
return this._whereParseSingleValueObject(key, field, valueKeys[0], value[valueKeys[0]], options); |
|
} |
|
return this._whereParseSingleValueObject(key, field, this.OperatorMap[Op.eq], value, options); |
|
} |
|
if (key === Op.placeholder) { |
|
const opValue2 = options.bindParam ? this.format(value, field, options, options.bindParam) : this.escape(value, field); |
|
return this._joinKeyValue(this.OperatorMap[key], opValue2, this.OperatorMap[Op.eq], options.prefix); |
|
} |
|
const opValue = options.bindParam ? this.format(value, field, options, options.bindParam) : this.escape(value, field); |
|
return this._joinKeyValue(key, opValue, this.OperatorMap[Op.eq], options.prefix); |
|
} |
|
_findField(key, options) { |
|
if (options.field) { |
|
return options.field; |
|
} |
|
if (options.model && options.model.rawAttributes && options.model.rawAttributes[key]) { |
|
return options.model.rawAttributes[key]; |
|
} |
|
if (options.model && options.model.fieldRawAttributesMap && options.model.fieldRawAttributesMap[key]) { |
|
return options.model.fieldRawAttributesMap[key]; |
|
} |
|
} |
|
_whereGroupBind(key, value, options) { |
|
const binding = key === Op.or ? this.OperatorMap[Op.or] : this.OperatorMap[Op.and]; |
|
const outerBinding = key === Op.not ? "NOT " : ""; |
|
if (Array.isArray(value)) { |
|
value = value.map((item) => { |
|
let itemQuery = this.whereItemsQuery(item, options, this.OperatorMap[Op.and]); |
|
if (itemQuery && itemQuery.length && (Array.isArray(item) || _.isPlainObject(item)) && Utils.getComplexSize(item) > 1) { |
|
itemQuery = `(${itemQuery})`; |
|
} |
|
return itemQuery; |
|
}).filter((item) => item && item.length); |
|
value = value.length && value.join(binding); |
|
} else { |
|
value = this.whereItemsQuery(value, options, binding); |
|
} |
|
if ((key === Op.or || key === Op.not) && !value) { |
|
return "0 = 1"; |
|
} |
|
return value ? `${outerBinding}(${value})` : void 0; |
|
} |
|
_whereBind(binding, key, value, options) { |
|
if (_.isPlainObject(value)) { |
|
value = Utils.getComplexKeys(value).map((prop) => { |
|
const item = value[prop]; |
|
return this.whereItemQuery(key, { [prop]: item }, options); |
|
}); |
|
} else { |
|
value = value.map((item) => this.whereItemQuery(key, item, options)); |
|
} |
|
value = value.filter((item) => item && item.length); |
|
return value.length ? `(${value.join(binding)})` : void 0; |
|
} |
|
_whereJSON(key, value, options) { |
|
const items = []; |
|
let baseKey = this.quoteIdentifier(key); |
|
if (options.prefix) { |
|
if (options.prefix instanceof Utils.Literal) { |
|
baseKey = `${this.handleSequelizeMethod(options.prefix)}.${baseKey}`; |
|
} else { |
|
baseKey = `${this.quoteTable(options.prefix)}.${baseKey}`; |
|
} |
|
} |
|
Utils.getOperators(value).forEach((op) => { |
|
const where = { |
|
[op]: value[op] |
|
}; |
|
items.push(this.whereItemQuery(key, where, __spreadProps(__spreadValues({}, options), { json: false }))); |
|
}); |
|
_.forOwn(value, (item, prop) => { |
|
this._traverseJSON(items, baseKey, prop, item, [prop]); |
|
}); |
|
const result = items.join(this.OperatorMap[Op.and]); |
|
return items.length > 1 ? `(${result})` : result; |
|
} |
|
_traverseJSON(items, baseKey, prop, item, path) { |
|
let cast; |
|
if (path[path.length - 1].includes("::")) { |
|
const tmp = path[path.length - 1].split("::"); |
|
cast = tmp[1]; |
|
path[path.length - 1] = tmp[0]; |
|
} |
|
let pathKey = this.jsonPathExtractionQuery(baseKey, path); |
|
if (_.isPlainObject(item)) { |
|
Utils.getOperators(item).forEach((op) => { |
|
const value = this._toJSONValue(item[op]); |
|
let isJson = false; |
|
if (typeof value === "string" && op === Op.contains) { |
|
try { |
|
JSON.stringify(value); |
|
isJson = true; |
|
} catch (e) { |
|
} |
|
} |
|
pathKey = this.jsonPathExtractionQuery(baseKey, path, isJson); |
|
items.push(this.whereItemQuery(this._castKey(pathKey, value, cast), { [op]: value })); |
|
}); |
|
_.forOwn(item, (value, itemProp) => { |
|
this._traverseJSON(items, baseKey, itemProp, value, path.concat([itemProp])); |
|
}); |
|
return; |
|
} |
|
item = this._toJSONValue(item); |
|
items.push(this.whereItemQuery(this._castKey(pathKey, item, cast), { [Op.eq]: item })); |
|
} |
|
_toJSONValue(value) { |
|
return value; |
|
} |
|
_castKey(key, value, cast, json) { |
|
cast = cast || this._getJsonCast(Array.isArray(value) ? value[0] : value); |
|
if (cast) { |
|
return new Utils.Literal(this.handleSequelizeMethod(new Utils.Cast(new Utils.Literal(key), cast, json))); |
|
} |
|
return new Utils.Literal(key); |
|
} |
|
_getJsonCast(value) { |
|
if (typeof value === "number") { |
|
return "double precision"; |
|
} |
|
if (value instanceof Date) { |
|
return "timestamptz"; |
|
} |
|
if (typeof value === "boolean") { |
|
return "boolean"; |
|
} |
|
return; |
|
} |
|
_joinKeyValue(key, value, comparator, prefix) { |
|
if (!key) { |
|
return value; |
|
} |
|
if (comparator === void 0) { |
|
throw new Error(`${key} and ${value} has no comparator`); |
|
} |
|
key = this._getSafeKey(key, prefix); |
|
return [key, value].join(` ${comparator} `); |
|
} |
|
_getSafeKey(key, prefix) { |
|
if (key instanceof Utils.SequelizeMethod) { |
|
key = this.handleSequelizeMethod(key); |
|
return this._prefixKey(this.handleSequelizeMethod(key), prefix); |
|
} |
|
if (Utils.isColString(key)) { |
|
key = key.substr(1, key.length - 2).split("."); |
|
if (key.length > 2) { |
|
key = [ |
|
key.slice(0, -1).join("->"), |
|
key[key.length - 1] |
|
]; |
|
} |
|
return key.map((identifier) => this.quoteIdentifier(identifier)).join("."); |
|
} |
|
return this._prefixKey(this.quoteIdentifier(key), prefix); |
|
} |
|
_prefixKey(key, prefix) { |
|
if (prefix) { |
|
if (prefix instanceof Utils.Literal) { |
|
return [this.handleSequelizeMethod(prefix), key].join("."); |
|
} |
|
return [this.quoteTable(prefix), key].join("."); |
|
} |
|
return key; |
|
} |
|
_whereParseSingleValueObject(key, field, prop, value, options) { |
|
if (prop === Op.not) { |
|
if (Array.isArray(value)) { |
|
prop = Op.notIn; |
|
} else if (value !== null && value !== true && value !== false) { |
|
prop = Op.ne; |
|
} |
|
} |
|
let comparator = this.OperatorMap[prop] || this.OperatorMap[Op.eq]; |
|
switch (prop) { |
|
case Op.in: |
|
case Op.notIn: |
|
if (value instanceof Utils.Literal) { |
|
return this._joinKeyValue(key, value.val, comparator, options.prefix); |
|
} |
|
if (value.length) { |
|
return this._joinKeyValue(key, `(${value.map((item) => this.escape(item, field)).join(", ")})`, comparator, options.prefix); |
|
} |
|
if (comparator === this.OperatorMap[Op.in]) { |
|
return this._joinKeyValue(key, "(NULL)", comparator, options.prefix); |
|
} |
|
return ""; |
|
case Op.any: |
|
case Op.all: |
|
comparator = `${this.OperatorMap[Op.eq]} ${comparator}`; |
|
if (value[Op.values]) { |
|
return this._joinKeyValue(key, `(VALUES ${value[Op.values].map((item) => `(${this.escape(item)})`).join(", ")})`, comparator, options.prefix); |
|
} |
|
return this._joinKeyValue(key, `(${this.escape(value, field)})`, comparator, options.prefix); |
|
case Op.between: |
|
case Op.notBetween: |
|
return this._joinKeyValue(key, `${this.escape(value[0], field)} AND ${this.escape(value[1], field)}`, comparator, options.prefix); |
|
case Op.raw: |
|
throw new Error("The `$raw` where property is no longer supported. Use `sequelize.literal` instead."); |
|
case Op.col: |
|
comparator = this.OperatorMap[Op.eq]; |
|
value = value.split("."); |
|
if (value.length > 2) { |
|
value = [ |
|
value.slice(0, -1).join("->"), |
|
value[value.length - 1] |
|
]; |
|
} |
|
return this._joinKeyValue(key, value.map((identifier) => this.quoteIdentifier(identifier)).join("."), comparator, options.prefix); |
|
case Op.startsWith: |
|
case Op.endsWith: |
|
case Op.substring: |
|
comparator = this.OperatorMap[Op.like]; |
|
if (value instanceof Utils.Literal) { |
|
value = value.val; |
|
} |
|
let pattern = `${value}%`; |
|
if (prop === Op.endsWith) |
|
pattern = `%${value}`; |
|
if (prop === Op.substring) |
|
pattern = `%${value}%`; |
|
return this._joinKeyValue(key, this.escape(pattern), comparator, options.prefix); |
|
} |
|
const escapeOptions = { |
|
acceptStrings: comparator.includes(this.OperatorMap[Op.like]) |
|
}; |
|
if (_.isPlainObject(value)) { |
|
if (value[Op.col]) { |
|
return this._joinKeyValue(key, this.whereItemQuery(null, value), comparator, options.prefix); |
|
} |
|
if (value[Op.any]) { |
|
escapeOptions.isList = true; |
|
return this._joinKeyValue(key, `(${this.escape(value[Op.any], field, escapeOptions)})`, `${comparator} ${this.OperatorMap[Op.any]}`, options.prefix); |
|
} |
|
if (value[Op.all]) { |
|
escapeOptions.isList = true; |
|
return this._joinKeyValue(key, `(${this.escape(value[Op.all], field, escapeOptions)})`, `${comparator} ${this.OperatorMap[Op.all]}`, options.prefix); |
|
} |
|
} |
|
if (value === null && comparator === this.OperatorMap[Op.eq]) { |
|
return this._joinKeyValue(key, this.escape(value, field, escapeOptions), this.OperatorMap[Op.is], options.prefix); |
|
} |
|
if (value === null && comparator === this.OperatorMap[Op.ne]) { |
|
return this._joinKeyValue(key, this.escape(value, field, escapeOptions), this.OperatorMap[Op.not], options.prefix); |
|
} |
|
return this._joinKeyValue(key, this.escape(value, field, escapeOptions), comparator, options.prefix); |
|
} |
|
getWhereConditions(smth, tableName, factory, options, prepend) { |
|
const where = {}; |
|
if (Array.isArray(tableName)) { |
|
tableName = tableName[0]; |
|
if (Array.isArray(tableName)) { |
|
tableName = tableName[1]; |
|
} |
|
} |
|
options = options || {}; |
|
if (prepend === void 0) { |
|
prepend = true; |
|
} |
|
if (smth && smth instanceof Utils.SequelizeMethod) { |
|
return this.handleSequelizeMethod(smth, tableName, factory, options, prepend); |
|
} |
|
if (_.isPlainObject(smth)) { |
|
return this.whereItemsQuery(smth, { |
|
model: factory, |
|
prefix: prepend && tableName, |
|
type: options.type |
|
}); |
|
} |
|
if (typeof smth === "number") { |
|
let primaryKeys = factory ? Object.keys(factory.primaryKeys) : []; |
|
if (primaryKeys.length > 0) { |
|
primaryKeys = primaryKeys[0]; |
|
} else { |
|
primaryKeys = "id"; |
|
} |
|
where[primaryKeys] = smth; |
|
return this.whereItemsQuery(where, { |
|
model: factory, |
|
prefix: prepend && tableName |
|
}); |
|
} |
|
if (typeof smth === "string") { |
|
return this.whereItemsQuery(smth, { |
|
model: factory, |
|
prefix: prepend && tableName |
|
}); |
|
} |
|
if (Buffer.isBuffer(smth)) { |
|
return this.escape(smth); |
|
} |
|
if (Array.isArray(smth)) { |
|
if (smth.length === 0 || smth.length > 0 && smth[0].length === 0) |
|
return "1=1"; |
|
if (Utils.canTreatArrayAsAnd(smth)) { |
|
const _smth = { [Op.and]: smth }; |
|
return this.getWhereConditions(_smth, tableName, factory, options, prepend); |
|
} |
|
throw new Error("Support for literal replacements in the `where` object has been removed."); |
|
} |
|
if (smth === null) { |
|
return this.whereItemsQuery(smth, { |
|
model: factory, |
|
prefix: prepend && tableName |
|
}); |
|
} |
|
return "1=1"; |
|
} |
|
parseConditionObject(conditions, path) { |
|
path = path || []; |
|
return _.reduce(conditions, (result, value, key) => { |
|
if (_.isObject(value)) { |
|
return result.concat(this.parseConditionObject(value, path.concat(key))); |
|
} |
|
result.push({ path: path.concat(key), value }); |
|
return result; |
|
}, []); |
|
} |
|
booleanValue(value) { |
|
return value; |
|
} |
|
authTestQuery() { |
|
return "SELECT 1+1 AS result"; |
|
} |
|
} |
|
Object.assign(QueryGenerator.prototype, require("./query-generator/operators")); |
|
Object.assign(QueryGenerator.prototype, require("./query-generator/transaction")); |
|
module.exports = QueryGenerator; |
|
//# sourceMappingURL=query-generator.js.map
|
|
|