2021-08-10 04:51:30 -05:00
|
|
|
const PrometheusClient = require("prom-client");
|
2022-04-13 10:33:37 -05:00
|
|
|
const { log } = require("../src/util");
|
2021-07-27 11:52:31 -05:00
|
|
|
|
|
|
|
const commonLabels = [
|
2021-08-10 04:51:30 -05:00
|
|
|
"monitor_name",
|
|
|
|
"monitor_type",
|
|
|
|
"monitor_url",
|
|
|
|
"monitor_hostname",
|
|
|
|
"monitor_port",
|
2021-10-10 03:37:53 -05:00
|
|
|
];
|
2021-07-27 11:52:31 -05:00
|
|
|
|
2022-04-16 12:39:49 -05:00
|
|
|
const monitorCertDaysRemaining = new PrometheusClient.Gauge({
|
2021-08-10 04:51:30 -05:00
|
|
|
name: "monitor_cert_days_remaining",
|
|
|
|
help: "The number of days remaining until the certificate expires",
|
2021-08-10 01:55:06 -05:00
|
|
|
labelNames: commonLabels
|
|
|
|
});
|
|
|
|
|
2022-04-16 12:39:49 -05:00
|
|
|
const monitorCertIsValid = new PrometheusClient.Gauge({
|
2021-08-10 04:51:30 -05:00
|
|
|
name: "monitor_cert_is_valid",
|
|
|
|
help: "Is the certificate still valid? (1 = Yes, 0= No)",
|
2021-08-10 01:55:06 -05:00
|
|
|
labelNames: commonLabels
|
|
|
|
});
|
2022-04-16 12:39:49 -05:00
|
|
|
const monitorResponseTime = new PrometheusClient.Gauge({
|
2021-08-10 04:51:30 -05:00
|
|
|
name: "monitor_response_time",
|
|
|
|
help: "Monitor Response Time (ms)",
|
2021-07-27 11:52:31 -05:00
|
|
|
labelNames: commonLabels
|
|
|
|
});
|
|
|
|
|
2022-04-16 12:39:49 -05:00
|
|
|
const monitorStatus = new PrometheusClient.Gauge({
|
2021-08-10 04:51:30 -05:00
|
|
|
name: "monitor_status",
|
|
|
|
help: "Monitor Status (1 = UP, 0= DOWN)",
|
2021-07-27 11:52:31 -05:00
|
|
|
labelNames: commonLabels
|
|
|
|
});
|
|
|
|
|
|
|
|
class Prometheus {
|
|
|
|
monitorLabelValues = {}
|
|
|
|
|
|
|
|
constructor(monitor) {
|
|
|
|
this.monitorLabelValues = {
|
|
|
|
monitor_name: monitor.name,
|
|
|
|
monitor_type: monitor.type,
|
|
|
|
monitor_url: monitor.url,
|
|
|
|
monitor_hostname: monitor.hostname,
|
|
|
|
monitor_port: monitor.port
|
2021-10-10 03:37:53 -05:00
|
|
|
};
|
2021-07-27 11:52:31 -05:00
|
|
|
}
|
|
|
|
|
2021-08-10 04:51:30 -05:00
|
|
|
update(heartbeat, tlsInfo) {
|
2021-10-10 03:37:53 -05:00
|
|
|
|
2021-08-10 07:00:59 -05:00
|
|
|
if (typeof tlsInfo !== "undefined") {
|
2021-08-10 06:14:13 -05:00
|
|
|
try {
|
2022-04-17 02:43:03 -05:00
|
|
|
let isValid;
|
|
|
|
if (tlsInfo.valid === true) {
|
2022-04-13 11:30:32 -05:00
|
|
|
isValid = 1;
|
2021-08-10 06:14:13 -05:00
|
|
|
} else {
|
2022-04-13 11:30:32 -05:00
|
|
|
isValid = 0;
|
2021-08-10 06:14:13 -05:00
|
|
|
}
|
2022-04-16 12:39:49 -05:00
|
|
|
monitorCertIsValid.set(this.monitorLabelValues, isValid);
|
2021-08-10 06:14:13 -05:00
|
|
|
} catch (e) {
|
2022-04-13 10:33:37 -05:00
|
|
|
log.error("prometheus", "Caught error");
|
|
|
|
log.error("prometheus", e);
|
2021-08-10 06:14:13 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
try {
|
2022-01-14 00:51:45 -06:00
|
|
|
if (tlsInfo.certInfo != null) {
|
2022-04-16 12:39:49 -05:00
|
|
|
monitorCertDaysRemaining.set(this.monitorLabelValues, tlsInfo.certInfo.daysRemaining);
|
2022-01-12 02:12:12 -06:00
|
|
|
}
|
2021-08-10 06:14:13 -05:00
|
|
|
} catch (e) {
|
2022-04-13 10:33:37 -05:00
|
|
|
log.error("prometheus", "Caught error");
|
|
|
|
log.error("prometheus", e);
|
2021-08-10 06:14:13 -05:00
|
|
|
}
|
|
|
|
}
|
2021-08-10 04:51:30 -05:00
|
|
|
|
2021-07-27 11:52:31 -05:00
|
|
|
try {
|
2022-04-16 12:39:49 -05:00
|
|
|
monitorStatus.set(this.monitorLabelValues, heartbeat.status);
|
2021-07-27 11:52:31 -05:00
|
|
|
} catch (e) {
|
2022-04-13 10:33:37 -05:00
|
|
|
log.error("prometheus", "Caught error");
|
|
|
|
log.error("prometheus", e);
|
2021-07-27 11:52:31 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
try {
|
2021-08-10 04:51:30 -05:00
|
|
|
if (typeof heartbeat.ping === "number") {
|
2022-04-16 12:39:49 -05:00
|
|
|
monitorResponseTime.set(this.monitorLabelValues, heartbeat.ping);
|
2021-07-27 11:52:31 -05:00
|
|
|
} else {
|
|
|
|
// Is it good?
|
2022-04-16 12:39:49 -05:00
|
|
|
monitorResponseTime.set(this.monitorLabelValues, -1);
|
2021-07-27 11:52:31 -05:00
|
|
|
}
|
|
|
|
} catch (e) {
|
2022-04-13 10:33:37 -05:00
|
|
|
log.error("prometheus", "Caught error");
|
|
|
|
log.error("prometheus", e);
|
2021-07-27 11:52:31 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-01-06 21:57:24 -06:00
|
|
|
remove() {
|
|
|
|
try {
|
2022-04-16 12:39:49 -05:00
|
|
|
monitorCertDaysRemaining.remove(this.monitorLabelValues);
|
|
|
|
monitorCertIsValid.remove(this.monitorLabelValues);
|
|
|
|
monitorResponseTime.remove(this.monitorLabelValues);
|
|
|
|
monitorStatus.remove(this.monitorLabelValues);
|
2022-01-06 21:57:24 -06:00
|
|
|
} catch (e) {
|
|
|
|
console.error(e);
|
|
|
|
}
|
|
|
|
}
|
2021-07-27 11:52:31 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
module.exports = {
|
|
|
|
Prometheus
|
2021-10-10 03:37:53 -05:00
|
|
|
};
|