unleash.unleash/packages/unleash-api/lib/routes/metrics.js

'use strict';

const logger = require('../logger');
const ClientMetrics = require('../client-metrics');
const ClientMetricsService = require('../client-metrics/service');

module.exports = function (app, config) {
    const {
        clientMetricsDb,
        clientStrategiesDb,
        clientInstancesDb,
    } = config;
    const metrics = new ClientMetrics();
    const service = new ClientMetricsService(clientMetricsDb);

    service.on('metrics', (entries) => {
        entries.forEach((m) => {
            metrics.addPayload(m.metrics);
        });
    });

    app.get('/metrics', (req, res) => {
        res.json(metrics.getMetricsOverview());
    });

    app.get('/toggle-metrics', (req, res) => {
        res.json(metrics.getTogglesMetrics());
    });

    app.post('/client/metrics', (req, res) => {
        try {
            const data = typeof req.body === 'string' ? JSON.parse(req.body) : req.body;
            service
                .insert(data)
                .catch(e => logger.error('Error inserting metrics data', e));
        } catch (e) {
            logger.error('Error receiving metrics', e);
        }

        res.end();
    });

    app.post('/client/register', (req, res) => {
        const data = req.body;
        clientStrategiesDb.insert(data.appName, data.strategies)
            .then(() => clientInstancesDb.insert({
                appName: data.appName,
                instanceId: data.instanceId,
                clientIp: req.ip,
            }))
            .then(() => console.log('new client registerd'))
            .catch((error) => logger.error('Error registering client', error));

        res.end();
    });

    app.get('/client/strategies', (req, res) => {
        clientStrategiesDb.getAll().then(data => res.json(data));
    });

    app.get('/client/instances', (req, res) => {
        clientInstancesDb.getAll()
            .then(data => res.json(data))
            .catch(err => console.error(err));
    });
};