-
Notifications
You must be signed in to change notification settings - Fork 3
/
Copy pathconsumer.js
45 lines (37 loc) · 1016 Bytes
/
consumer.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
var kafka = require('kafka-node'),
Consumer = kafka.Consumer,
knex = require('knex'),
glob = require("glob"),
path = require("path"),
client = new kafka.Client();
var config = require('./config.js');
for (var key in config.databases) {
config[key] = knex({client:'pg', connection:config.databases[key], debug:true});
}
var handlers = {};
var topics = [];
glob("./scripts/*.js", function(er, files) {
console.log(files);
files.forEach(function(f) {
var mod = require(f);
if (typeof(mod) == "object") {
for (var topic in mod) {
topics.push({topic: topic, partition:0});
handlers[topic] = mod[topic];
}
} else {
var topic = path.basename(f, '.js');
topics.push({topic: topic, partition:0});
handlers[topic] = mod;
}
});
console.log("Topics ", topics);
var consumer = new Consumer(client, topics);
consumer.on('message', function(message) {
try {
handlers[message.topic](config, JSON.parse(message.value));
} catch (e) {
console.log(e);
}
});
});