Infrastructure-Public/srcds_exporter
5
0
Fork 0
mirror of https://github.com/Sithranduil/srcds_exporter.git synced 2025-07-23 08:28:00 +02:00
Code Issues Projects Releases Wiki Activity

Découpage en module

Browse Source
This commit is contained in:
Sithranduil
2020-10-17 19:32:32 +02:00
parent 14fdcc39c2
commit 47ea30b234
12 changed files with 3197 additions and 200 deletions
Download Patch File Download Diff File Expand all files Collapse all files

242
index.js
View File

@@ -1,218 +1,68 @@
const express = require('express');
const { connect, TimeoutError } = require('working-rcon');
const prometheus = require('prom-client');
const { Gauge } = prometheus;
import express from 'express';
import { connect } from 'working-rcon';
import module from './modules/index.js';
const app = express();
async function rconCommand(client, command) {
try {
return client.command(command);
} catch (err) {
if (err instanceof TimeoutError) {
console.error('request timed out');
} else {
throw err;
}
}
return null;
}
async function getMetamod(client) {
return rconCommand(client, 'meta');
}
function metamodRequest(response) {
if (response) {
const isEnabled = isValidResponse(response);
metamodEnabledGauge.set(isEnabled ? Number(1) : Number(0));
} else {
metamodEnabledGauge.set((Number(-1)));
}
}
async function getSourcemod(client) {
return rconCommand(client, 'sm');
}
function sourcemodRequest(response) {
if (response) {
const isEnabled = isValidResponse(response);
sourcemodEnabledGauge.set(isEnabled ? Number(1) : Number(0));
} else {
sourcemodEnabledGauge.set((Number(-1)));
}
}
async function getSourcepython(client) {
return rconCommand(client, 'sp');
}
function sourcepythonRequest(response) {
if (response) {
const isEnabled = isValidResponse(response);
sourcepythonEnabledGauge.set(isEnabled ? Number(1) : Number(0));
} else {
sourcepythonEnabledGauge.set(Number(-1));
}
}
function isValidResponse(response) {
return !response.includes('Unknown');
}
function gameRequest(response, game) {
if (game === 'csgo') {
status.set((Number(1)));
cpu.set((Number(response[0])));
netin.set((Number(response[1])));
netout.set((Number(response[2])));
uptime.set((Number(response[3])));
maps.set((Number(response[4])));
fps.set((Number(response[5])));
players.set((Number(response[6])));
svms.set((Number(response[7])));
varms.set((Number(response[8])));
tick.set((Number(response[9])));
} else {
status.set((Number(1)));
cpu.set((Number(response[0])));
netin.set((Number(response[1])));
netout.set((Number(response[2])));
uptime.set((Number(response[3])));
maps.set((Number(response[4])));
fps.set((Number(response[5])));
players.set((Number(response[6])));
}
}
async function getStats(client, game) {
const stats = await rconCommand(client, 'stats');
let statusLine = null;
if (stats) {
statusLine = stats.split(/\r?\n/);
} else {
return null;
}
if (game === 'csgo') {
statusLine = stats.split(/\r?\n/);
statusLine.pop();
}
statusLine.shift();
statusLine = statusLine[0].split(/\s+/);
return statusLine;
}
const csgoRegistry = new prometheus.Registry();
const gmodRegistry = new prometheus.Registry();
const cstrikeRegistry = new prometheus.Registry();
// Global metrics, used accross all Source gameservers
const status = new Gauge({ name: 'srcds_status', help: "The server's status, 0 = offline/bad password, 1 = online", registers: [csgoRegistry, gmodRegistry, cstrikeRegistry] });
const cpu = new Gauge({ name: 'srcds_cpu', help: 'Probably the priority level of the srcds executable from an operating system point of view (0 - No priority, 10 - biggest priority)', registers: [csgoRegistry, gmodRegistry, cstrikeRegistry] });
const netin = new Gauge({ name: 'srcds_netin', help: 'Incoming bandwidth, in kbps, received by the server', registers: [csgoRegistry, gmodRegistry, cstrikeRegistry] });
const netout = new Gauge({ name: 'srcds_netout', help: 'Incoming bandwidth, in kbps, sent by the server', registers: [csgoRegistry, gmodRegistry, cstrikeRegistry] });
const uptime = new Gauge({ name: 'srcds_uptime', help: "The server's uptime, in minutes", registers: [csgoRegistry, gmodRegistry, cstrikeRegistry] });
const maps = new Gauge({ name: 'srcds_maps', help: "The number of maps played on that server since it's start", registers: [csgoRegistry, gmodRegistry, cstrikeRegistry] });
const fps = new Gauge({ name: 'srcds_fps', help: "The server's tick (10 fps on idle, 64 fps for 64 ticks server, 128 fps for 128 ticks..)", registers: [csgoRegistry, gmodRegistry, cstrikeRegistry] });
const players = new Gauge({ name: 'srcds_players', help: 'The number of real players actually connected on the server', registers: [csgoRegistry, gmodRegistry, cstrikeRegistry] });
// CSGO metrics
const svms = new Gauge({ name: 'srcds_svms', help: 'ms per sim frame', registers: [csgoRegistry] });
const varms = new Gauge({ name: 'srcds_varms', help: 'ms variance', registers: [csgoRegistry] });
const tick = new Gauge({ name: 'srcds_tick', help: 'The time in MS per tick', registers: [csgoRegistry] });
// Metamod metrics
const metamodEnabledGauge = new Gauge({ name: 'srcds_metamod_enabled', help: 'Is metamod enabled', registers: [csgoRegistry] });
// Sourcemod metrics
const sourcemodEnabledGauge = new Gauge({ name: 'srcds_sourcemod_enabled', help: 'Is sourcemod enabled', registers: [csgoRegistry] });
// SourcePython metrics
const sourcepythonEnabledGauge = new Gauge({ name: 'srcds_sourcepython_enabled', help: 'Is sourcepython enabled', registers: [csgoRegistry] });
app.get('/', (req, res) => {
res.send('use /metrics?ip=<srcds ip>&port=<srcds port>&password=<rcon password>&game=<game>&sm=&ltboolean&gt&meta=&ltboolean&gtsp=&ltboolean&gt to get data');
res.send('use /metrics?ip=<srcds ip>&port=<srcds port>&rconPassword=<rcon password>&game=<game>&sm=&ltboolean&gt&meta=&ltboolean&gtsp=&ltboolean&gt to get data');
});
app.get('/metrics', async (req, res) => {
const { ip } = req.query;
const { port } = req.query;
const { password } = req.query;
const { game } = req.query;
const metamod = req.query.metamod || false;
const sourcemod = req.query.sourcemod || false;
const sourcepython = req.query.sourcepython || false;
const config = {
ip: req.query.ip,
port: req.query.port,
game: req.query.game,
rconPassword: req.query.rconPassword,
};
config.metamod = req.query.metamod === 'true' || false;
config.sourcemod = req.query.sourcemod === 'true' || false;
config.sourcepython = req.query.sourcepython === 'true' || false;
config.tags = req.query.tags || [];
if (ip == null || port == null || password == null || game == null) {
res.send('Missing parameter, either IP, port, RCON password or game<br />use /metrics?ip=&lt;srcds ip&gt;&port=&lt;srcds port&gt;&password=&lt;rcon password&gt;&game=&lt;game&gt;&sm=<boolean>&meta=<boolean>sp=<boolean> to get data');
if (
config.ip == null
|| config.port == null
|| config.rconPassword == null
|| config.game == null
) {
res.send('Missing parameter');
return;
}
if (game !== 'csgo' && game !== 'gmod' && game !== 'cstrike') {
if (!['csgo', 'cstrike', 'gmod'].includes(config.game)) {
res.send('Incorrect game value, currently supported games are : csgo, gmod, cstrike');
return;
}
const client = await connect(ip, port, password, 5000);
const statsResponse = await getStats(client, game);
try {
if (metamod) {
const metamodResponse = await getMetamod(client);
metamodRequest(metamodResponse);
} else {
metamodRequest(null);
}
if (sourcemod) {
const sourcemodResponse = await getSourcemod(client);
sourcemodRequest(sourcemodResponse);
} else {
sourcemodRequest(null);
}
if (sourcepython) {
const sourcepythonResponse = await getSourcepython(client);
sourcepythonRequest(sourcepythonResponse);
} else {
sourcepythonRequest(null);
}
const defaultLabels = { server: `${ip}:${port}`, game };
gameRequest(statsResponse, game);
if (game === 'csgo') {
csgoRegistry.setDefaultLabels(defaultLabels);
res.end(csgoRegistry.metrics());
} else if (game === 'cstrike') {
cstrikeRegistry.setDefaultLabels(defaultLabels);
res.end(cstrikeRegistry.metrics());
} else if (game === 'gmod') {
gmodRegistry.setDefaultLabels(defaultLabels);
res.end(gmodRegistry.metrics());
}
const client = await connect(config.ip, config.port, config.rconPassword, 5000);
const response = await module.request(config, client);
await module.send(config, response, res);
await client.disconnect();
} catch (e) {
status.set((Number(0)));
cpu.set((Number(0)));
netin.set((Number(0)));
netout.set((Number(0)));
uptime.set((Number(0)));
maps.set((Number(0)));
fps.set((Number(0)));
players.set((Number(0)));
svms.set((Number(0)));
varms.set((Number(0)));
tick.set((Number(0)));
metamodEnabledGauge.set((Number(-1)));
sourcemodEnabledGauge.set((Number(-1)));
sourcepythonEnabledGauge.set((Number(-1)));
if (game === 'csgo') {
res.end(csgoRegistry.metrics());
} else if (game === 'gmod') {
res.end(gmodRegistry.metrics());
} else if (game === 'cstrike') {
res.end(cstrikeRegistry.metrics());
if (e.code === 'EHOSTUNREACH') {
console.error(`Unreachable host : ${e.address}:${e.port}`);
} else {
console.error(e);
}
// status.set((Number(0)));
// cpu.set((Number(0)));
// netin.set((Number(0)));
// netout.set((Number(0)));
// uptime.set((Number(0)));
// maps.set((Number(0)));
// fps.set((Number(0)));
// players.set((Number(0)));
// svms.set((Number(0)));
// varms.set((Number(0)));
// svMaxUpdateRate.set((Number(0)));
// metamodEnabledGauge.set((Number(-1)));
// sourcemodEnabledGauge.set((Number(-1)));
// sourcepythonEnabledGauge.set((Number(-1)));
// prepareData(ip, port, game, tags);
// sendData(game, res);
}
await client.disconnect();
});
app.listen(9591);