emit event when all pool are actually destroyed

Author: Ricky Ng-Adam

lib/index.js

@@ -19,15 +19,22 @@ util.inherits(PG, EventEmitter);
 
 PG.prototype.end = function() {
   var self = this;
-  Object.keys(self.pools.all).forEach(function(key) {
+  var keys = Object.keys(self.pools.all);
+  var count = keys.length;
+  keys.forEach(function(key) {
     var pool = self.pools.all[key];
     delete self.pools.all[key];
     pool.drain(function() {
       pool.destroyAllNow();
+      count--;
+      if(count === 0) {
+        self.emit('ended');
+      }
     });
   });
 };
 
+
 PG.prototype.connect = function(config, callback) {
   if(typeof config == "function") {
     callback = config;

test/integration/connection-pool/ending-pool-tests.js

@@ -4,6 +4,11 @@ var called = false;
 test('disconnects', function() {
   var sink = new helper.Sink(4, function() {
     called = true;
+    var eventSink = new helper.Sink(1, function() {});
+    helper.pg.on('ended', function() {
+      eventSink.add();
+    });
+
     //this should exit the process, killing each connection pool
     helper.pg.end();
   });