1
0
mirror of https://github.com/louislam/uptime-kuma.git synced 2024-12-04 09:42:56 +02:00
uptime-kuma/server/model/monitor.js

146 lines
3.9 KiB
JavaScript
Raw Normal View History

2021-07-01 08:03:06 +02:00
2021-06-25 15:55:49 +02:00
const dayjs = require("dayjs");
2021-06-27 10:10:55 +02:00
const utc = require('dayjs/plugin/utc')
var timezone = require('dayjs/plugin/timezone')
dayjs.extend(utc)
dayjs.extend(timezone)
const axios = require("axios");
2021-07-01 08:03:06 +02:00
const {tcping} = require("../util-server");
2021-06-27 10:10:55 +02:00
const {R} = require("redbean-node");
2021-06-25 15:55:49 +02:00
const {BeanModel} = require("redbean-node/dist/bean-model");
2021-06-27 10:10:55 +02:00
/**
* status:
* 0 = DOWN
* 1 = UP
*/
2021-06-25 15:55:49 +02:00
class Monitor extends BeanModel {
toJSON() {
return {
id: this.id,
name: this.name,
url: this.url,
2021-07-01 08:03:06 +02:00
hostname: this.hostname,
port: this.port,
2021-07-01 07:11:16 +02:00
weight: this.weight,
2021-06-25 15:55:49 +02:00
active: this.active,
type: this.type,
interval: this.interval,
};
}
start(io) {
2021-06-29 10:06:20 +02:00
let previousBeat = null;
2021-06-27 10:10:55 +02:00
const beat = async () => {
2021-06-25 15:55:49 +02:00
console.log(`Monitor ${this.id}: Heartbeat`)
2021-06-27 10:10:55 +02:00
2021-06-29 10:06:20 +02:00
if (! previousBeat) {
previousBeat = await R.findOne("heartbeat", " monitor_id = ? ORDER BY time DESC", [
this.id
])
}
2021-06-27 10:10:55 +02:00
let bean = R.dispense("heartbeat")
bean.monitor_id = this.id;
bean.time = R.isoDateTime(dayjs.utc());
bean.status = 0;
// Duration
if (previousBeat) {
bean.duration = dayjs(bean.time).diff(dayjs(previousBeat.time), 'second');
} else {
bean.duration = 0;
console.log(previousBeat)
}
2021-06-27 10:10:55 +02:00
try {
if (this.type === "http") {
let startTime = dayjs().valueOf();
let res = await axios.get(this.url)
bean.msg = `${res.status} - ${res.statusText}`
bean.ping = dayjs().valueOf() - startTime;
bean.status = 1;
2021-07-01 08:03:06 +02:00
} else if (this.type === "port") {
bean.ping = await tcping(this.hostname, this.port);
bean.status = 1;
2021-06-27 10:10:55 +02:00
}
} catch (error) {
bean.msg = error.message;
}
2021-06-29 10:06:20 +02:00
// Mark as important if status changed
if (! previousBeat || previousBeat.status !== bean.status) {
bean.important = true;
} else {
bean.important = false;
}
io.to(this.user_id).emit("heartbeat", bean.toJSON());
2021-06-27 10:10:55 +02:00
await R.store(bean)
2021-07-01 08:03:06 +02:00
Monitor.sendStats(io, this.id, this.user_id)
2021-06-29 10:06:20 +02:00
previousBeat = bean;
2021-06-25 15:55:49 +02:00
}
beat();
this.heartbeatInterval = setInterval(beat, this.interval * 1000);
}
stop() {
clearInterval(this.heartbeatInterval)
}
2021-06-30 15:04:58 +02:00
static async sendStats(io, monitorID, userID) {
Monitor.sendAvgPing(24, io, monitorID, userID);
2021-07-01 07:11:16 +02:00
Monitor.sendUptime(24, io, monitorID, userID);
Monitor.sendUptime(24 * 30, io, monitorID, userID);
2021-06-30 15:04:58 +02:00
}
2021-07-01 07:11:16 +02:00
/**
*
* @param duration : int Hours
*/
2021-06-30 15:04:58 +02:00
static async sendAvgPing(duration, io, monitorID, userID) {
let avgPing = parseInt(await R.getCell(`
SELECT AVG(ping)
FROM heartbeat
WHERE time > DATE('now', ? || ' hours')
2021-07-01 07:11:16 +02:00
AND ping IS NOT NULL
2021-06-30 15:04:58 +02:00
AND monitor_id = ? `, [
-duration,
monitorID
]));
io.to(userID).emit("avgPing", monitorID, avgPing);
}
2021-07-01 07:11:16 +02:00
/**
*
* @param duration : int Hours
*/
static async sendUptime(duration, io, monitorID, userID) {
let downtime = parseInt(await R.getCell(`
SELECT SUM(duration)
FROM heartbeat
WHERE time > DATE('now', ? || ' hours')
AND status = 0
AND monitor_id = ? `, [
-duration,
monitorID
]));
let sec = duration * 3600;
let uptime = (sec - downtime) / sec;
2021-06-30 15:04:58 +02:00
2021-07-01 07:11:16 +02:00
io.to(userID).emit("uptime", monitorID, duration, uptime);
2021-06-30 15:04:58 +02:00
}
2021-06-25 15:55:49 +02:00
}
module.exports = Monitor;