Support ending single pool

Call `pg.end(connString)` to do it.

Author: Suor

lib/index.js

@@ -17,27 +17,37 @@ var PG = function(clientConstructor) {
 
 util.inherits(PG, EventEmitter);
 
-PG.prototype.end = function() {
+PG.prototype.end = function(config) {
   var self = this;
-  var keys = Object.keys(self.pools.all);
+  var pool;
+  if (config) {
+    pool = self.pools.get(config);
+    if (pool) pool.end();
+  } else {
+    forEachOf(self.pools.all, function (pool, name, callback) {
+      pool.end(callback);
+    }, function () {
+      self.emit('end');
+    });
+  }
+};
+
+function forEachOf(object, iterator, callback) {
+  var keys = Object.keys(object);
   var count = keys.length;
-  if(count === 0) {
-    self.emit('end');
+  if (count === 0) {
+    callback();
   } else {
-    keys.forEach(function(key) {
-      var pool = self.pools.all[key];
-      delete self.pools.all[key];
-      pool.drain(function() {
-        pool.destroyAllNow(function() {
-          count--;
-          if(count === 0) {
-            self.emit('end');
-          }
-        });
+    keys.forEach(function (key) {
+      iterator(object[key], key, function () {
+        count--;
+        if (count === 0) {
+          callback();
+        }
       });
     });
   }
-};
+}
 
 
 PG.prototype.connect = function(config, callback) {

lib/pool.js

@@ -8,6 +8,10 @@ var pools = {
   all: {},
   //reference to the client constructor - can override in tests or for require('pg').native
   Client: require(__dirname + '/client'),
+  get: function(clientConfig) {
+    var name = JSON.stringify(clientConfig || {});
+    return pools.all[name];
+  },
   getOrCreate: function(clientConfig) {
     clientConfig = clientConfig || {};
     var name = JSON.stringify(clientConfig);
@@ -83,6 +87,12 @@ var pools = {
         });
       });
     };
+    pool.end = function(cb) {
+      delete pools.all[name];
+      pool.drain(function() {
+        pool.destroyAllNow(cb);
+      });
+    };
     return pool;
   }
 };