@lskjs/proxy/collectMetrics.js

"use strict";

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

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

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

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

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

var collectMetrics = /*#__PURE__*/function () {
  var _ref = (0, _asyncToGenerator2["default"])( /*#__PURE__*/_regenerator["default"].mark(function _callee() {
    var _this = this;

    var _ref2,
        _ref2$prefix,
        prefix,
        proxyList,
        proxyTestsResults,
        _args = arguments;

    return _regenerator["default"].wrap(function _callee$(_context) {
      while (1) {
        switch (_context.prev = _context.next) {
          case 0:
            _ref2 = _args.length > 0 && _args[0] !== undefined ? _args[0] : {}, _ref2$prefix = _ref2.prefix, prefix = _ref2$prefix === void 0 ? 'bg' : _ref2$prefix;
            if (!this.__metrics) this.__metrics = {};

            if (!this.__metrics.gauge) {
              this.__metrics.gauge = new _promClient.Gauge({
                name: [prefix, 'proxy'].filter(Boolean).join('_'),
                help: 'Информация по проксям',
                labelNames: ['provider', 'type', 'country', 'ipv', 'worker']
              });
            }

            if (!this.__metrics.gaugeTargetCount) {
              this.__metrics.gaugeTargetCount = new _promClient.Gauge({
                name: [prefix, 'proxy_count'].filter(Boolean).join('_'),
                help: 'Информация по статусам проксям',
                labelNames: ['target', 'provider', 'status', 'worker']
              });
            }

            _context.prev = 4;
            _context.next = 7;
            return this.getAllProxyList();

          case 7:
            proxyList = _context.sent;
            // console.log({ proxyList });
            proxyList.forEach(function (proxy) {
              // console.log({ proxy });
              var _ref3 = proxy || {},
                  provider = _ref3.provider,
                  type = _ref3.type,
                  tags = _ref3.tags; // console.log({provider})


              var _ref4 = tags || {},
                  country = _ref4.country,
                  ipv = _ref4.ipv,
                  worker = _ref4.worker; // console.log({country})


              _this.__metrics.gauge.inc({
                provider: provider,
                type: type,
                country: country,
                ipv: ipv,
                worker: worker
              });
            });
            _context.next = 14;
            break;

          case 11:
            _context.prev = 11;
            _context.t0 = _context["catch"](4);
            this.log.error('[collectMetrics] getProxyList', _context.t0);

          case 14:
            _context.prev = 14;
            _context.next = 17;
            return this.getProxyTestsResults();

          case 17:
            proxyTestsResults = _context.sent;
            Object.values(proxyTestsResults).forEach(function () {
              var _ref5 = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : {},
                  test = _ref5.test,
                  proxy = _ref5.proxy,
                  time = _ref5.time,
                  status = _ref5.status;

              var _ref6 = proxy || {},
                  provider = _ref6.provider;

              var _ref7 = test || {},
                  _ref7$tags = _ref7.tags,
                  tags = _ref7$tags === void 0 ? [] : _ref7$tags;

              var _ref8 = proxy.tags || {},
                  worker = _ref8.worker;

              tags.forEach(function (target) {
                if (!target) return;

                _this.__metrics.gaugeTargetCount.inc({
                  target: target,
                  provider: provider,
                  status: status,
                  worker: worker
                });
              });
            });
            _context.next = 24;
            break;

          case 21:
            _context.prev = 21;
            _context.t1 = _context["catch"](14);
            this.log.error('[collectMetrics] getProxyTests', _context.t1);

          case 24:
            return _context.abrupt("return", this.__metrics);

          case 25:
          case "end":
            return _context.stop();
        }
      }
    }, _callee, this, [[4, 11], [14, 21]]);
  }));

  return function collectMetrics() {
    return _ref.apply(this, arguments);
  };
}();

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