2020-05-16 14:33:26 +01:00

32 lines
846 B
JavaScript

require('dotenv').config()
const pg = require('pg')
const db = new pg.Client()
const task = process.argv[2]
const CREATE_TABLE_SQL = `CREATE TABLE IF NOT EXISTS tasks
(id SERIAL, task TEXT NOT NULL, PRIMARY KEY ( id ))`
const INSERT_TASK_SQL = `INSERT INTO tasks (task) VALUES ($1);`
const GET_TASKS_SQL = `SELECT * FROM tasks;`
db.connect((err) => {
if (err) throw err
db.query(CREATE_TABLE_SQL, (err) => {
if (err) throw err
if (task) {
db.query(INSERT_TASK_SQL, [task], (err) => {
if (err) throw err
listTasks()
})
} else {
listTasks()
}
})
})
function listTasks() {
db.query(GET_TASKS_SQL, (err, results) => {
if (err) throw err
console.log(results.rows);
db.end()
})
}