@lskjs/proxy/MetricsApi.js

"use strict";

var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports["default"] = exports.MetricsApi = void 0;

var _regenerator = _interopRequireDefault(require("@babel/runtime/regenerator"));

var _asyncToGenerator2 = _interopRequireDefault(require("@babel/runtime/helpers/asyncToGenerator"));

var _classCallCheck2 = _interopRequireDefault(require("@babel/runtime/helpers/classCallCheck"));

var _createClass2 = _interopRequireDefault(require("@babel/runtime/helpers/createClass"));

var _get2 = _interopRequireDefault(require("@babel/runtime/helpers/get"));

var _inherits2 = _interopRequireDefault(require("@babel/runtime/helpers/inherits"));

var _possibleConstructorReturn2 = _interopRequireDefault(require("@babel/runtime/helpers/possibleConstructorReturn"));

var _getPrototypeOf2 = _interopRequireDefault(require("@babel/runtime/helpers/getPrototypeOf"));

var _env = require("@lskjs/env");

var _serverApi = _interopRequireDefault(require("@lskjs/server-api"));

var _promClient = require("prom-client");

function _createSuper(Derived) { var hasNativeReflectConstruct = _isNativeReflectConstruct(); return function _createSuperInternal() { var Super = (0, _getPrototypeOf2["default"])(Derived), result; if (hasNativeReflectConstruct) { var NewTarget = (0, _getPrototypeOf2["default"])(this).constructor; result = Reflect.construct(Super, arguments, NewTarget); } else { result = Super.apply(this, arguments); } return (0, _possibleConstructorReturn2["default"])(this, result); }; }

function _isNativeReflectConstruct() { if (typeof Reflect === "undefined" || !Reflect.construct) return false; if (Reflect.construct.sham) return false; if (typeof Proxy === "function") return true; try { Boolean.prototype.valueOf.call(Reflect.construct(Boolean, [], function () {})); return true; } catch (e) { return false; } }

var MetricsApi = /*#__PURE__*/function (_Api) {
  (0, _inherits2["default"])(MetricsApi, _Api);

  var _super = _createSuper(MetricsApi);

  function MetricsApi() {
    (0, _classCallCheck2["default"])(this, MetricsApi);
    return _super.apply(this, arguments);
  }

  (0, _createClass2["default"])(MetricsApi, [{
    key: "getRoutes",
    value: function getRoutes() {
      return {
        '/': this.metrics.bind(this)
      };
    }
  }, {
    key: "init",
    value: function () {
      var _init = (0, _asyncToGenerator2["default"])( /*#__PURE__*/_regenerator["default"].mark(function _callee() {
        var _this = this;

        var proxy;
        return _regenerator["default"].wrap(function _callee$(_context) {
          while (1) {
            switch (_context.prev = _context.next) {
              case 0:
                _context.next = 2;
                return (0, _get2["default"])((0, _getPrototypeOf2["default"])(MetricsApi.prototype), "init", this).call(this);

              case 2:
                this.registry = new _promClient.Registry();
                this.registry.setDefaultLabels({
                  stage: _env.stage
                });
                _context.next = 6;
                return this.app.module('proxy');

              case 6:
                proxy = _context.sent;
                _context.t0 = Object;
                _context.next = 10;
                return proxy.metrics();

              case 10:
                _context.t1 = _context.sent;

                _context.t0.values.call(_context.t0, _context.t1).map(function (m) {
                  return _this.registry.registerMetric(m);
                });

              case 12:
              case "end":
                return _context.stop();
            }
          }
        }, _callee, this);
      }));

      function init() {
        return _init.apply(this, arguments);
      }

      return init;
    }()
  }, {
    key: "metrics",
    value: function () {
      var _metrics = (0, _asyncToGenerator2["default"])( /*#__PURE__*/_regenerator["default"].mark(function _callee2(req, res) {
        var proxy, raw;
        return _regenerator["default"].wrap(function _callee2$(_context2) {
          while (1) {
            switch (_context2.prev = _context2.next) {
              case 0:
                this.registry.resetMetrics();
                _context2.next = 3;
                return this.app.module('proxy');

              case 3:
                proxy = _context2.sent;
                _context2.next = 6;
                return proxy.metrics();

              case 6:
                _context2.next = 8;
                return this.registry.metrics();

              case 8:
                raw = _context2.sent;
                return _context2.abrupt("return", res.send(raw));

              case 10:
              case "end":
                return _context2.stop();
            }
          }
        }, _callee2, this);
      }));

      function metrics(_x, _x2) {
        return _metrics.apply(this, arguments);
      }

      return metrics;
    }()
  }]);
  return MetricsApi;
}(_serverApi["default"]);

exports.MetricsApi = MetricsApi;
var _default = MetricsApi;
exports["default"] = _default;
//# sourceMappingURL=MetricsApi.js.map