Infra-Mirrors/uptime-kuma
1
0
Fork 0
mirror of https://github.com/louislam/uptime-kuma.git synced 2024-12-18 20:24:36 -05:00
Code Issues Packages Projects Releases Wiki Activity

added MongoDB ping monitor

Browse Source
This commit is contained in:
rmarops 2022-11-16 20:50:34 -05:00
parent cb4e512dc6
commit b059a36e66
5 changed files with 2227 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2181
package-lock.json generated
View File

File diff suppressed because it is too large Load Diff

1
package.json
View File

@ -93,6 +93,7 @@
"jsonwebtoken": "~8.5.1",
"jwt-decode": "~3.1.2",
"limiter": "~2.1.0",
"mongodb": "^4.12.0",
"mqtt": "~4.3.7",
"mssql": "~8.1.4",
"node-cloudflared-tunnel": "~1.0.9",

10
server/model/monitor.js
View File

@ -3,7 +3,7 @@ const dayjs = require("dayjs");
const axios = require("axios");
const { Prometheus } = require("../prometheus");
const { log, UP, DOWN, PENDING, MAINTENANCE, flipStatus, TimeLogger } = require("../../src/util");
const { tcping, ping, dnsResolve, checkCertificate, checkStatusCode, getTotalClientInRoom, setting, mssqlQuery, postgresQuery, mqttAsync, setSetting, httpNtlm, radius, grpcQuery } = require("../util-server");
const { tcping, ping, dnsResolve, checkCertificate, checkStatusCode, getTotalClientInRoom, setting, mssqlQuery, postgresQuery, mongodbPing, mqttAsync, setSetting, httpNtlm, radius, grpcQuery } = require("../util-server");
const { R } = require("redbean-node");
const { BeanModel } = require("redbean-node/dist/bean-model");
const { Notification } = require("../notification");
@ -578,6 +578,14 @@ class Monitor extends BeanModel {
await postgresQuery(this.databaseConnectionString, this.databaseQuery);
bean.msg = "";
bean.status = UP;
bean.ping = dayjs().valueOf() - startTime;
} else if (this.type === "mongodb") {
let startTime = dayjs().valueOf();
await mongodbPing(this.databaseConnectionString);
bean.msg = "";
bean.status = UP;
bean.ping = dayjs().valueOf() - startTime;

22
server/util-server.js
View File

@ -13,6 +13,7 @@ const { badgeConstants } = require("./config");
const mssql = require("mssql");
const { Client } = require("pg");
const postgresConParse = require("pg-connection-string").parse;
const { MongoClient } = require("mongodb");
const { NtlmClient } = require("axios-ntlm");
const { Settings } = require("./settings");
const grpc = require("@grpc/grpc-js");
@ -294,6 +295,27 @@ exports.postgresQuery = function (connectionString, query) {
});
};
/**
* Connect to and Ping a MongoDB database
* @param {string} connectionString The database connection string
* @returns {Promise<(string[]|Object[]|Object)>}
*/
exports.mongodbPing = async function (connectionString) {
let client, db;
try {
client = await MongoClient.connect(connectionString, {useNewUrlParser: true})
db = client.db();
dbping = await db.command({ ping: 1 });
if (dbping["ok"] === 1) {
return 'UP'
}
throw Error("failed");
}
catch(err){ console.error(err); }
finally{ client.close(); }
}
/**
* Query radius server
* @param {string} hostname Hostname of radius server

14
src/pages/EditMonitor.vue
View File

@ -54,6 +54,9 @@
<option value="postgres">
PostgreSQL
</option>
<option value="mongodb">
MongoDB
</option>
<option value="radius">
Radius
</option>
@ -262,6 +265,17 @@
</div>
</template>
<!-- MongoDB -->
<template v-if="monitor.type === 'mongodb'">
<div class="my-3">
<label for="sqlConnectionString" class="form-label">{{ $t("Connection String") }}</label>
<template v-if="monitor.type === 'mongodb'">
<input id="sqlConnectionString" v-model="monitor.databaseConnectionString" type="text" class="form-control" placeholder="mongodb://username:password@host:port/database">
</template>
</div>
</template>
<!-- Interval -->
<div class="my-3">
<label for="interval" class="form-label">{{ $t("Heartbeat Interval") }} ({{ $t("checkEverySecond", [ monitor.interval ]) }})</label>