Add Connection Pooling

billwashere · billwashere · commit d4ab7ccea506 · 2018-03-17T17:02:26.000+11:00
diff --git a/index.js b/index.js
@@ -10,13 +10,16 @@ function PostgresDB(options) {
   this.closed = false;
 
   this.pg_config = options;
+  this.pool = new pg.Pool(this.pg_config)
 };
 module.exports = PostgresDB;
 
 PostgresDB.prototype = Object.create(DB.prototype);
 
 PostgresDB.prototype.close = function(callback) {
   this.closed = true;
+  this.pool.end()
+  
   if (callback) callback();
 };
 
@@ -39,7 +42,7 @@ PostgresDB.prototype.commit = function(collection, id, op, snapshot, options, ca
    * }
    * snapshot: PostgresSnapshot
    */
-  pg.connect(this.pg_config, function(err, client, done) {
+  this.pool.connect(function(err, client, done) {
     if (err) {
       done(client);
       callback(err);
@@ -123,7 +126,7 @@ PostgresDB.prototype.commit = function(collection, id, op, snapshot, options, ca
 // snapshot). A snapshot with a version of zero is returned if the docuemnt
 // has never been created in the database.
 PostgresDB.prototype.getSnapshot = function(collection, id, fields, options, callback) {
-  pg.connect(this.pg_config, function(err, client, done) {
+  this.pool.connect(function(err, client, done) {
     if (err) {
       done(client);
       callback(err);
@@ -173,7 +176,7 @@ PostgresDB.prototype.getSnapshot = function(collection, id, fields, options, cal
 //
 // Callback should be called as callback(error, [list of ops]);
 PostgresDB.prototype.getOps = function(collection, id, from, to, options, callback) {
-  pg.connect(this.pg_config, function(err, client, done) {
+  this.pool.connect(function(err, client, done) {
     if (err) {
       done(client);
       callback(err);
diff --git a/package.json b/package.json
@@ -11,7 +11,7 @@
   "keywords": ["sharedb", "sharejs", "share", "postgres"],
   "repository": "share/sharedb-postgres",
   "dependencies": {
-    "pg": "^4.5.1",
+    "pg": "^7.4.1",
     "sharedb": "^1.0.0-beta.7"
   }
-}
+}
