@lskjs/worker/lib/WorkerApp.js

"use strict";

var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
Object.defineProperty(exports, "__esModule", {
  value: true
});
exports["default"] = exports.WorkerApp = void 0;
var _regenerator = _interopRequireDefault(require("@babel/runtime/regenerator"));
var _defineProperty2 = _interopRequireDefault(require("@babel/runtime/helpers/defineProperty"));
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 _err = _interopRequireDefault(require("@lskjs/err"));
var _arrayToObject = _interopRequireDefault(require("@lskjs/utils/arrayToObject"));
var _asyncMapValues = _interopRequireDefault(require("@lskjs/utils/asyncMapValues"));
var _importFn = _interopRequireDefault(require("@lskjs/utils/importFn"));
var _get3 = _interopRequireDefault(require("lodash/get"));
var _createErrorInfo = require("./createErrorInfo");
var _SampleApp2 = require("./SampleApp");
function ownKeys(object, enumerableOnly) { var keys = Object.keys(object); if (Object.getOwnPropertySymbols) { var symbols = Object.getOwnPropertySymbols(object); enumerableOnly && (symbols = symbols.filter(function (sym) { return Object.getOwnPropertyDescriptor(object, sym).enumerable; })), keys.push.apply(keys, symbols); } return keys; }
function _objectSpread(target) { for (var i = 1; i < arguments.length; i++) { var source = null != arguments[i] ? arguments[i] : {}; i % 2 ? ownKeys(Object(source), !0).forEach(function (key) { (0, _defineProperty2["default"])(target, key, source[key]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(target, Object.getOwnPropertyDescriptors(source)) : ownKeys(Object(source)).forEach(function (key) { Object.defineProperty(target, key, Object.getOwnPropertyDescriptor(source, key)); }); } return target; }
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; } } // import { isDev } from '@lskjs/utils/env';
var WorkerApp = /*#__PURE__*/function (_SampleApp) {
  (0, _inherits2["default"])(WorkerApp, _SampleApp);
  var _super = _createSuper(WorkerApp);
  function WorkerApp() {
    var _this;
    (0, _classCallCheck2["default"])(this, WorkerApp);
    for (var _len = arguments.length, args = new Array(_len), _key = 0; _key < _len; _key++) {
      args[_key] = arguments[_key];
    }
    _this = _super.call.apply(_super, [this].concat(args));
    _this.preload = true;
    _this.initedWorkers = {};
    return _this;
  }
  (0, _createClass2["default"])(WorkerApp, [{
    key: "model",
    value: function () {
      var _model = (0, _asyncToGenerator2["default"])( /*#__PURE__*/_regenerator["default"].mark(function _callee() {
        var _this$log;
        var _len2,
          args,
          _key2,
          modelsModule,
          _args = arguments;
        return _regenerator["default"].wrap(function _callee$(_context) {
          while (1) switch (_context.prev = _context.next) {
            case 0:
              for (_len2 = _args.length, args = new Array(_len2), _key2 = 0; _key2 < _len2; _key2++) {
                args[_key2] = _args[_key2];
              }
              (_this$log = this.log).warn.apply(_this$log, ['app.model deprecated'].concat(args));
              _context.next = 4;
              return this.module('models');
            case 4:
              modelsModule = _context.sent;
              return _context.abrupt("return", modelsModule.model.apply(modelsModule, args));
            case 6:
            case "end":
              return _context.stop();
          }
        }, _callee, this);
      }));
      function model() {
        return _model.apply(this, arguments);
      }
      return model;
    }()
  }, {
    key: "setProp",
    value: function setProp(name, value) {
      if (name === 'workers') return (0, _get2["default"])((0, _getPrototypeOf2["default"])(WorkerApp.prototype), "setProp", this).call(this, 'availableWorkers', value);
      return (0, _get2["default"])((0, _getPrototypeOf2["default"])(WorkerApp.prototype), "setProp", this).call(this, name, value);
    }
  }, {
    key: "init",
    value: function () {
      var _init = (0, _asyncToGenerator2["default"])( /*#__PURE__*/_regenerator["default"].mark(function _callee2() {
        return _regenerator["default"].wrap(function _callee2$(_context2) {
          while (1) switch (_context2.prev = _context2.next) {
            case 0:
              _context2.next = 2;
              return (0, _get2["default"])((0, _getPrototypeOf2["default"])(WorkerApp.prototype), "init", this).call(this);
            case 2:
              if (!this.getErrorInfo) this.getErrorInfo = (0, _createErrorInfo.createErrorInfo)(this.errors);
            case 3:
            case "end":
              return _context2.stop();
          }
        }, _callee2, this);
      }));
      function init() {
        return _init.apply(this, arguments);
      }
      return init;
    }()
  }, {
    key: "isWorkerClass",
    value: function isWorkerClass(WorkerClass) {
      return !!(WorkerClass && WorkerClass.__worker);
    }
  }, {
    key: "isWorkerJob",
    value: function isWorkerJob(WorkerJob) {
      return !!(WorkerJob && WorkerJob.__workerJob);
    }
  }, {
    key: "worker",
    value: function () {
      var _worker = (0, _asyncToGenerator2["default"])( /*#__PURE__*/_regenerator["default"].mark(function _callee3(nameOrNames) {
        var _this2 = this;
        var name, CurrentWorker, config, worker;
        return _regenerator["default"].wrap(function _callee3$(_context3) {
          while (1) switch (_context3.prev = _context3.next) {
            case 0:
              // eslint-disable-next-line no-param-reassign
              if (nameOrNames === '*') nameOrNames = Object.keys(this.availableWorkers || {});
              if (!Array.isArray(nameOrNames)) {
                _context3.next = 3;
                break;
              }
              return _context3.abrupt("return", (0, _asyncMapValues["default"])((0, _arrayToObject["default"])(nameOrNames), function (n) {
                return _this2.module(n);
              }));
            case 3:
              name = nameOrNames;
              if (!this.initedWorkers[name]) {
                _context3.next = 6;
                break;
              }
              return _context3.abrupt("return", this.initedWorkers[name]);
            case 6:
              if (this.availableWorkers[name]) {
                _context3.next = 8;
                break;
              }
              throw new _err["default"]('WORKER_NOT_FOUND', "In worker \"".concat(name, "\" not found index with exported default function"));
            case 8:
              _context3.next = 10;
              return (0, _importFn["default"])(this.availableWorkers[name]);
            case 10:
              CurrentWorker = _context3.sent;
              _context3.next = 13;
              return this.getWorkerConfig(name);
            case 13:
              config = _context3.sent;
              if (this.isWorkerClass(CurrentWorker)) {
                _context3.next = 16;
                break;
              }
              throw new _err["default"]('notInstanceOfWorker', {
                data: {
                  name: name
                }
              });
            case 16:
              _context3.next = 18;
              return CurrentWorker.start({
                app: this,
                rabbit: this.rabbit,
                config: config
              });
            case 18:
              worker = _context3.sent;
              this.initedWorkers[name] = worker;
              return _context3.abrupt("return", worker);
            case 21:
            case "end":
              return _context3.stop();
          }
        }, _callee3, this);
      }));
      function worker(_x) {
        return _worker.apply(this, arguments);
      }
      return worker;
    }()
  }, {
    key: "getWorkerConfig",
    value: function () {
      var _getWorkerConfig = (0, _asyncToGenerator2["default"])( /*#__PURE__*/_regenerator["default"].mark(function _callee4(name) {
        var mConfig, wConfig, nConfig, res;
        return _regenerator["default"].wrap(function _callee4$(_context4) {
          while (1) switch (_context4.prev = _context4.next) {
            case 0:
              _context4.next = 2;
              return this.getModuleConfig(name);
            case 2:
              mConfig = _context4.sent;
              wConfig = (0, _get3["default"])(this, 'config.worker') || {};
              if (!(typeof wConfig === 'string')) {
                _context4.next = 6;
                break;
              }
              throw new _err["default"]('DEPRECATED config.worker === string');
            case 6:
              nConfig = (0, _get3["default"])(this, "config.workers.".concat(name)) || {};
              res = _objectSpread(_objectSpread(_objectSpread(_objectSpread({}, mConfig || {}), wConfig || {}), nConfig || {}), {}, {
                // ns: 'worker',
                log: _objectSpread(_objectSpread(_objectSpread(_objectSpread({}, (mConfig === null || mConfig === void 0 ? void 0 : mConfig.log) || {}), (wConfig === null || wConfig === void 0 ? void 0 : wConfig.log) || {}), (nConfig === null || nConfig === void 0 ? void 0 : nConfig.log) || {}), {}, {
                  ns: 'worker'
                })
              });
              return _context4.abrupt("return", res);
            case 9:
            case "end":
              return _context4.stop();
          }
        }, _callee4, this);
      }));
      function getWorkerConfig(_x2) {
        return _getWorkerConfig.apply(this, arguments);
      }
      return getWorkerConfig;
    }()
  }, {
    key: "runWorkers",
    value: function () {
      var _runWorkers = (0, _asyncToGenerator2["default"])( /*#__PURE__*/_regenerator["default"].mark(function _callee6() {
        var _this3 = this;
        var config, workerKey, workerNames;
        return _regenerator["default"].wrap(function _callee6$(_context6) {
          while (1) switch (_context6.prev = _context6.next) {
            case 0:
              _context6.next = 2;
              return this.getWorkerConfig();
            case 2:
              config = _context6.sent;
              workerKey = config.name;
              if (!(!workerKey || typeof workerKey !== 'string')) {
                _context6.next = 7;
                break;
              }
              this.log.warn('!config.worker', '[ignore]');
              return _context6.abrupt("return");
            case 7:
              this.log.trace("runWorkers(".concat(workerKey, ")"));
              if (workerKey === '*') {
                workerNames = Object.keys(this.availableWorkers || {});
              } else if (workerKey) {
                workerNames = workerKey.split(',').map(function (w) {
                  return w.trim();
                });
              }
              _context6.next = 11;
              return (0, _asyncMapValues["default"])(workerNames, /*#__PURE__*/function () {
                var _ref = (0, _asyncToGenerator2["default"])( /*#__PURE__*/_regenerator["default"].mark(function _callee5(name) {
                  return _regenerator["default"].wrap(function _callee5$(_context5) {
                    while (1) switch (_context5.prev = _context5.next) {
                      case 0:
                        return _context5.abrupt("return", _this3.worker(name));
                      case 1:
                      case "end":
                        return _context5.stop();
                    }
                  }, _callee5);
                }));
                return function (_x3) {
                  return _ref.apply(this, arguments);
                };
              }());
            case 11:
            case "end":
              return _context6.stop();
          }
        }, _callee6, this);
      }));
      function runWorkers() {
        return _runWorkers.apply(this, arguments);
      }
      return runWorkers;
    }()
  }, {
    key: "run",
    value: function () {
      var _run = (0, _asyncToGenerator2["default"])( /*#__PURE__*/_regenerator["default"].mark(function _callee7() {
        return _regenerator["default"].wrap(function _callee7$(_context7) {
          while (1) switch (_context7.prev = _context7.next) {
            case 0:
              _context7.next = 2;
              return (0, _get2["default"])((0, _getPrototypeOf2["default"])(WorkerApp.prototype), "run", this).call(this);
            case 2:
              _context7.next = 4;
              return this.runWorkers();
            case 4:
            case "end":
              return _context7.stop();
          }
        }, _callee7, this);
      }));
      function run() {
        return _run.apply(this, arguments);
      }
      return run;
    }()
  }]);
  return WorkerApp;
}(_SampleApp2.SampleApp);
exports.WorkerApp = WorkerApp;
var _default = WorkerApp;
exports["default"] = _default;
//# sourceMappingURL=WorkerApp.js.map