forked from disqus/grockets
-
Notifications
You must be signed in to change notification settings - Fork 0
/
grockets.js
110 lines (93 loc) · 2.71 KB
/
grockets.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
var io = require('socket.io').listen(8000),
http = require('http');
var graphiteHost = 'CHANGE-ME-YOU-NOOB',
handlers = [];
io.enable('browser client minification');
io.enable('browser client etag');
io.enable('browser client gzip');
io.set('log level', 1);
function buildQueryString(parameters) {
'use strict';
var queryParameters = [];
parameters.targets.forEach(function(target) {
queryParameters.push('target=' + target);
});
queryParameters.push('from=' + parameters.from);
queryParameters.push('format=json');
return(queryParameters.join('&'));
}
function processHandlers() {
'use strict';
handlers.map(function(handler) {
if (handler.shouldUpdate()) {
handler.data(function(data) {
handler.socket.emit('metricData', {id: handler.id, metrics: data});
});
}
});
setTimeout(processHandlers, 1000);
}
function MetricHandler(socket, params) {
'use strict';
var lastUpdateTime = 0,
queryParameters = {
'targets': params.targets,
'from': params.from
};
this.id = params.id;
this.socket = socket;
this.addTarget = function(target) {
queryParameters.targets.push(target);
};
this.removeTarget = function(target) {
var index = queryParameters.targets.indexOf(target);
queryParameters.targets.splice(index, 1);
};
this.data = function(callback) {
var httpOptions = {
host: graphiteHost,
port: 80,
path: '/render?' + buildQueryString(queryParameters)
},
metricData = '';
http.get(httpOptions, function(res) {
res.on('data', function(chunk) {
metricData += chunk.toString();
}).on('error', function(e) {
console.log('Got error: ' + e.message);
}).on('end', function() {
lastUpdateTime = new Date().getTime();
callback(metricData);
});
});
};
this.shouldUpdate = function() {
if ((params.refresh === true && (lastUpdateTime + params.updateInterval < new Date().getTime())) || lastUpdateTime === 0) {
if (lastUpdateTime > 0) {
queryParameters.from = Math.floor((lastUpdateTime - (params.updateInterval * 3)) / 1000);
}
return(true);
} else {
return(false);
}
};
}
io.sockets.on('connection', function(socket) {
'use strict';
socket.on('fetchMetrics', function(params) {
handlers.push(new MetricHandler(socket, params));
}).on('addTarget', function(target) {
handlers.forEach(function(handler) {
if (handler.socket === socket) {
handler.addTarget(target);
}
});
}).on('removeTarget', function(target) {
handlers.forEach(function(handler) {
if (handler.socket === socket) {
handler.removeTarget(target);
}
});
});
});
processHandlers();