mirror of
https://github.com/knex/knex.git
synced 2025-07-03 23:19:26 +00:00
90 lines
3.4 KiB
JavaScript
90 lines
3.4 KiB
JavaScript
// PostgreSQL Table Builder & Compiler
|
|
// -------
|
|
module.exports = function(client) {
|
|
|
|
var _ = require('lodash');
|
|
var inherits = require('inherits');
|
|
var Schema = require('../../../schema');
|
|
|
|
// Table
|
|
// ------
|
|
|
|
function TableBuilder_PG() {
|
|
this.client = client;
|
|
Schema.TableBuilder.apply(this, arguments);
|
|
}
|
|
inherits(TableBuilder_PG, Schema.TableBuilder);
|
|
|
|
function TableCompiler_PG() {
|
|
this.client = client;
|
|
this.Formatter = client.Formatter;
|
|
Schema.TableCompiler.apply(this, arguments);
|
|
}
|
|
inherits(TableCompiler_PG, Schema.TableCompiler);
|
|
|
|
// Compile a rename column command.
|
|
TableCompiler_PG.prototype.renameColumn = function(from, to) {
|
|
return this.pushQuery({
|
|
sql: 'alter table ' + this.tableName() + ' rename '+ this.formatter.wrap(from) + ' to ' + this.formatter.wrap(to)
|
|
});
|
|
};
|
|
|
|
TableCompiler_PG.prototype.compileAdd = function(builder) {
|
|
var table = this.formatter.wrap(builder);
|
|
var columns = this.prefixArray('add column', this.getColumns(builder));
|
|
return this.pushQuery({
|
|
sql: 'alter table ' + table + ' ' + columns.join(', ')
|
|
});
|
|
};
|
|
|
|
// Adds the "create" query to the query sequence.
|
|
TableCompiler_PG.prototype.createQuery = function(columns) {
|
|
this.pushQuery({
|
|
sql: (this._createIfNot ? 'create table if not exists ' : 'create table ') + this.tableName() + ' (' + columns.sql.join(', ') + ')',
|
|
bindings: columns.bindings
|
|
});
|
|
var hasComment = _.has(this.single, 'comment');
|
|
if (hasComment) this.comment(this.single.comment);
|
|
};
|
|
|
|
// Compiles the comment on the table.
|
|
TableCompiler_PG.prototype.comment = function(comment) {
|
|
this.pushQuery('comment on table ' + this.tableName() + ' is ' + "'" + (this.single.comment || '') + "'");
|
|
};
|
|
|
|
// Indexes:
|
|
// -------
|
|
|
|
TableCompiler_PG.prototype.primary = function(columns) {
|
|
this.pushQuery('alter table ' + this.tableName() + " add primary key (" + this.formatter.columnize(columns) + ")");
|
|
};
|
|
TableCompiler_PG.prototype.unique = function(columns, indexName) {
|
|
indexName = indexName || this._indexCommand('unique', this.tableNameRaw, columns);
|
|
this.pushQuery('alter table ' + this.tableName() + ' add constraint ' + indexName +
|
|
' unique (' + this.formatter.columnize(columns) + ')');
|
|
};
|
|
TableCompiler_PG.prototype.index = function(columns, indexName) {
|
|
indexName = indexName || this._indexCommand('index', this.tableNameRaw, columns);
|
|
this.pushQuery('create index ' + indexName + ' on ' + this.tableName() +
|
|
' (' + this.formatter.columnize(columns) + ')');
|
|
};
|
|
TableCompiler_PG.prototype.dropPrimary = function() {
|
|
this.pushQuery('alter table ' + this.tableName() + " drop constraint " + this.tableNameRaw + "_pkey");
|
|
};
|
|
TableCompiler_PG.prototype.dropIndex = function(columns, indexName) {
|
|
indexName = indexName || this._indexCommand('index', this.tableNameRaw, columns);
|
|
this.pushQuery('drop index ' + indexName);
|
|
};
|
|
TableCompiler_PG.prototype.dropUnique = function(columns, indexName) {
|
|
indexName = indexName || this._indexCommand('unique', this.tableNameRaw, columns);
|
|
this.pushQuery('alter table ' + this.tableName() + ' drop constraint ' + indexName);
|
|
};
|
|
TableCompiler_PG.prototype.dropForeign = function(columns, indexName) {
|
|
indexName = indexName || this._indexCommand('foreign', this.tableNameRaw, columns);
|
|
this.pushQuery('alter table ' + this.tableName() + ' drop constraint ' + indexName);
|
|
};
|
|
|
|
client.TableBuilder = TableBuilder_PG;
|
|
client.TableCompiler = TableCompiler_PG;
|
|
|
|
}; |