Update index.js

Author: dkarpman

lib/index.js

@@ -13,6 +13,8 @@ var PG = function(clientConstructor) {
   this.pools = pool;
   this.Connection = Connection;
   this.types = require('pg-types');
+  this.num_connections = 0;
+  this.max_connections = this.defaults.max_connections;
 };
 
 util.inherits(PG, EventEmitter);
@@ -35,28 +37,26 @@ PG.prototype.end = function() {
   });
 };
 
-
 PG.prototype.connect = function(config, callback) {
   if(typeof config == "function") {
     callback = config;
     config = null;
   }
+  if(this.max_connections && this.num_connections >= this.max_connections) {
+    return callback && callback(Error("Connection limit", this.max_connections, "reached"));
+  }
   var client = new this.Client(config);
+  // Max connections waiting to connect
+  this.num_connections++;
   client.connect(function(err) {
+    this.num_connections--;
     if(err) {
       return callback && callback(err);
     }
     return callback && callback(null, client, function() {
       client.end();
     });
-    // client.query()
   });
-  /* var pool = this.pools.getOrCreate(config);
-  pool.connect(callback);
-  if(!pool.listeners('error').length) {
-    //propagate errors up to pg object
-    pool.on('error', this.emit.bind(this, 'error'));
-  } */
 };
 
 // cancel the query runned by the given client