/* eslint max-len: 0 */ // Redshift Table Builder & Compiler // ------- const inherits = require('inherits'); const { has } = require('lodash'); const TableCompiler_PG = require('../../postgres/schema/tablecompiler'); function TableCompiler_Redshift() { TableCompiler_PG.apply(this, arguments); } inherits(TableCompiler_Redshift, TableCompiler_PG); TableCompiler_Redshift.prototype.index = function( columns, indexName, indexType ) { this.client.logger.warn('Redshift does not support the creation of indexes.'); }; TableCompiler_Redshift.prototype.dropIndex = function(columns, indexName) { this.client.logger.warn('Redshift does not support the deletion of indexes.'); }; // TODO: have to disable setting not null on columns that already exist... // Adds the "create" query to the query sequence. TableCompiler_Redshift.prototype.createQuery = function(columns, ifNot) { const createStatement = ifNot ? 'create table if not exists ' : 'create table '; let sql = createStatement + this.tableName() + ' (' + columns.sql.join(', ') + ')'; if (this.single.inherits) sql += ` like (${this.formatter.wrap(this.single.inherits)})`; this.pushQuery({ sql, bindings: columns.bindings, }); const hasComment = has(this.single, 'comment'); if (hasComment) this.comment(this.single.comment); }; TableCompiler_Redshift.prototype.primary = function(columns, constraintName) { const self = this; constraintName = constraintName ? self.formatter.wrap(constraintName) : self.formatter.wrap(`${this.tableNameRaw}_pkey`); if (columns.constructor !== Array) { columns = [columns]; } const thiscolumns = self.grouped.columns; if (thiscolumns) { for (let i = 0; i < columns.length; i++) { let exists = thiscolumns.find( (tcb) => tcb.grouping === 'columns' && tcb.builder && tcb.builder._method === 'add' && tcb.builder._args && tcb.builder._args.indexOf(columns[i]) > -1 ); if (exists) { exists = exists.builder; } const nullable = !( exists && exists._modifiers && exists._modifiers['nullable'] && exists._modifiers['nullable'][0] === false ); if (nullable) { if (exists) { return this.client.logger.warn( 'Redshift does not allow primary keys to contain nullable columns.' ); } else { return this.client.logger.warn( 'Redshift does not allow primary keys to contain nonexistent columns.' ); } } } } return self.pushQuery( `alter table ${self.tableName()} add constraint ${constraintName} primary key (${self.formatter.columnize( columns )})` ); }; // Compiles column add. Redshift can only add one column per ALTER TABLE, so core addColumns doesn't work. #2545 TableCompiler_Redshift.prototype.addColumns = function( columns, prefix, colCompilers ) { if (prefix === this.alterColumnsPrefix) { TableCompiler_PG.prototype.addColumns.call( this, columns, prefix, colCompilers ); } else { prefix = prefix || this.addColumnsPrefix; colCompilers = colCompilers || this.getColumns(); for (const col of colCompilers) { const quotedTableName = this.tableName(); const colCompiled = col.compileColumn(); this.pushQuery({ sql: `alter table ${quotedTableName} ${prefix}${colCompiled}`, bindings: [], }); } } }; module.exports = TableCompiler_Redshift;