server.js 2.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485
  1. var express = require('express'),
  2. async = require('async'),
  3. pg = require("pg"),
  4. cookieParser = require('cookie-parser'),
  5. bodyParser = require('body-parser'),
  6. methodOverride = require('method-override'),
  7. app = express(),
  8. server = require('http').Server(app),
  9. io = require('socket.io')(server);
  10. io.set('transports', ['polling']);
  11. var port = process.env.PORT || 4000;
  12. io.sockets.on('connection', function (socket) {
  13. socket.emit('message', { text : 'Welcome!' });
  14. socket.on('subscribe', function (data) {
  15. socket.join(data.channel);
  16. });
  17. });
  18. async.retry(
  19. {times: 1000, interval: 1000},
  20. function(callback) {
  21. pg.connect('postgres://postgres@db/postgres', function(err, client, done) {
  22. if (err) {
  23. console.error("Waiting for db");
  24. }
  25. callback(err, client);
  26. });
  27. },
  28. function(err, client) {
  29. if (err) {
  30. return console.err("Giving up");
  31. }
  32. console.log("Connected to db");
  33. getVotes(client);
  34. }
  35. );
  36. function getVotes(client) {
  37. client.query('SELECT vote, COUNT(id) AS count FROM votes GROUP BY vote', [], function(err, result) {
  38. if (err) {
  39. console.error("Error performing query: " + err);
  40. } else {
  41. var votes = collectVotesFromResult(result);
  42. io.sockets.emit("scores", JSON.stringify(votes));
  43. }
  44. setTimeout(function() {getVotes(client) }, 1000);
  45. });
  46. }
  47. function collectVotesFromResult(result) {
  48. var votes = {a: 0, b: 0};
  49. result.rows.forEach(function (row) {
  50. votes[row.vote] = parseInt(row.count);
  51. });
  52. return votes;
  53. }
  54. app.use(cookieParser());
  55. app.use(bodyParser());
  56. app.use(methodOverride('X-HTTP-Method-Override'));
  57. app.use(function(req, res, next) {
  58. res.header("Access-Control-Allow-Origin", "*");
  59. res.header("Access-Control-Allow-Headers", "Origin, X-Requested-With, Content-Type, Accept");
  60. res.header("Access-Control-Allow-Methods", "PUT, GET, POST, DELETE, OPTIONS");
  61. next();
  62. });
  63. app.use(express.static(__dirname + '/views'));
  64. app.get('/', function (req, res) {
  65. res.sendFile(path.resolve(__dirname + '/views/index.html'));
  66. });
  67. server.listen(port, function () {
  68. var port = server.address().port;
  69. console.log('App running on port ' + port);
  70. });