var EventEmitter = require('events').EventEmitter; var util = require('util'); var Client = require('./client'); var defaults = require('./defaults'); var Connection = require('./connection'); var ConnectionParameters = require('./connection-parameters'); var Pool = require('pg-pool'); var PG = function(clientConstructor) { EventEmitter.call(this); this.defaults = defaults; this.Client = clientConstructor; this.Query = this.Client.Query; this.Pool = Pool; this.pools = []; this.Connection = Connection; this.types = require('pg-types'); }; util.inherits(PG, EventEmitter); PG.prototype.end = function() { var self = this; var keys = Object.keys(this.pools); var count = keys.length; if(count === 0) { self.emit('end'); } else { keys.forEach(function(key) { var pool = self.pools[key]; delete self.pools[key]; pool.pool.drain(function() { pool.pool.destroyAllNow(function() { count--; if(count === 0) { self.emit('end'); } }); }); }); } }; PG.prototype.connect = function(config, callback) { if(typeof config == "function") { callback = config; config = null; } var poolName = JSON.stringify(config || {}); if (typeof config == 'string') { config = new ConnectionParameters(config); } config = config || {}; //for backwards compatibility config.max = config.max || config.poolSize || defaults.poolSize; config.idleTimeoutMillis = config.idleTimeoutMillis || config.poolIdleTimeout || defaults.poolIdleTimeout; config.log = config.log || config.poolLog || defaults.poolLog; this.pools[poolName] = this.pools[poolName] || new Pool(config, this.Client); var pool = this.pools[poolName]; pool.connect(callback); if(!pool.listeners('error').length) { //propagate errors up to pg object pool.on('error', function(e) { this.emit('error', e, e.client); }.bind(this)); } }; // cancel the query runned by the given client PG.prototype.cancel = function(config, client, query) { if(client.native) { return client.cancel(query); } var c = config; //allow for no config to be passed if(typeof c === 'function') { c = defaults; } var cancellingClient = new this.Client(c); cancellingClient.cancel(client, query); }; if(typeof process.env.NODE_PG_FORCE_NATIVE != 'undefined') { module.exports = new PG(require('./native')); } else { module.exports = new PG(Client); //lazy require native module...the native module may not have installed module.exports.__defineGetter__("native", function() { delete module.exports.native; var native = null; try { native = new PG(require('./native')); } catch (err) { if (err.code !== 'MODULE_NOT_FOUND') { throw err; } console.error(err.message); } module.exports.native = native; return native; }); }