Add Basic Auth for /metrics

server/model/monitor.js

@@ -1,4 +1,3 @@
-const Prometheus = require('prom-client');
 const https = require('https');
 const dayjs = require("dayjs");
 const utc = require('dayjs/plugin/utc')
@@ -6,6 +5,7 @@ var timezone = require('dayjs/plugin/timezone')
 dayjs.extend(utc)
 dayjs.extend(timezone)
 const axios = require("axios");
+const {Prometheus} = require("../prometheus");
 const {debug, UP, DOWN, PENDING} = require("../util");
 const {tcping, ping, checkCertificate} = require("../util-server");
 const {R} = require("redbean-node");
@@ -18,26 +18,6 @@ const customAgent = new https.Agent({
     maxCachedSessions: 0
 });
 
-const commonLabels = [
-    'monitor_name',
-    'monitor_type',
-    'monitor_url',
-    'monitor_hostname',
-    'monitor_port',
-]
-
-const monitor_response_time = new Prometheus.Gauge({
-    name: 'monitor_response_time',
-    help: 'Monitor Response Time (ms)',
-    labelNames: commonLabels
-});
-
-const monitor_status = new Prometheus.Gauge({
-    name: 'monitor_status',
-    help: 'Monitor Status (1 = UP, 0= DOWN)',
-    labelNames: commonLabels
-});
-
 /**
  * status:
  *      0 = DOWN
@@ -76,13 +56,7 @@ class Monitor extends BeanModel {
         let previousBeat = null;
         let retries = 0;
 
-        const monitorLabelValues = {
-                monitor_name: this.name,
-                monitor_type: this.type,
-                monitor_url: this.url,
-                monitor_hostname: this.hostname,
-                monitor_port: this.port
-        }
+        let prometheus = new Prometheus(this);
 
         const beat = async () => {
 
@@ -219,7 +193,6 @@ class Monitor extends BeanModel {
                 bean.important = false;
             }
 
-            monitor_status.set(monitorLabelValues, bean.status)
 
             if (bean.status === UP) {
                 console.info(`Monitor #${this.id} '${this.name}': Successful Response: ${bean.ping} ms | Interval: ${this.interval} seconds | Type: ${this.type}`)
@@ -229,7 +202,7 @@ class Monitor extends BeanModel {
                 console.warn(`Monitor #${this.id} '${this.name}': Failing: ${bean.msg} | Type: ${this.type}`)
             }
 
-            monitor_response_time.set(monitorLabelValues, bean.ping)
+            prometheus.update(bean)
 
             io.to(this.user_id).emit("heartbeat", bean.toJSON());