1 | "use strict";
|
2 |
|
3 | var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
|
4 | Object.defineProperty(exports, "__esModule", {
|
5 | value: true
|
6 | });
|
7 | exports["default"] = exports.DbModule = void 0;
|
8 | var _regenerator = _interopRequireDefault(require("@babel/runtime/regenerator"));
|
9 | var _asyncToGenerator2 = _interopRequireDefault(require("@babel/runtime/helpers/asyncToGenerator"));
|
10 | var _defineProperty2 = _interopRequireDefault(require("@babel/runtime/helpers/defineProperty"));
|
11 | var _classCallCheck2 = _interopRequireDefault(require("@babel/runtime/helpers/classCallCheck"));
|
12 | var _createClass2 = _interopRequireDefault(require("@babel/runtime/helpers/createClass"));
|
13 | var _get2 = _interopRequireDefault(require("@babel/runtime/helpers/get"));
|
14 | var _inherits2 = _interopRequireDefault(require("@babel/runtime/helpers/inherits"));
|
15 | var _possibleConstructorReturn2 = _interopRequireDefault(require("@babel/runtime/helpers/possibleConstructorReturn"));
|
16 | var _getPrototypeOf2 = _interopRequireDefault(require("@babel/runtime/helpers/getPrototypeOf"));
|
17 | var _err = _interopRequireDefault(require("@lskjs/err"));
|
18 | var _module = _interopRequireDefault(require("@lskjs/module"));
|
19 | var _maskUriPassword = _interopRequireDefault(require("@lskjs/utils/maskUriPassword"));
|
20 | var _bluebird = _interopRequireDefault(require("bluebird"));
|
21 | var _pick = _interopRequireDefault(require("lodash/pick"));
|
22 | var _mongoose = require("mongoose");
|
23 | var _validOptionNames = require("./mongo/validOptionNames");
|
24 | 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; }
|
25 | 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; }
|
26 | 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); }; }
|
27 | 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; } }
|
28 |
|
29 | var DbModule = function (_Module) {
|
30 | (0, _inherits2["default"])(DbModule, _Module);
|
31 | var _super = _createSuper(DbModule);
|
32 | function DbModule() {
|
33 | var _this;
|
34 | (0, _classCallCheck2["default"])(this, DbModule);
|
35 | for (var _len = arguments.length, args = new Array(_len), _key = 0; _key < _len; _key++) {
|
36 | args[_key] = arguments[_key];
|
37 | }
|
38 | _this = _super.call.apply(_super, [this].concat(args));
|
39 | _this.config = {
|
40 | keepAlive: true,
|
41 | useUnifiedTopology: true,
|
42 | useNewUrlParser: true,
|
43 |
|
44 |
|
45 |
|
46 | connectTimeoutMS: 5000
|
47 |
|
48 | };
|
49 | _this.provider = 'mongo';
|
50 | _this.client = 'mongoose';
|
51 | _this.Mongoose = _mongoose.Mongoose;
|
52 | _this.reconnectios = 0;
|
53 | return _this;
|
54 | }
|
55 | (0, _createClass2["default"])(DbModule, [{
|
56 | key: "getOptions",
|
57 | value: function getOptions() {
|
58 | return _objectSpread(_objectSpread({}, (0, _pick["default"])(this.config, _validOptionNames.validOptionNames)), this.config.options || {});
|
59 | }
|
60 | }, {
|
61 | key: "init",
|
62 | value: function () {
|
63 | var _init = (0, _asyncToGenerator2["default"])( _regenerator["default"].mark(function _callee2() {
|
64 | var _this2 = this;
|
65 | var uri, options, Mongoose;
|
66 | return _regenerator["default"].wrap(function _callee2$(_context2) {
|
67 | while (1) switch (_context2.prev = _context2.next) {
|
68 | case 0:
|
69 | _context2.next = 2;
|
70 | return (0, _get2["default"])((0, _getPrototypeOf2["default"])(DbModule.prototype), "init", this).call(this);
|
71 | case 2:
|
72 | uri = this.config.uri;
|
73 | options = this.getOptions();
|
74 | if (uri) {
|
75 | _context2.next = 6;
|
76 | break;
|
77 | }
|
78 | throw new _err["default"]('!uri');
|
79 | case 6:
|
80 | if (this.debug) this.log.trace('config', (0, _maskUriPassword["default"])(uri), options);
|
81 |
|
82 |
|
83 | Mongoose = this.Mongoose;
|
84 | this.client = new Mongoose();
|
85 | this.client.Promise = Promise;
|
86 | this.client.connection.on('connected', function () {
|
87 | if (_this2.debug) _this2.log.trace('connected');
|
88 | _this2.reconnectios = 0;
|
89 | });
|
90 | this.client.connection.on('error', function () {
|
91 | var _ref = (0, _asyncToGenerator2["default"])( _regenerator["default"].mark(function _callee(err) {
|
92 | var interval;
|
93 | return _regenerator["default"].wrap(function _callee$(_context) {
|
94 | while (1) switch (_context.prev = _context.next) {
|
95 | case 0:
|
96 | _this2.log.error('error', err);
|
97 | interval = _this2.reconnectios++ * 2000 + 1000;
|
98 | if (_this2.debug) _this2.log.warn("reconnect [delay] ".concat(interval, "ms"));
|
99 | _context.next = 5;
|
100 | return _bluebird["default"].delay(interval);
|
101 | case 5:
|
102 | setTimeout(_this2.reconnect.bind(_this2), interval);
|
103 | case 6:
|
104 | case "end":
|
105 | return _context.stop();
|
106 | }
|
107 | }, _callee);
|
108 | }));
|
109 | return function (_x) {
|
110 | return _ref.apply(this, arguments);
|
111 | };
|
112 | }());
|
113 | this.client.connection.on('disconnected', function () {
|
114 | if (_this2.debug) _this2.log.trace('disconnected');
|
115 | });
|
116 | this.client.set('debug', this.debug || false);
|
117 | return _context2.abrupt("return", this.client);
|
118 | case 15:
|
119 | case "end":
|
120 | return _context2.stop();
|
121 | }
|
122 | }, _callee2, this);
|
123 | }));
|
124 | function init() {
|
125 | return _init.apply(this, arguments);
|
126 | }
|
127 | return init;
|
128 | }()
|
129 | }, {
|
130 | key: "stop",
|
131 | value: function () {
|
132 | var _stop = (0, _asyncToGenerator2["default"])( _regenerator["default"].mark(function _callee3() {
|
133 | return _regenerator["default"].wrap(function _callee3$(_context3) {
|
134 | while (1) switch (_context3.prev = _context3.next) {
|
135 | case 0:
|
136 | _context3.next = 2;
|
137 | return (0, _get2["default"])((0, _getPrototypeOf2["default"])(DbModule.prototype), "stop", this).call(this);
|
138 | case 2:
|
139 | if (this.client) {
|
140 | if (this.client.connection) this.client.connection.close();
|
141 | this.client.disconnect();
|
142 | } else {
|
143 | this.log.warn('!client', 'cannot client.disconnect');
|
144 | }
|
145 | return _context3.abrupt("return", _bluebird["default"].delay(10));
|
146 | case 4:
|
147 | case "end":
|
148 | return _context3.stop();
|
149 | }
|
150 | }, _callee3, this);
|
151 | }));
|
152 | function stop() {
|
153 | return _stop.apply(this, arguments);
|
154 | }
|
155 | return stop;
|
156 | }()
|
157 | }, {
|
158 | key: "reconnect",
|
159 | value: function () {
|
160 | var _reconnect = (0, _asyncToGenerator2["default"])( _regenerator["default"].mark(function _callee4() {
|
161 | return _regenerator["default"].wrap(function _callee4$(_context4) {
|
162 | while (1) switch (_context4.prev = _context4.next) {
|
163 | case 0:
|
164 | if (this.debug) this.log.trace('reconnect');
|
165 | if (this.client) {
|
166 | this.client.stop();
|
167 | this.client.run();
|
168 | } else {
|
169 | this.log.warn('!client', 'cannot client.stop and client.run');
|
170 | }
|
171 | case 2:
|
172 | case "end":
|
173 | return _context4.stop();
|
174 | }
|
175 | }, _callee4, this);
|
176 | }));
|
177 | function reconnect() {
|
178 | return _reconnect.apply(this, arguments);
|
179 | }
|
180 | return reconnect;
|
181 | }()
|
182 | }, {
|
183 | key: "run",
|
184 | value: function () {
|
185 | var _run = (0, _asyncToGenerator2["default"])( _regenerator["default"].mark(function _callee5() {
|
186 | var _this3 = this;
|
187 | var uri, options, connection;
|
188 | return _regenerator["default"].wrap(function _callee5$(_context5) {
|
189 | while (1) switch (_context5.prev = _context5.next) {
|
190 | case 0:
|
191 | _context5.next = 2;
|
192 | return (0, _get2["default"])((0, _getPrototypeOf2["default"])(DbModule.prototype), "run", this).call(this);
|
193 | case 2:
|
194 | this.reconnectios = 0;
|
195 | uri = this.config.uri;
|
196 | options = this.getOptions();
|
197 | _context5.next = 7;
|
198 | return new Promise(function (resolve, reject) {
|
199 | return _this3.client.connect(uri, options).then(resolve, reject);
|
200 | });
|
201 | case 7:
|
202 | connection = _context5.sent;
|
203 | this.log.debug('[ready]', (0, _maskUriPassword["default"])(uri));
|
204 | return _context5.abrupt("return", connection);
|
205 | case 10:
|
206 | case "end":
|
207 | return _context5.stop();
|
208 | }
|
209 | }, _callee5, this);
|
210 | }));
|
211 | function run() {
|
212 | return _run.apply(this, arguments);
|
213 | }
|
214 | return run;
|
215 | }()
|
216 | }, {
|
217 | key: "healthcheck",
|
218 | value: function () {
|
219 | var _healthcheck = (0, _asyncToGenerator2["default"])( _regenerator["default"].mark(function _callee6() {
|
220 | return _regenerator["default"].wrap(function _callee6$(_context6) {
|
221 | while (1) switch (_context6.prev = _context6.next) {
|
222 | case 0:
|
223 | if (this.client) {
|
224 | _context6.next = 2;
|
225 | break;
|
226 | }
|
227 | throw '!client';
|
228 | case 2:
|
229 | return _context6.abrupt("return", this.client.connection.db.command({
|
230 | ping: 1
|
231 | }).then(function (_ref2) {
|
232 | var operationTime = _ref2.operationTime;
|
233 | return operationTime;
|
234 | }));
|
235 | case 3:
|
236 | case "end":
|
237 | return _context6.stop();
|
238 | }
|
239 | }, _callee6, this);
|
240 | }));
|
241 | function healthcheck() {
|
242 | return _healthcheck.apply(this, arguments);
|
243 | }
|
244 | return healthcheck;
|
245 | }()
|
246 | }]);
|
247 | return DbModule;
|
248 | }(_module["default"]);
|
249 | exports.DbModule = DbModule;
|
250 | var _default = DbModule;
|
251 | exports["default"] = _default;
|
252 |
|
\ | No newline at end of file |