@lskjs/db/lib/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 = _interopRequireDefault(require("mongoose"));
var _validOptionNames = require("./mongo/validOptionNames");
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; } }
require('mongoose-long')(_mongoose["default"]);
var Mongoose = _mongoose["default"].Mongoose;
// 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;
    _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