Weishen/Hydro
2
0 Fork 0
mirror of https://gitcode.com/gh_mirrors/hy/Hydro.git synced 2026-10-17 09:36:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

prom-client: catch push errors

Browse Source
This commit is contained in:
undefined committed 2024-09-18 02:45:15 +08:00
1 parent cc331e75f3
commit e04e6c6f96
1 file changed
+22 -8
Download Patch File Download Diff File Expand all files Collapse all files
packages/prom-client/index.ts
+22 -8
View File
@@ -46,15 +46,29 @@ export function apply(ctx: Context) {
if (process.env.HYDRO_CLI) return;
const registry = createRegistry(ctx);
ctx.on('metrics', (id, metrics) => { instances[id] = metrics; });
let pushError = '';
ctx.setInterval(async () => {
const [gateway, name, pass] = SystemModel.getMany(['prom-client.gateway', 'prom-client.name', 'prom-client.password']);
if (gateway) {
const prefix = gateway.endsWith('/') ? gateway : `${gateway}/`;
const endpoint = `${prefix}metrics/job/hydro-web/instance/${encodeURIComponent(hostname())}:${process.env.NODE_APP_INSTANCE}`;
let req = superagent.post(endpoint);
if (name) req = req.set('Authorization', `Basic ${Buffer.from(`${name}:${pass}`).toString('base64')}`);
await req.send(await registry.metrics());
} else ctx.broadcast('metrics', `${hostname()}/${process.env.NODE_APP_INSTANCE}`, await registry.getMetricsAsJSON());
try {
const [gateway, name, pass] = SystemModel.getMany(['prom-client.gateway', 'prom-client.name', 'prom-client.password']);
if (gateway) {
const prefix = gateway.endsWith('/') ? gateway : `${gateway}/`;
const endpoint = `${prefix}metrics/job/hydro-web/instance/${encodeURIComponent(hostname())}:${process.env.NODE_APP_INSTANCE}`;
let req = superagent.post(endpoint);
if (name) req = req.auth(name, pass, { type: 'basic' });
await req.send(await registry.metrics());
} else ctx.broadcast('metrics', `${hostname()}/${process.env.NODE_APP_INSTANCE}`, await registry.getMetricsAsJSON());
pushError = '';
} catch (e) {
pushError = e.message;
}
}, 5000 * (+SystemModel.get('prom-client.collect_rate') || 1));
ctx.inject(['check'], (c) => {
const gateway = SystemModel.get('prom-client.gateway');
c.check.addChecker('prom-client', async (_, log, warn, error) => {
if (pushError) error(`Prometheus push error: ${pushError}`);
else if (gateway) log(`Prometheus pushed to gateway: ${gateway}`);
else log('Prometheus metrics server running.');
});
});
ctx.Route('metrics', '/metrics', MetricsHandler);
}