This repository has been archived by the owner on Sep 7, 2022. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 0
/
influx_collector.js
83 lines (68 loc) · 2.13 KB
/
influx_collector.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
// Copyright 2018 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
var InfluxNodeClient = require('./influx_node_client');
var InfluxCollector = require('zetta-device-data-influxdb');
var url = require('url');
module.exports = function(server) {
var opts = {
host: process.env.COREOS_PRIVATE_IPV4
};
// allow a list of peers to be passed, overides COREOS_PRIVATE_IPV4
if (process.env.ETCD_PEER_HOSTS) {
opts.host = process.env.ETCD_PEER_HOSTS.split(',');
}
var client = new InfluxNodeClient(opts);
var db = process.env.INFLUX_DATABASE;
client.findAll(function(err, results) {
client.on('change', function(results) {
if(results.length) {
update(results[0]);
}
});
if(!err && results.length) {
return connect(results[0]);
}
if(err) {
server.log('Error searching for influx: ' + err);
}
});
var influx = null;
function connect(influxUrl) {
server.log('Connecting to influxdb at: ' + influxUrl.url);
var endpoint = influxUrl.url;
var endpointUrl = url.parse(endpoint);
var opts = {
host: endpointUrl.hostname,
port: endpointUrl.port,
database: db
}
influx = new InfluxCollector(opts);
influx._collect(server);
}
function update(influxUrl) {
if(influx) {
server.log('Updating influx url: ' + influxUrl.url);
var endpoint = influxUrl.url;
var endpointUrl = url.parse(endpoint);
var opts = {
host: endpointUrl.hostname,
port: endpointUrl.port,
database: db
}
influx.configure(opts);
} else {
connect(influxUrl);
}
}
}