aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--routes/index.js16
1 files changed, 8 insertions, 8 deletions
diff --git a/routes/index.js b/routes/index.js
index 782d361..7c9d170 100644
--- a/routes/index.js
+++ b/routes/index.js
@@ -44,7 +44,7 @@ function pager(req, count) {
};
}
-function connect(cb) {
+function connect(res, cb) {
pg.connect(process.env.DATABASE_URL, function(err, client) {
if(err) {
res.send('Unable to connect to PostgreSQL.', {'Content-Type': 'text/plain'}, 500);
@@ -167,11 +167,11 @@ exports.index = method({
});
function getDepartments(req, res) {
- connect(function(client) {
- var sql = 'SELECT n_live_tup FROM pg_stat_user_tables WHERE relname=\'departments\'';
+ connect(res, function(client) {
+ var sql = 'SELECT COUNT(dept_no) AS count FROM departments';
var sql2 = 'SELECT dept_no, dept_name FROM departments OFFSET $1 LIMIT $2';
client.query(sql, after(res, function(rs) {
- var p = pager(req, parseInt(rs.rows[0].n_live_tup));
+ var p = pager(req, parseInt(rs.rows[0].count));
client.query(sql2, [ p.offset, p.limit ], after(res, function(rs2) {
switch(req.accept.types.getBestMatch(['text/html', 'application/vnd.collection+json'])) {
case 'text/html':
@@ -238,7 +238,7 @@ exports.department = method({
});
function getEmployeesInDepartment(req, res) {
- connect(function(client) {
+ connect(res, function(client) {
var dept_no = req.params.dept_no;
// TODO: Add dept_name to view
// TODO: Add manager as a link
@@ -283,7 +283,7 @@ exports.employeesInDepartment = method({
});
function getEmployees(req, res) {
- connect(function(client) {
+ connect(res, function(client) {
var emp_no = req.params.emp_no;
var sql1, sql2;
var sql1Params = [];
@@ -351,7 +351,7 @@ exports.employees = method({
});
function getEmployee(req, res) {
- connect(function(client) {
+ connect(res, function(client) {
var emp_no = req.params.emp_no;
var sql = 'SELECT * FROM employees WHERE employees.emp_no=$1';
client.query(sql, [ emp_no ], after(res, function(rs) {
@@ -394,7 +394,7 @@ function getEmployee(req, res) {
}
function deleteEmployee(req, res) {
- connect(function(client) {
+ connect(res, function(client) {
var emp_no = parseInt(req.params.emp_no) || 0;
if(emp_no < 10000 || emp_no > 999999) {
return send_text(res, 404, 'Illegal emp_no.');