server-apm.js 2.42 KB
const Sender = require('influx-batch-sender');
const MysqlSender = require('../lib/mysql-sender');
const config = require('../common/config');
const msg2row = require('./msg2row');
const lineparse = require('../lib/line-parse');

const errorSender = new MysqlSender(config.table.error);
const slowRouterSender = new MysqlSender(config.table.slow);

const routeSender = new Sender(config.reportRoute);
const apiSender = new Sender(config.reportApi);
const _ = require('lodash');

const server = {
    async handle(data) {
        let msgs = lineparse.parse(data);

        for (let m of msgs) {
            if (m.measurement === 'web-server-duration') {
                let duration = parseInt(m.fields.duration);

                if (duration > config.slowRoute.min && duration < config.slowRoute.max) {
                    slowRouterSender.addMessage(msg2row.slowRouter(m));
                }

                if (m.tags.type.toLowerCase() === 'api') {
                    apiSender.addMessage({
                        tags: {
                            app: m.tags.app,
                            host: m.tags.hostname,
                            api: m.tags.api
                        },
                        fields: {
                            duration: duration,
                            times: 1
                        }
                    });
                }

                if (m.tags.type.toLowerCase() === 'route') {
                    routeSender.addMessage({
                        tags: {
                            app: m.tags.app,
                            host: m.tags.hostname,
                            route: m.tags.route
                        },
                        fields: {
                            duration: duration
                        }
                    });
                }
            } else if (m.measurement === 'error-report') {
                errorSender.addMessage(msg2row.errorRouter(m));
            } else if (m.measurement === 'process-info') {
                m.measurement = 'process-info';
                _.set(m, 'fields.memory', parseInt(_.get(m, 'fields.memory', '0')));
                _.set(m, 'fields.cpu', parseInt(_.get(m, 'fields.cpu', '0')));
                routeSender.addMessage(m);
            }
        }
    }
};

module.exports = function(req, res, next) {
    let data = req.body || '';

    server.handle(data).then(() => {
        res.status(204).json();
    }).catch(next);
};