Seluard
2 months ago
parent
9e01917a08
commit
1666659d48
@ -0,0 +1,21 @@
|
||||
exports.up = function (knex) {
|
||||
// Add new columns monitor.datadog*
|
||||
return knex.schema
|
||||
.alterTable("monitor", function (table) {
|
||||
table.string("datadog_site", 255);
|
||||
table.string("datadog_api_key", 255);
|
||||
table.string("datadog_app_key", 255);
|
||||
table.string("datadog_monitor_id", 255);
|
||||
});
|
||||
};
|
||||
|
||||
exports.down = function (knex) {
|
||||
// Drop column monitor.datadog*
|
||||
return knex.schema
|
||||
.alterTable("monitor", function (table) {
|
||||
table.dropColumn("datadog_site");
|
||||
table.dropColumn("datadog_api_key");
|
||||
table.dropColumn("datadog_app_key");
|
||||
table.dropColumn("datadog_monitor_id");
|
||||
});
|
||||
};
|
||||
@ -0,0 +1,58 @@
|
||||
const { MonitorType } = require("./monitor-type");
|
||||
const { UP, log, DOWN, PENDING } = require("../../src/util");
|
||||
const { client, v1 } = require("@datadog/datadog-api-client");
|
||||
const { json } = require("express");
|
||||
|
||||
/**
|
||||
* A DataDog class extends the MonitorType.
|
||||
* It will query DataDog api to get datadog monitor state and sync the monitor status.
|
||||
*/
|
||||
class DataDog extends MonitorType {
|
||||
|
||||
name = "datadog";
|
||||
/**
|
||||
* @param {Object} monitor - The monitor object associated with the check.
|
||||
* @param {Object} heartbeat - The heartbeat object to update.
|
||||
* @throws Will throw an error if the API call found any.
|
||||
*/
|
||||
async check(monitor, heartbeat) {
|
||||
try {
|
||||
const configurationOpts = {
|
||||
authMethods: {
|
||||
apiKeyAuth: monitor.datadog_api_key,
|
||||
appKeyAuth: monitor.datadog_app_key
|
||||
},
|
||||
};
|
||||
const configuration = client.createConfiguration(configurationOpts);
|
||||
configuration.setServerVariables({
|
||||
site: monitor.datadog_site
|
||||
});
|
||||
const apiInstance = new v1.MonitorsApi(configuration)
|
||||
let params = {
|
||||
monitorId: monitor.datadog_monitor_id,
|
||||
};
|
||||
let status = await apiInstance.getMonitor(params).then((data) => {
|
||||
log.debug("datadog",'DataDog API called successfully');
|
||||
let state = data["overallState"]
|
||||
if (state === "OK") {
|
||||
log.debug("datadog","UP")
|
||||
return UP;
|
||||
} else if (state === "No Data") {
|
||||
log.debug("datadog","PENDING")
|
||||
return PENDING;
|
||||
} else {
|
||||
log.debug("datadog","DOWN")
|
||||
return DOWN;
|
||||
}
|
||||
})
|
||||
heartbeat.status = status
|
||||
} catch (err) {
|
||||
// trigger log function somewhere to display a notification or alert to the user (but how?)
|
||||
throw new Error(`Error checking datadog monitor: ${err}`);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
module.exports = {
|
||||
DataDog,
|
||||
};
|
||||
Loading…
Reference in new issue