You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
uptime-kuma/server/model/monitor.js

267 lines
8.2 KiB

3 years ago
3 years ago
const dayjs = require("dayjs");
3 years ago
const utc = require('dayjs/plugin/utc')
var timezone = require('dayjs/plugin/timezone')
dayjs.extend(utc)
dayjs.extend(timezone)
const axios = require("axios");
const {tcping, ping} = require("../util-server");
3 years ago
const {R} = require("redbean-node");
3 years ago
const {BeanModel} = require("redbean-node/dist/bean-model");
const {Notification} = require("../notification")
3 years ago
/**
* status:
* 0 = DOWN
* 1 = UP
*/
3 years ago
class Monitor extends BeanModel {
async toJSON() {
let notificationIDList = {};
let list = await R.find("monitor_notification", " monitor_id = ? ", [
this.id
])
for (let bean of list) {
notificationIDList[bean.notification_id] = true;
}
3 years ago
return {
id: this.id,
name: this.name,
url: this.url,
3 years ago
hostname: this.hostname,
port: this.port,
maxretries: this.maxretries,
3 years ago
weight: this.weight,
3 years ago
active: this.active,
type: this.type,
interval: this.interval,
keyword: this.keyword,
notificationIDList
3 years ago
};
}
start(io) {
3 years ago
let previousBeat = null;
let retries = 0;
3 years ago
3 years ago
const beat = async () => {
3 years ago
if (! previousBeat) {
previousBeat = await R.findOne("heartbeat", " monitor_id = ? ORDER BY time DESC", [
this.id
])
}
3 years ago
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;
}
3 years ago
try {
if (this.type === "http" || this.type === "keyword") {
3 years ago
let startTime = dayjs().valueOf();
let res = await axios.get(this.url, {
headers: { 'User-Agent':'Uptime-Kuma' }
})
3 years ago
bean.msg = `${res.status} - ${res.statusText}`
bean.ping = dayjs().valueOf() - startTime;
if (this.type === "http") {
bean.status = 1;
} else {
let data = res.data;
// Convert to string for object/array
if (typeof data !== "string") {
data = JSON.stringify(data)
}
if (data.includes(this.keyword)) {
bean.msg += ", keyword is found"
bean.status = 1;
} else {
throw new Error(bean.msg + ", but keyword is not found")
}
}
3 years ago
} else if (this.type === "port") {
bean.ping = await tcping(this.hostname, this.port);
bean.msg = ""
3 years ago
bean.status = 1;
} else if (this.type === "ping") {
bean.ping = await ping(this.hostname);
bean.msg = ""
bean.status = 1;
3 years ago
}
retries = 0;
3 years ago
} catch (error) {
if ((this.maxretries > 0) && (retries < this.maxretries)) {
retries++;
bean.status = 2;
}
3 years ago
bean.msg = error.message;
}
// Mark as important if status changed, ignore pending pings,
// Don't notify if disrupted changes to up
if ((!previousBeat) || ((previousBeat.status !== bean.status) && bean.status !== 2 && !(previousBeat.status === 2 && bean.status !== 0))) {
3 years ago
bean.important = true;
// Do not send if first beat is UP
if (previousBeat || bean.status !== 1) {
let notificationList = await R.getAll(`SELECT notification.* FROM notification, monitor_notification WHERE monitor_id = ? AND monitor_notification.notification_id = notification.id `, [
this.id
])
let text;
if (bean.status === 1) {
text = "✅ Up"
} else {
text = "🔴 Down"
}
let msg = `[${this.name}] [${text}] ${bean.msg}`;
for(let notification of notificationList) {
try {
await Notification.send(JSON.parse(notification.config), msg, await this.toJSON(), bean.toJSON())
} catch (e) {
console.error("Cannot send notification to " + notification.name)
}
}
}
3 years ago
} else {
bean.important = false;
}
if (bean.status === 1) {
console.info(`Monitor #${this.id} '${this.name}': Successful Response: ${bean.ping} ms | Interval: ${this.interval} seconds | Type: ${this.type}`)
} else {
console.warn(`Monitor #${this.id} '${this.name}': Failing: ${bean.msg} | Type: ${this.type}`)
}
3 years ago
io.to(this.user_id).emit("heartbeat", bean.toJSON());
3 years ago
await R.store(bean)
3 years ago
Monitor.sendStats(io, this.id, this.user_id)
3 years ago
previousBeat = bean;
3 years ago
}
beat();
this.heartbeatInterval = setInterval(beat, this.interval * 1000);
}
stop() {
clearInterval(this.heartbeatInterval)
}
static async sendStats(io, monitorID, userID) {
Monitor.sendAvgPing(24, io, monitorID, userID);
3 years ago
Monitor.sendUptime(24, io, monitorID, userID);
Monitor.sendUptime(24 * 30, io, monitorID, userID);
}
3 years ago
/**
*
* @param duration : int Hours
*/
static async sendAvgPing(duration, io, monitorID, userID) {
let avgPing = parseInt(await R.getCell(`
SELECT AVG(ping)
FROM heartbeat
WHERE time > DATETIME('now', ? || ' hours')
3 years ago
AND ping IS NOT NULL
AND monitor_id = ? `, [
-duration,
monitorID
]));
io.to(userID).emit("avgPing", monitorID, avgPing);
}
3 years ago
/**
* Uptime with calculation
* Calculation based on:
* https://www.uptrends.com/support/kb/reporting/calculation-of-uptime-and-downtime
3 years ago
* @param duration : int Hours
*/
static async sendUptime(duration, io, monitorID, userID) {
let sec = duration * 3600;
let heartbeatList = await R.getAll(`
SELECT duration, time, status
3 years ago
FROM heartbeat
WHERE time > DATETIME('now', ? || ' hours')
3 years ago
AND monitor_id = ? `, [
-duration,
monitorID
]);
let downtime = 0;
let total = 0;
let uptime;
// Special handle for the first heartbeat only
if (heartbeatList.length === 1) {
if (heartbeatList[0].status === 1) {
uptime = 1;
} else {
uptime = 0;
}
} else {
for (let row of heartbeatList) {
let value = parseInt(row.duration)
let time = row.time
// Handle if heartbeat duration longer than the target duration
// e.g. Heartbeat duration = 28hrs, but target duration = 24hrs
if (value > sec) {
let trim = dayjs.utc().diff(dayjs(time), 'second');
value = sec - trim;
if (value < 0) {
value = 0;
}
}
total += value;
if (row.status === 0 || row.status === 2) {
downtime += value;
}
}
3 years ago
uptime = (total - downtime) / total;
if (uptime < 0) {
uptime = 0;
}
}
3 years ago
io.to(userID).emit("uptime", monitorID, duration, uptime);
}
3 years ago
}
module.exports = Monitor;