forked from stelligent/dromedary
-
Notifications
You must be signed in to change notification settings - Fork 0
/
app.js
159 lines (140 loc) · 4.69 KB
/
app.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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
var express = require('express');
var app = express();
var CS = require(__dirname + '/lib/inMemoryStorage.js');
var commitSha = require(__dirname + '/lib/sha.js');
var reqThrottle = require(__dirname + '/lib/requestThrottle.js');
var DDBP = require(__dirname + '/lib/dynamoDbPersist.js');
var serverPort = 8080;
var siteChartStore = {};
var ddbLastFetch = {};
var ddbPersist = new DDBP();
if (process.env.hasOwnProperty('AUTOMATED_ACCEPTANCE_TEST')) {
serverPort = 0;
}
/* Helper to refresh in memory store w/ data from DDB */
function updateColorCountsFromDdb(siteName, cb) {
var chartData = siteChartStore[siteName];
ddbPersist.getSiteCounts(siteName, chartData.getAllCounts(), function(err, data) {
if (err) {
cb(err);
} else {
chartData.setCounts(data);
ddbLastFetch[siteName] = Date.now();
cb(null, chartData);
}
});
}
/* Returns in memory store (refreshed with DDB data no more often than every second) */
function getChartData(siteName, cb) {
if (!siteChartStore.hasOwnProperty(siteName)) {
siteChartStore[siteName] = new CS(siteName);
ddbLastFetch[siteName] = 0;
}
if (Date.now() - ddbLastFetch[siteName] > 1000) {
// Fetch from DDB if it's been more than a second since last refresh
updateColorCountsFromDdb(siteName, cb);
} else {
cb(null, siteChartStore[siteName]);
}
}
/* Helper to send responses to frontend */
function sendJsonResponse(res, obj) {
res.setHeader('Content-Type', 'application/json');
res.setHeader('Access-Control-Allow-Origin', '*');
res.send(JSON.stringify(obj));
}
/* helper to determine client ip */
function getClientIp(req) {
var ip = req.ip;
if (req.headers.hasOwnProperty('x-real-ip')) {
ip = req.headers['x-real-ip'];
}
return ip;
}
/* clean up throttle map every minute to keep it tidy */
setInterval(reqThrottle.gcMap, 1000);
/* Host static content from /public */
app.use(express.static(__dirname + '/public'));
/* GET requests to /sha returns git commit sha */
app.get('/sha', function (req, res) {
console.log('Request received from %s for /sha', getClientIp(req));
sendJsonResponse(res, {sha: commitSha});
});
/* GET requests to /data return chart data values */
app.get('/data', function (req, res) {
console.log('Request received from %s for /data', getClientIp(req));
getChartData(req.headers.host, function (err, data) {
var chartData = data;
if (err) {
console.log(err);
sendJsonResponse(res, {error: err});
} else {
if (req.query.hasOwnProperty('countsOnly')) {
sendJsonResponse(res, chartData.getAllCounts());
} else {
sendJsonResponse(res, chartData.getForChartJs());
}
}
});
});
/* GET requests to /increment to increment counts */
app.get('/increment', function (req, res) {
var ip = getClientIp(req);
if (! reqThrottle.checkIp(ip) ) {
console.log('Request throttled from %s for /increment', ip);
sendJsonResponse(res, {error: 'Request throttled'});
return;
}
if (!req.query.hasOwnProperty('color')) {
console.log('No color specified in params');
sendJsonResponse(res, {count: 0});
return;
}
getChartData(req.headers.host, function (err, data) {
console.log('Request received from %s for /increment', ip);
reqThrottle.logIp(ip);
if (err) {
console.log(err);
sendJsonResponse(res, {error: err});
return;
}
if (! data.colorExists(req.query.color)) {
console.log('Increment received for unknown color ' + req.query.color);
sendJsonResponse(res, {error: 'Unknown color'});
return;
}
ddbPersist.incrementCount(req.headers.host, req.query.color, function (err) {
console.log('Incrementing count for ' + req.query.color);
if (err) {
console.log(err);
sendJsonResponse( res, {error: 'Failed to increment color count in DDB'});
return;
}
updateColorCountsFromDdb(req.headers.host, function (err, data) {
if (err) {
console.log(err);
sendJsonResponse( res, {error: 'Failed to increment color count in DDB'});
return;
}
sendJsonResponse(res, {count: data.getCount(req.query.color)});
});
});
});
});
ddbPersist.init(function(err) {
var server;
if (err) {
console.log('Failed to init DynamoDB persistence');
console.log(err);
process.exit(1);
}
server = app.listen(serverPort, function () {
var host = server.address().address;
var port = server.address().port;
console.log('Listening on %s:%s', host, port);
if (process.env.hasOwnProperty('AUTOMATED_ACCEPTANCE_TEST')) {
require('fs').writeFileSync(__dirname + '/dev-lib/targetPort.js',
'module.exports = ' + port + ';\n');
}
});
});