forked from hoppity/kafka-consumer-monitor
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathserver.js
76 lines (66 loc) · 2.15 KB
/
server.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
var zkLib = require('./lib/ZooKeeper.js');
var kafkaLib = require('./lib/Kafka.js');
var cache = require('./lib/Cache.js');
var logger = require('./logger.js').logger;
var config = require('./config');
var express = require('express');
var Promise = require('promise');
var app = express();
var loadMonitorData = function() {
logger.trace('loading consumer metadata');
return zkLib
.loadConsumerMetaData()
.then(kafkaLib.getTopicOffsets)
.then(function() {
cache
.getContents()
.then(function(result){
logger.info(result.items, "consumer lags");
});
})
.then(function() {
setTimeout(function() {
loadMonitorData();
}, config.refreshInterval);
});
};
// set a delay to stop zk throwing errors from us trying to connect to quickly
loadMonitorData();
// CORS headers for external access from JS
app.use(function(req, res, next) {
res.header("Access-Control-Allow-Origin", "*");
res.header("Access-Control-Allow-Headers", "Origin, X-Requested-With, Content-Type, Accept");
next();
});
app.get('/monitor/refresh', function(req, res) {
logger.trace('load metadata called from external source');
loadMonitorData()
.then(function(){
logger.trace('completed loading metadata');
return res.status(200).send();
})
.catch(function(err){
if (err) {
return res.status(500).json({'error_code': 500, 'message': err});
}
});
});
app.get('/consumergroups', function(req, res){
cache.keys(function(err, keys){
logger.debug({keys: keys}, 'returned keys');
res.status(200).send(keys);
});
});
app.get('/consumers/:consumer/lag', function(req, res) {
var consumer = req.params.consumer;
cache.get(consumer, function(err, value){
if (!err) {
// TODO : sort the response here if required
res.send(value);
}
else {
res.status(500).send();
}
});
});
app.listen(config.server.port);