knex/lib/dialects/redshift/query/redshift-querycompiler.js
2021-11-09 09:34:19 +01:00

101 lines
2.6 KiB
JavaScript

// Redshift Query Builder & Compiler
// ------
const QueryCompiler = require('../../../query/querycompiler');
const QueryCompiler_PG = require('../../postgres/query/pg-querycompiler');
const identity = require('lodash/identity');
class QueryCompiler_Redshift extends QueryCompiler_PG {
truncate() {
return `truncate ${this.tableName.toLowerCase()}`;
}
// Compiles an `insert` query, allowing for multiple
// inserts using a single query statement.
insert() {
const sql = QueryCompiler.prototype.insert.apply(this, arguments);
if (sql === '') return sql;
this._slightReturn();
return {
sql,
};
}
// Compiles an `update` query, warning on unsupported returning
update() {
const sql = QueryCompiler.prototype.update.apply(this, arguments);
this._slightReturn();
return {
sql,
};
}
// Compiles an `delete` query, warning on unsupported returning
del() {
const sql = QueryCompiler.prototype.del.apply(this, arguments);
this._slightReturn();
return {
sql,
};
}
// simple: if trying to return, warn
_slightReturn() {
if (this.single.isReturning) {
this.client.logger.warn(
'insert/update/delete returning is not supported by redshift dialect'
);
}
}
forUpdate() {
this.client.logger.warn('table lock is not supported by redshift dialect');
return '';
}
forShare() {
this.client.logger.warn(
'lock for share is not supported by redshift dialect'
);
return '';
}
forNoKeyUpdate() {
this.client.logger.warn('table lock is not supported by redshift dialect');
return '';
}
forKeyShare() {
this.client.logger.warn(
'lock for share is not supported by redshift dialect'
);
return '';
}
// Compiles a columnInfo query
columnInfo() {
const column = this.single.columnInfo;
let schema = this.single.schema;
// The user may have specified a custom wrapIdentifier function in the config. We
// need to run the identifiers through that function, but not format them as
// identifiers otherwise.
const table = this.client.customWrapIdentifier(this.single.table, identity);
if (schema) {
schema = this.client.customWrapIdentifier(schema, identity);
}
const sql =
'select * from information_schema.columns where table_name = ? and table_catalog = ?';
const bindings = [
table.toLowerCase(),
this.client.database().toLowerCase(),
];
return this._buildColumnInfoQuery(schema, sql, bindings, column);
}
}
module.exports = QueryCompiler_Redshift;