clients/server/postgres.js

PostgreSQL

Other dependencies, including the pg library, which needs to be added as a dependency to the project using this database.

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

All other local project modules needed in this scope.

var ServerBase = require('./base').ServerBase; var Helpers = require('../../lib/helpers').Helpers; var Promise = require('../../lib/promise').Promise; var grammar = require('./postgres/grammar').grammar; var schemaGrammar = require('./postgres/schemagrammar').schemaGrammar;

Constructor for the PostgreSQL Client

exports.Client = ServerBase.extend({ dialect: 'postgresql',

Attach the appropriate grammar definitions onto the current client.

attachGrammars: function() { this.grammar = grammar; this.schemaGrammar = schemaGrammar; },

Runs the query on the specified connection, providing the bindings and any other necessary prep work.

runQuery: function(connection, sql, bindings, builder) { if (!connection) throw new Error('No database connection exists for the query'); var questionCount = 0; sql = sql.replace(/\?/g, function() { questionCount++; return '$' + questionCount; }); if (builder && builder.flags.options) sql = _.extend({text: sql}, builder.flags.options); return Promise.promisify(connection.query, connection)(sql, bindings); },

Get a raw connection, called by the pool whenever a new connection needs to be added to the pool.

getRawConnection: function(callback) { var connection = new pg.Client(this.connectionSettings); return Promise.promisify(connection.connect, connection)().bind(this).tap(function() { if (!this.version) return this.checkVersion(connection); }).bind().yield(connection); },

Used to explicitly close a connection, called internally by the pool when a connection times out or the pool is shutdown.

destroyRawConnection: function(connection) { connection.end(); },

In PostgreSQL, we need to do a version check to do some feature checking on the database.

checkVersion: function(connection) { var instance = this; this.runQuery(connection, 'select version();').then(function(resp) { instance.version = /^PostgreSQL (.*?) /.exec(resp.rows[0].version)[1]; }); } });