@shopify/koa-metrics/dist/middleware.js

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
var tslib_1 = require("tslib");
var statsd_1 = require("@shopify/statsd");
var tags_1 = require("./tags");
var timing_1 = require("./timing");
var content_1 = require("./content");
var timer_1 = require("./timer");
var CustomMetric;
(function (CustomMetric) {
    CustomMetric["ContentLength"] = "request_content_length";
    CustomMetric["QueuingTime"] = "request_queuing_time";
    CustomMetric["RequestDuration"] = "request_time";
})(CustomMetric = exports.CustomMetric || (exports.CustomMetric = {}));
function metrics(_a) {
    var prefix = _a.prefix, host = _a.host, _b = _a.skipInstrumentation, skipInstrumentation = _b === void 0 ? false : _b, logger = _a.logger;
    var statsdHost = host.split(':')[0];
    var statsdPort = parseInt(host.split(':')[1], 10);
    return function statsdMiddleware(ctx, next) {
        return tslib_1.__awaiter(this, void 0, void 0, function () {
            var metrics, promises, timer, queuingTime, duration, contentLength;
            return tslib_1.__generator(this, function (_a) {
                switch (_a.label) {
                    case 0:
                        metrics = new statsd_1.StatsDClient({
                            host: statsdHost,
                            port: statsdPort,
                            prefix: prefix,
                            globalTags: tags_1.tagsForRequest(ctx),
                            // eslint-disable-next-line no-console
                            logger: logger || ctx.log || console.log,
                        });
                        promises = [];
                        if (!skipInstrumentation) {
                            timer = timer_1.initTimer();
                            queuingTime = timing_1.getQueuingTime(ctx);
                            if (queuingTime) {
                                promises.push(metrics.distribution(CustomMetric.QueuingTime, queuingTime));
                            }
                        }
                        // allow later middleware to add metrics
                        ctx.metrics = metrics;
                        _a.label = 1;
                    case 1:
                        _a.trys.push([1, , 3, 6]);
                        return [4 /*yield*/, next()];
                    case 2:
                        _a.sent();
                        return [3 /*break*/, 6];
                    case 3:
                        metrics.addGlobalTags(tags_1.tagsForResponse(ctx));
                        if (!skipInstrumentation) {
                            if (timer) {
                                duration = timer.stop();
                                promises.push(metrics.distribution(CustomMetric.RequestDuration, duration));
                            }
                            contentLength = content_1.getContentLength(ctx);
                            if (contentLength) {
                                promises.push(metrics.distribution(CustomMetric.ContentLength, contentLength));
                            }
                        }
                        return [4 /*yield*/, Promise.all(promises)];
                    case 4:
                        _a.sent();
                        return [4 /*yield*/, metrics.close()];
                    case 5:
                        _a.sent();
                        return [7 /*endfinally*/];
                    case 6: return [2 /*return*/];
                }
            });
        });
    };
}
exports.default = metrics;