238 lines
8.2 KiB
JavaScript
238 lines
8.2 KiB
JavaScript
"use strict";
|
|
const AbstractQuery = require("../abstract/query");
|
|
const sequelizeErrors = require("../../errors");
|
|
const _ = require("lodash");
|
|
const { logger } = require("../../utils/logger");
|
|
const ER_DUP_ENTRY = 1062;
|
|
const ER_DEADLOCK = 1213;
|
|
const ER_ROW_IS_REFERENCED = 1451;
|
|
const ER_NO_REFERENCED_ROW = 1452;
|
|
const debug = logger.debugContext("sql:snowflake");
|
|
class Query extends AbstractQuery {
|
|
static formatBindParameters(sql, values, dialect) {
|
|
const bindParam = [];
|
|
const replacementFunc = (_match, key, values_) => {
|
|
if (values_[key] !== void 0) {
|
|
bindParam.push(values_[key]);
|
|
return "?";
|
|
}
|
|
return void 0;
|
|
};
|
|
sql = AbstractQuery.formatBindParameters(sql, values, dialect, replacementFunc)[0];
|
|
return [sql, bindParam.length > 0 ? bindParam : void 0];
|
|
}
|
|
async run(sql, parameters) {
|
|
this.sql = sql;
|
|
const { connection, options } = this;
|
|
const showWarnings = this.sequelize.options.showWarnings || options.showWarnings;
|
|
const complete = this._logQuery(sql, debug, parameters);
|
|
if (parameters) {
|
|
debug("parameters(%j)", parameters);
|
|
}
|
|
let results;
|
|
try {
|
|
results = await new Promise((resolve, reject) => {
|
|
connection.execute({
|
|
sqlText: sql,
|
|
binds: parameters,
|
|
complete(err, _stmt, rows) {
|
|
if (err) {
|
|
reject(err);
|
|
} else {
|
|
resolve(rows);
|
|
}
|
|
}
|
|
});
|
|
});
|
|
} catch (error) {
|
|
if (options.transaction && error.errno === ER_DEADLOCK) {
|
|
try {
|
|
await options.transaction.rollback();
|
|
} catch (error_) {
|
|
}
|
|
options.transaction.finished = "rollback";
|
|
}
|
|
error.sql = sql;
|
|
error.parameters = parameters;
|
|
throw this.formatError(error);
|
|
} finally {
|
|
complete();
|
|
}
|
|
if (showWarnings && results && results.warningStatus > 0) {
|
|
await this.logWarnings(results);
|
|
}
|
|
return this.formatResults(results);
|
|
}
|
|
formatResults(data) {
|
|
let result = this.instance;
|
|
if (this.isInsertQuery(data)) {
|
|
this.handleInsertQuery(data);
|
|
if (!this.instance) {
|
|
if (data.constructor.name === "ResultSetHeader" && this.model && this.model.autoIncrementAttribute && this.model.autoIncrementAttribute === this.model.primaryKeyAttribute && this.model.rawAttributes[this.model.primaryKeyAttribute]) {
|
|
const startId = data[this.getInsertIdField()];
|
|
result = [];
|
|
for (let i = startId; i < startId + data.affectedRows; i++) {
|
|
result.push({ [this.model.rawAttributes[this.model.primaryKeyAttribute].field]: i });
|
|
}
|
|
} else {
|
|
result = data[this.getInsertIdField()];
|
|
}
|
|
}
|
|
}
|
|
if (this.isSelectQuery()) {
|
|
if (this.options.raw === false && this.sequelize.options.quoteIdentifiers === false) {
|
|
const sfAttrMap = _.reduce(this.model.rawAttributes, (m, v, k) => {
|
|
m[k.toUpperCase()] = k;
|
|
return m;
|
|
}, {});
|
|
data = data.map((data2) => _.reduce(data2, (prev, value, key) => {
|
|
if (value !== void 0 && sfAttrMap[key]) {
|
|
prev[sfAttrMap[key]] = value;
|
|
delete prev[key];
|
|
}
|
|
return prev;
|
|
}, data2));
|
|
}
|
|
this.options.fieldMap = _.mapKeys(this.options.fieldMap, (v, k) => {
|
|
return k.toUpperCase();
|
|
});
|
|
return this.handleSelectQuery(data);
|
|
}
|
|
if (this.isShowTablesQuery()) {
|
|
return this.handleShowTablesQuery(data);
|
|
}
|
|
if (this.isDescribeQuery()) {
|
|
result = {};
|
|
for (const _result of data) {
|
|
result[_result.Field] = {
|
|
type: _result.Type.toUpperCase(),
|
|
allowNull: _result.Null === "YES",
|
|
defaultValue: _result.Default,
|
|
primaryKey: _result.Key === "PRI",
|
|
autoIncrement: Object.prototype.hasOwnProperty.call(_result, "Extra") && _result.Extra.toLowerCase() === "auto_increment",
|
|
comment: _result.Comment ? _result.Comment : null
|
|
};
|
|
}
|
|
return result;
|
|
}
|
|
if (this.isShowIndexesQuery()) {
|
|
return this.handleShowIndexesQuery(data);
|
|
}
|
|
if (this.isCallQuery()) {
|
|
return data[0];
|
|
}
|
|
if (this.isBulkUpdateQuery() || this.isBulkDeleteQuery()) {
|
|
return data[0]["number of rows updated"];
|
|
}
|
|
if (this.isVersionQuery()) {
|
|
return data[0].version;
|
|
}
|
|
if (this.isForeignKeysQuery()) {
|
|
return data;
|
|
}
|
|
if (this.isUpsertQuery()) {
|
|
return [result, data.affectedRows === 1];
|
|
}
|
|
if (this.isInsertQuery() || this.isUpdateQuery()) {
|
|
return [result, data.affectedRows];
|
|
}
|
|
if (this.isShowConstraintsQuery()) {
|
|
return data;
|
|
}
|
|
if (this.isRawQuery()) {
|
|
return [data, data];
|
|
}
|
|
return result;
|
|
}
|
|
async logWarnings(results) {
|
|
const warningResults = await this.run("SHOW WARNINGS");
|
|
const warningMessage = `Snowflake Warnings (${this.connection.uuid || "default"}): `;
|
|
const messages = [];
|
|
for (const _warningRow of warningResults) {
|
|
if (_warningRow === void 0 || typeof _warningRow[Symbol.iterator] !== "function") {
|
|
continue;
|
|
}
|
|
for (const _warningResult of _warningRow) {
|
|
if (Object.prototype.hasOwnProperty.call(_warningResult, "Message")) {
|
|
messages.push(_warningResult.Message);
|
|
} else {
|
|
for (const _objectKey of _warningResult.keys()) {
|
|
messages.push([_objectKey, _warningResult[_objectKey]].join(": "));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
this.sequelize.log(warningMessage + messages.join("; "), this.options);
|
|
return results;
|
|
}
|
|
formatError(err) {
|
|
const errCode = err.errno || err.code;
|
|
switch (errCode) {
|
|
case ER_DUP_ENTRY: {
|
|
const match = err.message.match(/Duplicate entry '([\s\S]*)' for key '?((.|\s)*?)'?$/);
|
|
let fields = {};
|
|
let message = "Validation error";
|
|
const values = match ? match[1].split("-") : void 0;
|
|
const fieldKey = match ? match[2] : void 0;
|
|
const fieldVal = match ? match[1] : void 0;
|
|
const uniqueKey = this.model && this.model.uniqueKeys[fieldKey];
|
|
if (uniqueKey) {
|
|
if (uniqueKey.msg)
|
|
message = uniqueKey.msg;
|
|
fields = _.zipObject(uniqueKey.fields, values);
|
|
} else {
|
|
fields[fieldKey] = fieldVal;
|
|
}
|
|
const errors = [];
|
|
_.forOwn(fields, (value, field) => {
|
|
errors.push(new sequelizeErrors.ValidationErrorItem(this.getUniqueConstraintErrorMessage(field), "unique violation", field, value, this.instance, "not_unique"));
|
|
});
|
|
return new sequelizeErrors.UniqueConstraintError({ message, errors, parent: err, fields });
|
|
}
|
|
case ER_ROW_IS_REFERENCED:
|
|
case ER_NO_REFERENCED_ROW: {
|
|
const match = err.message.match(/CONSTRAINT ([`"])(.*)\1 FOREIGN KEY \(\1(.*)\1\) REFERENCES \1(.*)\1 \(\1(.*)\1\)/);
|
|
const quoteChar = match ? match[1] : "`";
|
|
const fields = match ? match[3].split(new RegExp(`${quoteChar}, *${quoteChar}`)) : void 0;
|
|
return new sequelizeErrors.ForeignKeyConstraintError({
|
|
reltype: String(errCode) === String(ER_ROW_IS_REFERENCED) ? "parent" : "child",
|
|
table: match ? match[4] : void 0,
|
|
fields,
|
|
value: fields && fields.length && this.instance && this.instance[fields[0]] || void 0,
|
|
index: match ? match[2] : void 0,
|
|
parent: err
|
|
});
|
|
}
|
|
default:
|
|
return new sequelizeErrors.DatabaseError(err);
|
|
}
|
|
}
|
|
handleShowIndexesQuery(data) {
|
|
data = data.reduce((acc, item) => {
|
|
if (!(item.Key_name in acc)) {
|
|
acc[item.Key_name] = item;
|
|
item.fields = [];
|
|
}
|
|
acc[item.Key_name].fields[item.Seq_in_index - 1] = {
|
|
attribute: item.Column_name,
|
|
length: item.Sub_part || void 0,
|
|
order: item.Collation === "A" ? "ASC" : void 0
|
|
};
|
|
delete item.column_name;
|
|
return acc;
|
|
}, {});
|
|
return _.map(data, (item) => ({
|
|
primary: item.Key_name === "PRIMARY",
|
|
fields: item.fields,
|
|
name: item.Key_name,
|
|
tableName: item.Table,
|
|
unique: item.Non_unique !== 1,
|
|
type: item.Index_type
|
|
}));
|
|
}
|
|
}
|
|
module.exports = Query;
|
|
module.exports.Query = Query;
|
|
module.exports.default = Query;
|
|
//# sourceMappingURL=query.js.map
|