io/atajo.consul.js at master · Atajo/io · GitHub

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
const RxJs = require('@reactivex/rxjs');

let consul = null;


class Consul {

    constructor(opts) {

        this.consul = require('consul')(config.get('CONSUL').api);
        this.fetch = [];

        let fetchList = config.get('CONSUL').fetch;
        for (var i in fetchList) {
            this.add(i, fetchList[i]);
        }

        return this;
    }


    add(key, token) {
        this.fetch.push({ key: key, token: token });
    }


    start(interval = config.get('CONSUL').checkInterval) {

        let that = this;
        log.debug("Starting Consul Client [ Refresh every " + interval + "ms ]");

        return new RxJs.Observable(observer => {

            that.observer = observer;
            that.interval = setInterval((() => that.process()), interval);
            that.process();


        });


    }

    stop() {
        log.debug("Stopping Consul Client");
        clearInterval(this.interval);
    }


    process() {

        var that = this;

        for (var i in this.fetch) {

            let parameters = this.fetch[i];


            //log.debug("GETTING : ", parameters);

            that.consul.kv.get(parameters).then(response => {

                //log.debug("CONSUL:RESPONSE : ", response);

                that.observer.next({ key: response.Key, value: response.Value });

            }).catch(err => {
                log.error(err);
                that.observer.error(err);

            });

        }

    }


    map(domain) {

        var that = this;

        log.debug("GETTING MAP FOR " + domain + " -> ", config.get('CONSUL').map);
        let map = config.get('CONSUL').map;
        that.consul.kv.get(map).then(response => {

            let value = JSON.parse(response.Value);
            for (var i in value) {
                if (i == domain) {
                    log.debug("CONSUL:ADDING DOMAIN : ", i, value[i]);
                    that.add(i, value[i]);
                    break;
                }
            }

            if (that.fetch.length == 0) {
                log.error("DOMAIN " + domain + " NOT FOUND");
                process.exit(1);
            }

            that.process();


        }).catch(err => {
            log.error(err);

        });


    }

}


module.exports = Consul;