· 6 years ago · Mar 21, 2019, 07:34 PM
1const { Pool } = require('pg');
2const dotenv = require('dotenv');
3
4dotenv.config();
5
6const pool = new Pool({
7 connectionString: process.env.DATABASE_URL
8});
9
10pool.on('connect', () => {
11 console.log('connected to the db');
12});
13
14/**
15 * Create Tables
16 */
17const createTables = () => {
18 const queryText =
19 `CREATE TABLE IF NOT EXISTS
20 reflections(
21 id UUID PRIMARY KEY,
22 success VARCHAR(128) NOT NULL,
23 low_point VARCHAR(128) NOT NULL,
24 take_away VARCHAR(128) NOT NULL,
25 created_date TIMESTAMP,
26 modified_date TIMESTAMP
27 )`;
28
29 pool.query(queryText)
30 .then((res) => {
31 console.log(res);
32 pool.end();
33 })
34 .catch((err) => {
35 console.log(err);
36 pool.end();
37 });
38}
39
40/**
41 * Drop Tables
42 */
43const dropTables = () => {
44 const queryText = 'DROP TABLE IF EXISTS reflections';
45 pool.query(queryText)
46 .then((res) => {
47 console.log(res);
48 pool.end();
49 })
50 .catch((err) => {
51 console.log(err);
52 pool.end();
53 });
54}
55
56pool.on('remove', () => {
57 console.log('client removed');
58 process.exit(0);
59});
60
61module.exports = {
62 createTables,
63 dropTables
64};
65
66// Install make-runnable as devDependency
67require('make-runnable');