UNPKG

dynamodb-dashboard

Version:
295 lines (291 loc) 18.7 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 _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) && (u = i[(c = i[4]) ? 5 : (c = 3, 3)], i[4] = 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) { function o(r, n) { _regeneratorDefine2(e, r, function (e) { return this._invoke(r, n, e); }); } r ? i ? i(e, r, { value: n, enumerable: !t, configurable: !t, writable: !t }) : e[r] = n : (o("next", 0), o("throw", 1), o("return", 2)); }, _regeneratorDefine2(e, r, n, t); } 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 _t3 in e) "default" !== _t3 && {}.hasOwnProperty.call(e, _t3) && ((i = (o = Object.defineProperty) && Object.getOwnPropertyDescriptor(e, _t3)) && (i.get || i.set) ? o(f, _t3, i) : f[_t3] = e[_t3]); 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__*/_regenerator().m(function _callee(credentials) { var _yield$import, AWS; return _regenerator().w(function (_context) { while (1) switch (_context.n) { case 0: _context.n = 1; return Promise.resolve().then(function () { return _interopRequireWildcard(require("../config/aws")); }); case 1: _yield$import = _context.v; AWS = _yield$import["default"]; AWS.initialize(credentials); case 2: return _context.a(2); } }, _callee); })); function update(_x) { return _update.apply(this, arguments); } return update; }()) }, { key: "reset", value: function () { var _reset = _asyncToGenerator(/*#__PURE__*/_regenerator().m(function _callee2() { var _yield$import2, AWS; return _regenerator().w(function (_context2) { while (1) switch (_context2.n) { case 0: _context2.n = 1; return Promise.resolve().then(function () { return _interopRequireWildcard(require("../config/aws")); }); case 1: _yield$import2 = _context2.v; AWS = _yield$import2["default"]; AWS.initialize(); case 2: return _context2.a(2); } }, _callee2); })); function reset() { return _reset.apply(this, arguments); } return reset; }() /** * @returns {Promise<Array<string>>} */ }, { key: "all", value: (function () { var _all = _asyncToGenerator(/*#__PURE__*/_regenerator().m(function _callee3() { var tables; return _regenerator().w(function (_context3) { while (1) switch (_context3.n) { case 0: _context3.n = 1; return this.SOURCE.TableService.all(); case 1: tables = _context3.v; return _context3.a(2, tables); } }, _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__*/_regenerator().m(function _callee4(_ref) { var source, target, tableName, _restore, _iteratorAbruptCompletion, _didIteratorError, _iteratorError, _iterator, _step, _step$value, ItemCount, TotalItemCount, _t, _t2; return _regenerator().w(function (_context4) { while (1) switch (_context4.p = _context4.n) { case 0: source = _ref.source, target = _ref.target; tableName = source; _context4.p = 1; eventEmitter.emit(_event.EVENTS.BEGIN, uid, { tableName: tableName }); _restore = _table["default"].restore(source, target, _this, abortController); _iteratorAbruptCompletion = false; _didIteratorError = false; _context4.p = 2; _iterator = _asyncIterator(_restore); case 3: _context4.n = 4; return _iterator.next(); case 4: if (!(_iteratorAbruptCompletion = !(_step = _context4.v).done)) { _context4.n = 6; 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 5: _iteratorAbruptCompletion = false; _context4.n = 3; break; case 6: _context4.n = 8; break; case 7: _context4.p = 7; _t = _context4.v; _didIteratorError = true; _iteratorError = _t; case 8: _context4.p = 8; _context4.p = 9; if (!(_iteratorAbruptCompletion && _iterator["return"] != null)) { _context4.n = 10; break; } _context4.n = 10; return _iterator["return"](); case 10: _context4.p = 10; if (!_didIteratorError) { _context4.n = 11; break; } throw _iteratorError; case 11: return _context4.f(10); case 12: return _context4.f(8); case 13: eventEmitter.emit(_event.EVENTS.SUCCESS, uid, { tableName: tableName }); _context4.n = 15; break; case 14: _context4.p = 14; _t2 = _context4.v; eventEmitter.emit(_event.EVENTS.FAILURE, uid, { tableName: tableName, error: { message: _t2.message } }); console.error(_t2); case 15: _context4.p = 15; eventEmitter.emit(_event.EVENTS.END, uid, { tableName: tableName }); return _context4.f(15); case 16: return _context4.a(2); } }, _callee4, null, [[9,, 10, 12], [2, 7, 8, 13], [1, 14, 15, 16]]); })); 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__*/_regenerator().m(function _callee5(sourceTableName, targetTableName) { var _yield$Promise$all, _yield$Promise$all2, TargetTable, SourceTable, ItemCount, TotalItemCount; return _regenerator().w(function (_context5) { while (1) switch (_context5.n) { case 0: _context5.n = 1; return Promise.all([this.TARGET.TableService.describe(targetTableName), this.SOURCE.TableService.describe(sourceTableName)]); case 1: _yield$Promise$all = _context5.v; _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.a(2, [ItemCount, TotalItemCount]); } }, _callee5, this); })); function compare(_x3, _x4) { return _compare.apply(this, arguments); } return compare; }()) }]); }();