|
3 | 3 |
|
4 | 4 | var pg = require('pg'); |
5 | 5 |
|
6 | | -exports["connect'"] = function (conString) { |
7 | | - return function(success, error) { |
8 | | - var client = new pg.Client(conString); |
9 | | - client.connect(function(err) { |
10 | | - if (err) { |
11 | | - error(err); |
12 | | - } else { |
13 | | - success(client); |
14 | | - } |
15 | | - }) |
16 | | - return client; |
| 6 | +exports.mkPool = function (conInfo) { |
| 7 | + return function () { |
| 8 | + return new pg.Pool(conInfo); |
17 | 9 | }; |
18 | 10 | } |
19 | 11 |
|
20 | | -exports._withClient = function (conString, cb) { |
21 | | - return function(success, error) { |
22 | | - pg.connect(conString, function(err, client, done) { |
| 12 | +exports["connect'"] = function (pool) { |
| 13 | + return function(error, success) { |
| 14 | + pool.connect(function(err, client) { |
23 | 15 | if (err) { |
24 | | - done(true); |
25 | | - return error(err); |
26 | | - } |
27 | | - cb(client)(function(v) { |
28 | | - done(); |
29 | | - success(v); |
30 | | - }, function(err) { |
31 | | - done(); |
32 | 16 | error(err); |
33 | | - }) |
| 17 | + } else { |
| 18 | + success(client); |
| 19 | + } |
34 | 20 | }); |
| 21 | + return function(cancelError, onCancelerError, onCancelerSuccess) { |
| 22 | + onCancelerSuccess(); |
| 23 | + }; |
35 | 24 | }; |
36 | 25 | } |
37 | 26 |
|
38 | | -exports.runQuery_ = function (queryStr) { |
| 27 | +exports.runQuery_ = function(queryStr) { |
39 | 28 | return function(client) { |
40 | | - return function(success, error) { |
| 29 | + return function(error, success) { |
41 | 30 | client.query(queryStr, function(err, result) { |
42 | 31 | if (err) { |
43 | 32 | error(err); |
44 | 33 | } else { |
45 | 34 | success(result.rows); |
46 | 35 | } |
47 | | - }) |
| 36 | + }); |
| 37 | + return function(cancelError, onCancelerError, onCancelerSuccess) { |
| 38 | + onCancelerSuccess(); |
| 39 | + }; |
48 | 40 | }; |
49 | 41 | }; |
50 | 42 | } |
51 | 43 |
|
52 | | -exports.runQuery = function (queryStr) { |
| 44 | +exports.runQuery = function(queryStr) { |
53 | 45 | return function(params) { |
54 | 46 | return function(client) { |
55 | | - return function(success, error) { |
| 47 | + return function(error, success) { |
56 | 48 | client.query(queryStr, params, function(err, result) { |
57 | 49 | if (err) return error(err); |
58 | 50 | success(result.rows); |
59 | | - }) |
| 51 | + }); |
| 52 | + return function(cancelError, onCancelerError, onCancelerSuccess) { |
| 53 | + onCancelerSuccess(); |
| 54 | + }; |
60 | 55 | }; |
61 | 56 | }; |
62 | 57 | }; |
63 | 58 | } |
64 | 59 |
|
65 | | -exports.runQueryValue_ = function (queryStr) { |
| 60 | +exports.runQueryValue_ = function(queryStr) { |
66 | 61 | return function(client) { |
67 | | - return function(success, error) { |
| 62 | + return function(error, success) { |
68 | 63 | client.query(queryStr, function(err, result) { |
69 | 64 | if (err) return error(err); |
70 | 65 | success(result.rows.length > 0 ? result.rows[0][result.fields[0].name] : undefined); |
71 | | - }) |
| 66 | + }); |
| 67 | + return function(cancelError, onCancelerError, onCancelerSuccess) { |
| 68 | + onCancelerSuccess(); |
| 69 | + }; |
72 | 70 | }; |
73 | 71 | }; |
74 | 72 | } |
75 | 73 |
|
76 | | -exports.runQueryValue = function (queryStr) { |
| 74 | +exports.runQueryValue = function(queryStr) { |
77 | 75 | return function(params) { |
78 | 76 | return function(client) { |
79 | | - return function(success, error) { |
| 77 | + return function(error, success) { |
80 | 78 | client.query(queryStr, params, function(err, result) { |
81 | 79 | if (err) return error(err); |
82 | 80 | success(result.rows.length > 0 ? result.rows[0][result.fields[0].name] : undefined); |
83 | | - }) |
| 81 | + }); |
| 82 | + return function(cancelError, onCancelerError, onCancelerSuccess) { |
| 83 | + onCancelerSuccess(); |
| 84 | + }; |
84 | 85 | }; |
85 | 86 | }; |
86 | 87 | }; |
87 | 88 | } |
88 | 89 |
|
89 | | -exports.end = function (client) { |
90 | | - return function() { |
91 | | - client.end(); |
| 90 | +exports.release = function (client) { |
| 91 | + return function () { |
| 92 | + client.release(); |
92 | 93 | }; |
93 | 94 | } |
94 | 95 |
|
95 | | -exports.disconnect = function () { |
96 | | - pg.end(); |
| 96 | +exports.end = function(pool) { |
| 97 | + return function() { |
| 98 | + pool.end(); |
| 99 | + }; |
97 | 100 | } |
0 commit comments