UNPKG

dynamodb-dashboard

Version:
309 lines (305 loc) 22.6 kB
"use strict"; Object.defineProperty(exports, "__esModule", { value: true }); exports["default"] = void 0; var _aws = require("../config/aws"); var _event = require("../constants/event"); var _item = _interopRequireDefault(require("./item.service")); var _table = _interopRequireDefault(require("./table.service")); var _dynamodb = require("../constants/dynamodb"); function _interopRequireDefault(e) { return e && e.__esModule ? e : { "default": e }; } function _typeof(o) { "@babel/helpers - typeof"; return _typeof = "function" == typeof Symbol && "symbol" == typeof Symbol.iterator ? function (o) { return typeof o; } : function (o) { return o && "function" == typeof Symbol && o.constructor === Symbol && o !== Symbol.prototype ? "symbol" : typeof o; }, _typeof(o); } function _slicedToArray(r, e) { return _arrayWithHoles(r) || _iterableToArrayLimit(r, e) || _unsupportedIterableToArray(r, e) || _nonIterableRest(); } function _nonIterableRest() { throw new TypeError("Invalid attempt to destructure non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); } function _iterableToArrayLimit(r, l) { var t = null == r ? null : "undefined" != typeof Symbol && r[Symbol.iterator] || r["@@iterator"]; if (null != t) { var e, n, i, u, a = [], f = !0, o = !1; try { if (i = (t = t.call(r)).next, 0 === l) { if (Object(t) !== t) return; f = !1; } else for (; !(f = (e = i.call(t)).done) && (a.push(e.value), a.length !== l); f = !0); } catch (r) { o = !0, n = r; } finally { try { if (!f && null != t["return"] && (u = t["return"](), Object(u) !== u)) return; } finally { if (o) throw n; } } return a; } } function _arrayWithHoles(r) { if (Array.isArray(r)) return r; } function _toConsumableArray(r) { return _arrayWithoutHoles(r) || _iterableToArray(r) || _unsupportedIterableToArray(r) || _nonIterableSpread(); } function _nonIterableSpread() { throw new TypeError("Invalid attempt to spread non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); } function _unsupportedIterableToArray(r, a) { if (r) { if ("string" == typeof r) return _arrayLikeToArray(r, a); var t = {}.toString.call(r).slice(8, -1); return "Object" === t && r.constructor && (t = r.constructor.name), "Map" === t || "Set" === t ? Array.from(r) : "Arguments" === t || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(t) ? _arrayLikeToArray(r, a) : void 0; } } function _iterableToArray(r) { if ("undefined" != typeof Symbol && null != r[Symbol.iterator] || null != r["@@iterator"]) return Array.from(r); } function _arrayWithoutHoles(r) { if (Array.isArray(r)) return _arrayLikeToArray(r); } function _arrayLikeToArray(r, a) { (null == a || a > r.length) && (a = r.length); for (var e = 0, n = Array(a); e < a; e++) n[e] = r[e]; return n; } function _regeneratorRuntime() { "use strict"; var r = _regenerator(), e = r.m(_regeneratorRuntime), t = (Object.getPrototypeOf ? Object.getPrototypeOf(e) : e.__proto__).constructor; function n(r) { var e = "function" == typeof r && r.constructor; return !!e && (e === t || "GeneratorFunction" === (e.displayName || e.name)); } var o = { "throw": 1, "return": 2, "break": 3, "continue": 3 }; function a(r) { var e, t; return function (n) { e || (e = { stop: function stop() { return t(n.a, 2); }, "catch": function _catch() { return n.v; }, abrupt: function abrupt(r, e) { return t(n.a, o[r], e); }, delegateYield: function delegateYield(r, o, a) { return e.resultName = o, t(n.d, _regeneratorValues(r), a); }, finish: function finish(r) { return t(n.f, r); } }, t = function t(r, _t2, o) { n.p = e.prev, n.n = e.next; try { return r(_t2, o); } finally { e.next = n.n; } }), e.resultName && (e[e.resultName] = n.v, e.resultName = void 0), e.sent = n.v, e.next = n.n; try { return r.call(this, e); } finally { n.p = e.prev, n.n = e.next; } }; } return (_regeneratorRuntime = function _regeneratorRuntime() { return { wrap: function wrap(e, t, n, o) { return r.w(a(e), t, n, o && o.reverse()); }, isGeneratorFunction: n, mark: r.m, awrap: function awrap(r, e) { return new _OverloadYield(r, e); }, AsyncIterator: _regeneratorAsyncIterator, async: function async(r, e, t, o, u) { return (n(e) ? _regeneratorAsyncGen : _regeneratorAsync)(a(r), e, t, o, u); }, keys: _regeneratorKeys, values: _regeneratorValues }; })(); } function _regeneratorValues(e) { if (null != e) { var t = e["function" == typeof Symbol && Symbol.iterator || "@@iterator"], r = 0; if (t) return t.call(e); if ("function" == typeof e.next) return e; if (!isNaN(e.length)) return { next: function next() { return e && r >= e.length && (e = void 0), { value: e && e[r++], done: !e }; } }; } throw new TypeError(_typeof(e) + " is not iterable"); } function _regeneratorKeys(e) { var n = Object(e), r = []; for (var t in n) r.unshift(t); return function e() { for (; r.length;) if ((t = r.pop()) in n) return e.value = t, e.done = !1, e; return e.done = !0, e; }; } function _regeneratorAsync(n, e, r, t, o) { var a = _regeneratorAsyncGen(n, e, r, t, o); return a.next().then(function (n) { return n.done ? n.value : a.next(); }); } function _regeneratorAsyncGen(r, e, t, o, n) { return new _regeneratorAsyncIterator(_regenerator().w(r, e, t, o), n || Promise); } function _regeneratorAsyncIterator(t, e) { function n(r, o, i, f) { try { var c = t[r](o), u = c.value; return u instanceof _OverloadYield ? e.resolve(u.v).then(function (t) { n("next", t, i, f); }, function (t) { n("throw", t, i, f); }) : e.resolve(u).then(function (t) { c.value = t, i(c); }, function (t) { return n("throw", t, i, f); }); } catch (t) { f(t); } } var r; this.next || (_regeneratorDefine2(_regeneratorAsyncIterator.prototype), _regeneratorDefine2(_regeneratorAsyncIterator.prototype, "function" == typeof Symbol && Symbol.asyncIterator || "@asyncIterator", function () { return this; })), _regeneratorDefine2(this, "_invoke", function (t, o, i) { function f() { return new e(function (e, r) { n(t, i, e, r); }); } return r = r ? r.then(f, f) : f(); }, !0); } function _regenerator() { /*! regenerator-runtime -- Copyright (c) 2014-present, Facebook, Inc. -- license (MIT): https://github.com/babel/babel/blob/main/packages/babel-helpers/LICENSE */ var e, t, r = "function" == typeof Symbol ? Symbol : {}, n = r.iterator || "@@iterator", o = r.toStringTag || "@@toStringTag"; function i(r, n, o, i) { var c = n && n.prototype instanceof Generator ? n : Generator, u = Object.create(c.prototype); return _regeneratorDefine2(u, "_invoke", function (r, n, o) { var i, c, u, f = 0, p = o || [], y = !1, G = { p: 0, n: 0, v: e, a: d, f: d.bind(e, 4), d: function d(t, r) { return i = t, c = 0, u = e, G.n = r, a; } }; function d(r, n) { for (c = r, u = n, t = 0; !y && f && !o && t < p.length; t++) { var o, i = p[t], d = G.p, l = i[2]; r > 3 ? (o = l === n) && (c = i[4] || 3, u = i[5] === e ? i[3] : i[5], i[4] = 3, i[5] = e) : i[0] <= d && ((o = r < 2 && d < i[1]) ? (c = 0, G.v = n, G.n = i[1]) : d < l && (o = r < 3 || i[0] > n || n > l) && (i[4] = r, i[5] = n, G.n = l, c = 0)); } if (o || r > 1) return a; throw y = !0, n; } return function (o, p, l) { if (f > 1) throw TypeError("Generator is already running"); for (y && 1 === p && d(p, l), c = p, u = l; (t = c < 2 ? e : u) || !y;) { i || (c ? c < 3 ? (c > 1 && (G.n = -1), d(c, u)) : G.n = u : G.v = u); try { if (f = 2, i) { if (c || (o = "next"), t = i[o]) { if (!(t = t.call(i, u))) throw TypeError("iterator result is not an object"); if (!t.done) return t; u = t.value, c < 2 && (c = 0); } else 1 === c && (t = i["return"]) && t.call(i), c < 2 && (u = TypeError("The iterator does not provide a '" + o + "' method"), c = 1); i = e; } else if ((t = (y = G.n < 0) ? u : r.call(n, G)) !== a) break; } catch (t) { i = e, c = 1, u = t; } finally { f = 1; } } return { value: t, done: y }; }; }(r, o, i), !0), u; } var a = {}; function Generator() {} function GeneratorFunction() {} function GeneratorFunctionPrototype() {} t = Object.getPrototypeOf; var c = [][n] ? t(t([][n]())) : (_regeneratorDefine2(t = {}, n, function () { return this; }), t), u = GeneratorFunctionPrototype.prototype = Generator.prototype = Object.create(c); function f(e) { return Object.setPrototypeOf ? Object.setPrototypeOf(e, GeneratorFunctionPrototype) : (e.__proto__ = GeneratorFunctionPrototype, _regeneratorDefine2(e, o, "GeneratorFunction")), e.prototype = Object.create(u), e; } return GeneratorFunction.prototype = GeneratorFunctionPrototype, _regeneratorDefine2(u, "constructor", GeneratorFunctionPrototype), _regeneratorDefine2(GeneratorFunctionPrototype, "constructor", GeneratorFunction), GeneratorFunction.displayName = "GeneratorFunction", _regeneratorDefine2(GeneratorFunctionPrototype, o, "GeneratorFunction"), _regeneratorDefine2(u), _regeneratorDefine2(u, o, "Generator"), _regeneratorDefine2(u, n, function () { return this; }), _regeneratorDefine2(u, "toString", function () { return "[object Generator]"; }), (_regenerator = function _regenerator() { return { w: i, m: f }; })(); } function _regeneratorDefine2(e, r, n, t) { var i = Object.defineProperty; try { i({}, "", {}); } catch (e) { i = 0; } _regeneratorDefine2 = function _regeneratorDefine(e, r, n, t) { if (r) i ? i(e, r, { value: n, enumerable: !t, configurable: !t, writable: !t }) : e[r] = n;else { var o = function o(r, n) { _regeneratorDefine2(e, r, function (e) { return this._invoke(r, n, e); }); }; o("next", 0), o("throw", 1), o("return", 2); } }, _regeneratorDefine2(e, r, n, t); } function _OverloadYield(e, d) { this.v = e, this.k = d; } function _interopRequireWildcard(e, t) { if ("function" == typeof WeakMap) var r = new WeakMap(), n = new WeakMap(); return (_interopRequireWildcard = function _interopRequireWildcard(e, t) { if (!t && e && e.__esModule) return e; var o, i, f = { __proto__: null, "default": e }; if (null === e || "object" != _typeof(e) && "function" != typeof e) return f; if (o = t ? n : r) { if (o.has(e)) return o.get(e); o.set(e, f); } for (var _t in e) "default" !== _t && {}.hasOwnProperty.call(e, _t) && ((i = (o = Object.defineProperty) && Object.getOwnPropertyDescriptor(e, _t)) && (i.get || i.set) ? o(f, _t, i) : f[_t] = e[_t]); return f; })(e, t); } function asyncGeneratorStep(n, t, e, r, o, a, c) { try { var i = n[a](c), u = i.value; } catch (n) { return void e(n); } i.done ? t(u) : Promise.resolve(u).then(r, o); } function _asyncToGenerator(n) { return function () { var t = this, e = arguments; return new Promise(function (r, o) { var a = n.apply(t, e); function _next(n) { asyncGeneratorStep(a, r, o, _next, _throw, "next", n); } function _throw(n) { asyncGeneratorStep(a, r, o, _next, _throw, "throw", n); } _next(void 0); }); }; } function _classCallCheck(a, n) { if (!(a instanceof n)) throw new TypeError("Cannot call a class as a function"); } function _defineProperties(e, r) { for (var t = 0; t < r.length; t++) { var o = r[t]; o.enumerable = o.enumerable || !1, o.configurable = !0, "value" in o && (o.writable = !0), Object.defineProperty(e, _toPropertyKey(o.key), o); } } function _createClass(e, r, t) { return r && _defineProperties(e.prototype, r), t && _defineProperties(e, t), Object.defineProperty(e, "prototype", { writable: !1 }), e; } function _toPropertyKey(t) { var i = _toPrimitive(t, "string"); return "symbol" == _typeof(i) ? i : i + ""; } function _toPrimitive(t, r) { if ("object" != _typeof(t) || !t) return t; var e = t[Symbol.toPrimitive]; if (void 0 !== e) { var i = e.call(t, r || "default"); if ("object" != _typeof(i)) return i; throw new TypeError("@@toPrimitive must return a primitive value."); } return ("string" === r ? String : Number)(t); } function _asyncIterator(r) { var n, t, o, e = 2; for ("undefined" != typeof Symbol && (t = Symbol.asyncIterator, o = Symbol.iterator); e--;) { if (t && null != (n = r[t])) return n.call(r); if (o && null != (n = r[o])) return new AsyncFromSyncIterator(n.call(r)); t = "@@asyncIterator", o = "@@iterator"; } throw new TypeError("Object is not async iterable"); } function AsyncFromSyncIterator(r) { function AsyncFromSyncIteratorContinuation(r) { if (Object(r) !== r) return Promise.reject(new TypeError(r + " is not an object.")); var n = r.done; return Promise.resolve(r.value).then(function (r) { return { value: r, done: n }; }); } return AsyncFromSyncIterator = function AsyncFromSyncIterator(r) { this.s = r, this.n = r.next; }, AsyncFromSyncIterator.prototype = { s: null, n: null, next: function next() { return AsyncFromSyncIteratorContinuation(this.n.apply(this.s, arguments)); }, "return": function _return(r) { var n = this.s["return"]; return void 0 === n ? Promise.resolve({ value: r, done: !0 }) : AsyncFromSyncIteratorContinuation(n.apply(this.s, arguments)); }, "throw": function _throw(r) { var n = this.s["return"]; return void 0 === n ? Promise.reject(r) : AsyncFromSyncIteratorContinuation(n.apply(this.s, arguments)); } }, new AsyncFromSyncIterator(r); } var DatabaseServiceProvider = exports["default"] = /*#__PURE__*/function () { /** * @param {AWS} _AWS_ * @param {object} credentials */ function DatabaseServiceProvider(_AWS_, credentials) { _classCallCheck(this, DatabaseServiceProvider); // TARGET this.TARGET = { AWS: _AWS_, ItemService: new _item["default"](_AWS_), TableService: new _table["default"](_AWS_) }; // SOURCE this.SOURCE = {}; this.SOURCE.AWS = new _aws.AWS(); this.SOURCE.AWS.initialize(credentials); this.SOURCE.ItemService = new _item["default"](this.SOURCE.AWS); this.SOURCE.TableService = new _table["default"](this.SOURCE.AWS); } /** * @param {object} credentials */ return _createClass(DatabaseServiceProvider, [{ key: "update", value: (function () { var _update = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee(credentials) { var _yield$import, AWS; return _regeneratorRuntime().wrap(function _callee$(_context) { while (1) switch (_context.prev = _context.next) { case 0: _context.next = 2; return Promise.resolve().then(function () { return _interopRequireWildcard(require("../config/aws")); }); case 2: _yield$import = _context.sent; AWS = _yield$import["default"]; AWS.initialize(credentials); case 5: case "end": return _context.stop(); } }, _callee); })); function update(_x) { return _update.apply(this, arguments); } return update; }()) }, { key: "reset", value: function () { var _reset = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee2() { var _yield$import2, AWS; return _regeneratorRuntime().wrap(function _callee2$(_context2) { while (1) switch (_context2.prev = _context2.next) { case 0: _context2.next = 2; return Promise.resolve().then(function () { return _interopRequireWildcard(require("../config/aws")); }); case 2: _yield$import2 = _context2.sent; AWS = _yield$import2["default"]; AWS.initialize(); case 5: case "end": return _context2.stop(); } }, _callee2); })); function reset() { return _reset.apply(this, arguments); } return reset; }() /** * @returns {Promise<Array<string>>} */ }, { key: "all", value: (function () { var _all = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee3() { var tables; return _regeneratorRuntime().wrap(function _callee3$(_context3) { while (1) switch (_context3.prev = _context3.next) { case 0: _context3.next = 2; return this.SOURCE.TableService.all(); case 2: tables = _context3.sent; return _context3.abrupt("return", tables); case 4: case "end": return _context3.stop(); } }, _callee3, this); })); function all() { return _all.apply(this, arguments); } return all; }() /** * @param {Array<{source: string, target: string}>} tables * @param {string} uid * @param {EventEmitter} eventEmitter * * @returns {Array<string>} */ ) }, { key: "restore", value: function restore() { var _this = this; var tables = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : []; var uid = arguments.length > 1 ? arguments[1] : undefined; var eventEmitter = arguments.length > 2 ? arguments[2] : undefined; var abortController = arguments.length > 3 ? arguments[3] : undefined; var queue = _toConsumableArray(tables); var jobs = queue.splice(0, _dynamodb.POOL_SIZE); var counter = jobs.length; eventEmitter.on(_event.EVENTS.END, function () { if (queue.length) { var job = queue.shift(); run(job); return; } if (--counter === 0) { eventEmitter.emit(_event.EVENTS.CLOSE, uid); } }); var run = /*#__PURE__*/function () { var _ref2 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee4(_ref) { var source, target, tableName, _restore, _iteratorAbruptCompletion, _didIteratorError, _iteratorError, _iterator, _step, _step$value, ItemCount, TotalItemCount; return _regeneratorRuntime().wrap(function _callee4$(_context4) { while (1) switch (_context4.prev = _context4.next) { case 0: source = _ref.source, target = _ref.target; tableName = source; _context4.prev = 2; eventEmitter.emit(_event.EVENTS.BEGIN, uid, { tableName: tableName }); _restore = _table["default"].restore(source, target, _this, abortController); _iteratorAbruptCompletion = false; _didIteratorError = false; _context4.prev = 7; _iterator = _asyncIterator(_restore); case 9: _context4.next = 11; return _iterator.next(); case 11: if (!(_iteratorAbruptCompletion = !(_step = _context4.sent).done)) { _context4.next = 17; break; } _step$value = _slicedToArray(_step.value, 2), ItemCount = _step$value[0], TotalItemCount = _step$value[1]; eventEmitter.emit(_event.EVENTS.PROGRESS, uid, { tableName: tableName, data: [ItemCount, TotalItemCount] }); case 14: _iteratorAbruptCompletion = false; _context4.next = 9; break; case 17: _context4.next = 23; break; case 19: _context4.prev = 19; _context4.t0 = _context4["catch"](7); _didIteratorError = true; _iteratorError = _context4.t0; case 23: _context4.prev = 23; _context4.prev = 24; if (!(_iteratorAbruptCompletion && _iterator["return"] != null)) { _context4.next = 28; break; } _context4.next = 28; return _iterator["return"](); case 28: _context4.prev = 28; if (!_didIteratorError) { _context4.next = 31; break; } throw _iteratorError; case 31: return _context4.finish(28); case 32: return _context4.finish(23); case 33: eventEmitter.emit(_event.EVENTS.SUCCESS, uid, { tableName: tableName }); _context4.next = 40; break; case 36: _context4.prev = 36; _context4.t1 = _context4["catch"](2); eventEmitter.emit(_event.EVENTS.FAILURE, uid, { tableName: tableName, error: _context4.t1 }); console.error(_context4.t1); case 40: _context4.prev = 40; eventEmitter.emit(_event.EVENTS.END, uid, { tableName: tableName }); return _context4.finish(40); case 43: case "end": return _context4.stop(); } }, _callee4, null, [[2, 36, 40, 43], [7, 19, 23, 33], [24,, 28, 32]]); })); return function run(_x2) { return _ref2.apply(this, arguments); }; }(); jobs.map(run); return tables; } /* * @param {string} sourceTableName * @param {string} targetTableName * @return {Promise<Array<number>>} */ }, { key: "compare", value: (function () { var _compare = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee5(sourceTableName, targetTableName) { var _yield$Promise$all, _yield$Promise$all2, TargetTable, SourceTable, ItemCount, TotalItemCount; return _regeneratorRuntime().wrap(function _callee5$(_context5) { while (1) switch (_context5.prev = _context5.next) { case 0: _context5.next = 2; return Promise.all([this.TARGET.TableService.describe(targetTableName), this.SOURCE.TableService.describe(sourceTableName)]); case 2: _yield$Promise$all = _context5.sent; _yield$Promise$all2 = _slicedToArray(_yield$Promise$all, 2); TargetTable = _yield$Promise$all2[0].Table; SourceTable = _yield$Promise$all2[1].Table; ItemCount = TargetTable.ItemCount; TotalItemCount = SourceTable.ItemCount; return _context5.abrupt("return", [ItemCount, TotalItemCount]); case 9: case "end": return _context5.stop(); } }, _callee5, this); })); function compare(_x3, _x4) { return _compare.apply(this, arguments); } return compare; }()) }]); }();