-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathdb.js
65 lines (57 loc) · 1.88 KB
/
db.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
/*
* ===================================================
* ===================================================
* ===================================================
* ===================================================
* ====== CONFIGURATION =========
* ===================================================
* ===================================================
* ===================================================
* ===================================================
*/
const pg = require('pg');
const url = require('url');
require('dotenv').config();
var configs;
if (process.env.DATABASE_URL) {
const params = url.parse(process.env.DATABASE_URL);
const auth = params.auth.split(':');
configs = {
user: auth[0],
password: auth[1],
host: params.hostname,
port: params.port,
database: params.pathname.split('/')[1],
ssl: true,
};
} else {
configs = {
user: process.env.PG_USER,
password: process.env.PG_PASSWORD,
host: process.env.PG_HOST,
port: process.env.PG_PORT,
database: process.env.PG_DATABASE,
};
}
const pool = new pg.Pool(configs);
pool.on('error', function (err) {
console.log('idle client error', err.message, err.stack);
});
const allMainModelsFunction = require('./models/main');
const allUserModelsFunction = require('./models/user');
const allGroupModelsFunction = require('./models/group');
const allBillModelsFunction = require('./models/bill');
const mainModelsObject = allMainModelsFunction(pool);
const userModelsObject = allUserModelsFunction(pool);
const groupModelsObject = allGroupModelsFunction(pool);
const billModelsObject = allBillModelsFunction(pool);
module.exports = {
queryInterface: (text, params, callback) => {
return pool.query(text, params, callback);
},
pool: pool,
main: mainModelsObject,
user: userModelsObject,
group: groupModelsObject,
bill: billModelsObject,
};