98 lines
3.0 KiB
JavaScript
Raw Normal View History

// PostgreSQL Schema Compiler
// -------
'use strict';
2015-05-09 13:58:18 -04:00
var inherits = require('inherits');
var SchemaCompiler = require('../../../schema/compiler');
function SchemaCompiler_PG() {
SchemaCompiler.apply(this, arguments);
}
inherits(SchemaCompiler_PG, SchemaCompiler);
// Check whether the current table
2015-05-09 13:58:18 -04:00
SchemaCompiler_PG.prototype.hasTable = function (tableName) {
2015-08-09 22:24:55 -03:00
var sql = 'select * from information_schema.tables where table_name = ?';
var bindings = [tableName];
if (this.schema) {
sql += ' and table_schema = ?';
bindings.push(this.schema);
} else {
sql += ' and table_schema = current_schema';
}
2014-04-16 01:23:50 -04:00
this.pushQuery({
2015-08-09 22:24:55 -03:00
sql: sql,
bindings: bindings,
2015-05-09 13:58:18 -04:00
output: function output(resp) {
return resp.rows.length > 0;
}
});
};
// Compile the query to determine if a column exists in a table.
2015-05-09 13:58:18 -04:00
SchemaCompiler_PG.prototype.hasColumn = function (tableName, columnName) {
2015-08-09 22:24:55 -03:00
var sql = 'select * from information_schema.columns where table_name = ? and column_name = ?';
var bindings = [tableName, columnName];
if (this.schema) {
sql += ' and table_schema = ?';
bindings.push(this.schema);
} else {
sql += ' and table_schema = current_schema';
}
2014-04-16 01:23:50 -04:00
this.pushQuery({
2015-08-09 22:24:55 -03:00
sql: sql,
bindings: bindings,
2015-05-09 13:58:18 -04:00
output: function output(resp) {
return resp.rows.length > 0;
}
});
};
2015-08-09 22:24:55 -03:00
SchemaCompiler_PG.prototype.qualifiedTableName = function (tableName) {
var name = this.schema ? this.schema + '.' + tableName : tableName;
return this.formatter.wrap(name);
};
// Compile a rename table command.
2015-05-09 13:58:18 -04:00
SchemaCompiler_PG.prototype.renameTable = function (from, to) {
2015-08-09 22:24:55 -03:00
this.pushQuery('alter table ' + this.qualifiedTableName(from) + ' rename to ' + this.qualifiedTableName(to));
};
2015-05-09 13:58:18 -04:00
SchemaCompiler_PG.prototype.createSchema = function (schemaName) {
this.pushQuery('create schema ' + this.formatter.wrap(schemaName));
};
2015-05-09 13:58:18 -04:00
SchemaCompiler_PG.prototype.createSchemaIfNotExists = function (schemaName) {
this.pushQuery('create schema if not exists ' + this.formatter.wrap(schemaName));
};
2015-05-09 13:58:18 -04:00
SchemaCompiler_PG.prototype.dropSchema = function (schemaName) {
this.pushQuery('drop schema ' + this.formatter.wrap(schemaName));
};
2015-05-09 13:58:18 -04:00
SchemaCompiler_PG.prototype.dropSchemaIfExists = function (schemaName) {
this.pushQuery('drop schema if exists ' + this.formatter.wrap(schemaName));
};
2015-05-09 13:58:18 -04:00
SchemaCompiler_PG.prototype.dropExtension = function (extensionName) {
this.pushQuery('drop extension ' + this.formatter.wrap(extensionName));
};
2015-05-09 13:58:18 -04:00
SchemaCompiler_PG.prototype.dropExtensionIfExists = function (extensionName) {
this.pushQuery('drop extension if exists ' + this.formatter.wrap(extensionName));
};
2015-05-09 13:58:18 -04:00
SchemaCompiler_PG.prototype.createExtension = function (extensionName) {
this.pushQuery('create extension ' + this.formatter.wrap(extensionName));
};
2015-05-09 13:58:18 -04:00
SchemaCompiler_PG.prototype.createExtensionIfNotExists = function (extensionName) {
this.pushQuery('create extension if not exists ' + this.formatter.wrap(extensionName));
};
2015-05-09 13:58:18 -04:00
module.exports = SchemaCompiler_PG;