knex/lib/dialects/postgres/query.js

// PostgreSQL Query Builder & Compiler
// ------
module.exports = function(client) {

var inherits = require('inherits');
var _ = require('lodash');

var QueryBuilder  = require('../../query/builder');
var QueryCompiler = require('../../query/compiler');

// Query Builder
// ------

function QueryBuilder_PG() {
  this.client = client;
  QueryBuilder.apply(this, arguments);
}
inherits(QueryBuilder_PG, QueryBuilder);

// Query Compiler
// ------

function QueryCompiler_PG() {
  this.formatter = new client.Formatter();
  QueryCompiler.apply(this, arguments);
}
inherits(QueryCompiler_PG, QueryCompiler);

// Compiles a truncate query.
QueryCompiler_PG.prototype.truncate = function() {
  return 'truncate ' + this.tableName + ' restart identity';
};

// Compiles an `insert` query, allowing for multiple
// inserts using a single query statement.
QueryCompiler_PG.prototype.insert = function() {
  var sql = 'insert into ' + this.tableName + ' ';
  if (_.isEmpty(this.single.insert)) {
    sql += 'default values';
  } else {
    var columns;
    var insertData = this._prepInsert(this.single.insert);
    sql += '(' + this.formatter.columnize(insertData.columns) + ') values (' +
      _.map(insertData.values, this.formatter.parameterize, this.formatter).join('), (') + ')';
  }
  var returning = this.single.returning;
  return sql + (returning ? ' ' + returning : '');
};

// Compiles an `update` query, allowing for a return value.
QueryCompiler_PG.prototype.update = function() {
  var wheres     = this.where();
  var updateData = this._prepUpdate(this.single.update);
  var returning  = this.single.returning;
  return 'update ' + this.tableName + ' set ' + updateData.columns + ' ' + wheres.sql +
    (returning ? ' ' + returning : '');
};

client.QueryBuilder = QueryBuilder_PG;
client.QueryCompiler = QueryCompiler_PG;

};