@lskjs/db/DbModule.js

"use strict";

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

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

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

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

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

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 _module = _interopRequireDefault(require("@lskjs/module"));

var _maskUriPassword = _interopRequireDefault(require("@lskjs/utils/maskUriPassword"));

var _bluebird = _interopRequireDefault(require("bluebird"));

var _pick = _interopRequireDefault(require("lodash/pick"));

var _mongoose = require("mongoose");

var _validOptionNames = require("./mongo/validOptionNames");

function ownKeys(object, enumerableOnly) { var keys = Object.keys(object); if (Object.getOwnPropertySymbols) { var symbols = Object.getOwnPropertySymbols(object); if (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 = arguments[i] != null ? arguments[i] : {}; if (i % 2) { ownKeys(Object(source), true).forEach(function (key) { (0, _defineProperty2["default"])(target, key, source[key]); }); } else if (Object.getOwnPropertyDescriptors) { Object.defineProperties(target, Object.getOwnPropertyDescriptors(source)); } else { 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; } }

// abstract
var DbModule = /*#__PURE__*/function (_Module) {
  (0, _inherits2["default"])(DbModule, _Module);

  var _super = _createSuper(DbModule);

  function DbModule() {
    var _this;

    (0, _classCallCheck2["default"])(this, DbModule);

    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.config = {
      keepAlive: true,
      useUnifiedTopology: true,
      useNewUrlParser: true,
      // autoReconnect: true,
      // reconnectTries: __DEV__ ? 10000 : 30,
      // reconnectInterval: __DEV__ ? 30000 : 1000,
      connectTimeoutMS: 5000 // sets the delay between every retry (milliseconds)

    };
    _this.provider = 'mongo';
    _this.client = 'mongoose';
    _this.Mongoose = _mongoose.Mongoose;
    _this.reconnectios = 0;
    return _this;
  }

  (0, _createClass2["default"])(DbModule, [{
    key: "getOptions",
    value: function getOptions() {
      return _objectSpread(_objectSpread({}, (0, _pick["default"])(this.config, _validOptionNames.validOptionNames)), this.config.options || {});
    }
  }, {
    key: "init",
    value: function () {
      var _init = (0, _asyncToGenerator2["default"])( /*#__PURE__*/_regenerator["default"].mark(function _callee2() {
        var _this2 = this;

        var uri, options, Mongoose;
        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"])(DbModule.prototype), "init", this).call(this);

              case 2:
                uri = this.config.uri;
                options = this.getOptions();

                if (uri) {
                  _context2.next = 6;
                  break;
                }

                throw new _err["default"]('!uri');

              case 6:
                if (this.debug) this.log.trace('config', (0, _maskUriPassword["default"])(uri), options); // finalOptions
                // eslint-disable-next-line no-shadow

                Mongoose = this.Mongoose;
                this.client = new Mongoose();
                this.client.Promise = Promise;
                this.client.connection.on('connected', function () {
                  if (_this2.debug) _this2.log.trace('connected');
                  _this2.reconnectios = 0;
                });
                this.client.connection.on('error', /*#__PURE__*/function () {
                  var _ref = (0, _asyncToGenerator2["default"])( /*#__PURE__*/_regenerator["default"].mark(function _callee(err) {
                    var interval;
                    return _regenerator["default"].wrap(function _callee$(_context) {
                      while (1) {
                        switch (_context.prev = _context.next) {
                          case 0:
                            _this2.log.error('error', err);

                            interval = _this2.reconnectios++ * 2000 + 1000; // eslint-disable-line no-plusplus

                            if (_this2.debug) _this2.log.warn("reconnect [delay] ".concat(interval, "ms"));
                            _context.next = 5;
                            return _bluebird["default"].delay(interval);

                          case 5:
                            setTimeout(_this2.reconnect.bind(_this2), interval);

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

                  return function (_x) {
                    return _ref.apply(this, arguments);
                  };
                }());
                this.client.connection.on('disconnected', function () {
                  if (_this2.debug) _this2.log.trace('disconnected');
                });
                this.client.set('debug', this.debug || false);
                return _context2.abrupt("return", this.client);

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

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

      return init;
    }()
  }, {
    key: "stop",
    value: function () {
      var _stop = (0, _asyncToGenerator2["default"])( /*#__PURE__*/_regenerator["default"].mark(function _callee3() {
        return _regenerator["default"].wrap(function _callee3$(_context3) {
          while (1) {
            switch (_context3.prev = _context3.next) {
              case 0:
                _context3.next = 2;
                return (0, _get2["default"])((0, _getPrototypeOf2["default"])(DbModule.prototype), "stop", this).call(this);

              case 2:
                if (this.client) {
                  if (this.client.connection) this.client.connection.close();
                  this.client.disconnect();
                } else {
                  this.log.warn('!client', 'cannot client.disconnect');
                }

                return _context3.abrupt("return", _bluebird["default"].delay(10));

              case 4:
              case "end":
                return _context3.stop();
            }
          }
        }, _callee3, this);
      }));

      function stop() {
        return _stop.apply(this, arguments);
      }

      return stop;
    }()
  }, {
    key: "reconnect",
    value: function () {
      var _reconnect = (0, _asyncToGenerator2["default"])( /*#__PURE__*/_regenerator["default"].mark(function _callee4() {
        return _regenerator["default"].wrap(function _callee4$(_context4) {
          while (1) {
            switch (_context4.prev = _context4.next) {
              case 0:
                if (this.debug) this.log.trace('reconnect');

                if (this.client) {
                  this.client.stop();
                  this.client.run();
                } else {
                  this.log.warn('!client', 'cannot client.stop and client.run');
                }

              case 2:
              case "end":
                return _context4.stop();
            }
          }
        }, _callee4, this);
      }));

      function reconnect() {
        return _reconnect.apply(this, arguments);
      }

      return reconnect;
    }()
  }, {
    key: "run",
    value: function () {
      var _run = (0, _asyncToGenerator2["default"])( /*#__PURE__*/_regenerator["default"].mark(function _callee5() {
        var _this3 = this;

        var uri, options, connection;
        return _regenerator["default"].wrap(function _callee5$(_context5) {
          while (1) {
            switch (_context5.prev = _context5.next) {
              case 0:
                _context5.next = 2;
                return (0, _get2["default"])((0, _getPrototypeOf2["default"])(DbModule.prototype), "run", this).call(this);

              case 2:
                this.reconnectios = 0;
                uri = this.config.uri;
                options = this.getOptions();
                _context5.next = 7;
                return new Promise(function (resolve, reject) {
                  return _this3.client.connect(uri, options).then(resolve, reject);
                });

              case 7:
                connection = _context5.sent;
                this.log.debug('[ready]', (0, _maskUriPassword["default"])(uri));
                return _context5.abrupt("return", connection);

              case 10:
              case "end":
                return _context5.stop();
            }
          }
        }, _callee5, this);
      }));

      function run() {
        return _run.apply(this, arguments);
      }

      return run;
    }()
  }, {
    key: "healthcheck",
    value: function () {
      var _healthcheck = (0, _asyncToGenerator2["default"])( /*#__PURE__*/_regenerator["default"].mark(function _callee6() {
        return _regenerator["default"].wrap(function _callee6$(_context6) {
          while (1) {
            switch (_context6.prev = _context6.next) {
              case 0:
                if (this.client) {
                  _context6.next = 2;
                  break;
                }

                throw '!client';

              case 2:
                return _context6.abrupt("return", this.client.connection.db.command({
                  ping: 1
                }).then(function (_ref2) {
                  var operationTime = _ref2.operationTime;
                  return operationTime;
                }));

              case 3:
              case "end":
                return _context6.stop();
            }
          }
        }, _callee6, this);
      }));

      function healthcheck() {
        return _healthcheck.apply(this, arguments);
      }

      return healthcheck;
    }()
  }]);
  return DbModule;
}(_module["default"]);

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