mirror of
https://github.com/seejohnrun/haste-server.git
synced 2024-11-21 20:31:20 +00:00
Switch to using pg.Pool
This commit is contained in:
parent
9a692ed652
commit
f147acb51c
1 changed files with 11 additions and 14 deletions
|
@ -1,14 +1,16 @@
|
|||
/*global require,module,process*/
|
||||
|
||||
var winston = require('winston');
|
||||
const {Client} = require('pg');
|
||||
const {Pool} = require('pg');
|
||||
|
||||
// create table entries (id serial primary key, key varchar(255) not null, value text not null, expiration int, unique(key));
|
||||
|
||||
// A postgres document store
|
||||
var PostgresDocumentStore = function (options) {
|
||||
this.expireJS = options.expire;
|
||||
this.connectionUrl = process.env.DATABASE_URL || options.connectionUrl;
|
||||
|
||||
const connectionString = process.env.DATABASE_URL || options.connectionUrl;
|
||||
this.pool = new Pool({connectionString});
|
||||
};
|
||||
|
||||
PostgresDocumentStore.prototype = {
|
||||
|
@ -64,20 +66,15 @@ PostgresDocumentStore.prototype = {
|
|||
|
||||
// A connection wrapper
|
||||
safeConnect: function (callback) {
|
||||
const client = new Client({connectionString: this.connectionUrl});
|
||||
const connectionPromise = client.connect();
|
||||
|
||||
const done = () => {};
|
||||
|
||||
connectionPromise.then(() => {
|
||||
callback(undefined, client, done);
|
||||
}).catch((error) => {
|
||||
console.log(error);
|
||||
this.pool.connect((err, client, done) => {
|
||||
if (err) {
|
||||
winston.error('error connecting to postgres', {error});
|
||||
callback(error);
|
||||
} else {
|
||||
callback(undefined, client, done);
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
};
|
||||
|
||||
module.exports = PostgresDocumentStore;
|
||||
|
|
Loading…
Reference in a new issue