Windshaft-cartodb/lib/api/middlewares/profiler.js

'use strict';

const Profiler = require('../../stats/profiler-proxy');
const debug = require('debug')('windshaft:cartodb:stats');
const { name: prefix } = require('../../../package.json');

module.exports = function profiler (options) {
    const { enabled = true, statsClient } = options;

    return function profilerMiddleware (req, res, next) {
        const { logger } = res.locals;

        // TODO: stop using profiler and log stats instead of adding them to the profiler
        req.profiler = new Profiler({
            statsd_client: statsClient,
            profile: enabled
        });

        req.profiler.start(prefix);

        res.on('finish', () => {
            req.profiler.done('response');
            req.profiler.end();
            logger.info({ stats: req.profiler.toJSON() });

            try {
                // May throw due to dns, see: http://github.com/CartoDB/Windshaft/issues/166
                req.profiler.sendStats();
            } catch (err) {
                debug('error sending profiling stats: ' + err);
            }
        });

        next();
    };
};