@trap_stevo/star-vault
Version:
Deterministic data engine that eliminates query-time joins and enables normalized data execution. Architect secure, scalable, real-time systems with integrated sharding, encryption, and event-driven data flows. Manage hierarchical structures, execute adva
710 lines (709 loc) • 41.1 kB
JavaScript
"use strict";
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 _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 _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 _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 ownKeys(e, r) { var t = Object.keys(e); if (Object.getOwnPropertySymbols) { var o = Object.getOwnPropertySymbols(e); r && (o = o.filter(function (r) { return Object.getOwnPropertyDescriptor(e, r).enumerable; })), t.push.apply(t, o); } return t; }
function _objectSpread(e) { for (var r = 1; r < arguments.length; r++) { var t = null != arguments[r] ? arguments[r] : {}; r % 2 ? ownKeys(Object(t), !0).forEach(function (r) { _defineProperty(e, r, t[r]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(e, Object.getOwnPropertyDescriptors(t)) : ownKeys(Object(t)).forEach(function (r) { Object.defineProperty(e, r, Object.getOwnPropertyDescriptor(t, r)); }); } return e; }
function _defineProperty(e, r, t) { return (r = _toPropertyKey(r)) in e ? Object.defineProperty(e, r, { value: t, enumerable: !0, configurable: !0, writable: !0 }) : e[r] = t, e; }
function _regeneratorRuntime() { "use strict"; /*! regenerator-runtime -- Copyright (c) 2014-present, Facebook, Inc. -- license (MIT): https://github.com/facebook/regenerator/blob/main/LICENSE */ _regeneratorRuntime = function _regeneratorRuntime() { return e; }; var t, e = {}, r = Object.prototype, n = r.hasOwnProperty, o = Object.defineProperty || function (t, e, r) { t[e] = r.value; }, i = "function" == typeof Symbol ? Symbol : {}, a = i.iterator || "@@iterator", c = i.asyncIterator || "@@asyncIterator", u = i.toStringTag || "@@toStringTag"; function define(t, e, r) { return Object.defineProperty(t, e, { value: r, enumerable: !0, configurable: !0, writable: !0 }), t[e]; } try { define({}, ""); } catch (t) { define = function define(t, e, r) { return t[e] = r; }; } function wrap(t, e, r, n) { var i = e && e.prototype instanceof Generator ? e : Generator, a = Object.create(i.prototype), c = new Context(n || []); return o(a, "_invoke", { value: makeInvokeMethod(t, r, c) }), a; } function tryCatch(t, e, r) { try { return { type: "normal", arg: t.call(e, r) }; } catch (t) { return { type: "throw", arg: t }; } } e.wrap = wrap; var h = "suspendedStart", l = "suspendedYield", f = "executing", s = "completed", y = {}; function Generator() {} function GeneratorFunction() {} function GeneratorFunctionPrototype() {} var p = {}; define(p, a, function () { return this; }); var d = Object.getPrototypeOf, v = d && d(d(values([]))); v && v !== r && n.call(v, a) && (p = v); var g = GeneratorFunctionPrototype.prototype = Generator.prototype = Object.create(p); function defineIteratorMethods(t) { ["next", "throw", "return"].forEach(function (e) { define(t, e, function (t) { return this._invoke(e, t); }); }); } function AsyncIterator(t, e) { function invoke(r, o, i, a) { var c = tryCatch(t[r], t, o); if ("throw" !== c.type) { var u = c.arg, h = u.value; return h && "object" == _typeof(h) && n.call(h, "__await") ? e.resolve(h.__await).then(function (t) { invoke("next", t, i, a); }, function (t) { invoke("throw", t, i, a); }) : e.resolve(h).then(function (t) { u.value = t, i(u); }, function (t) { return invoke("throw", t, i, a); }); } a(c.arg); } var r; o(this, "_invoke", { value: function value(t, n) { function callInvokeWithMethodAndArg() { return new e(function (e, r) { invoke(t, n, e, r); }); } return r = r ? r.then(callInvokeWithMethodAndArg, callInvokeWithMethodAndArg) : callInvokeWithMethodAndArg(); } }); } function makeInvokeMethod(e, r, n) { var o = h; return function (i, a) { if (o === f) throw Error("Generator is already running"); if (o === s) { if ("throw" === i) throw a; return { value: t, done: !0 }; } for (n.method = i, n.arg = a;;) { var c = n.delegate; if (c) { var u = maybeInvokeDelegate(c, n); if (u) { if (u === y) continue; return u; } } if ("next" === n.method) n.sent = n._sent = n.arg;else if ("throw" === n.method) { if (o === h) throw o = s, n.arg; n.dispatchException(n.arg); } else "return" === n.method && n.abrupt("return", n.arg); o = f; var p = tryCatch(e, r, n); if ("normal" === p.type) { if (o = n.done ? s : l, p.arg === y) continue; return { value: p.arg, done: n.done }; } "throw" === p.type && (o = s, n.method = "throw", n.arg = p.arg); } }; } function maybeInvokeDelegate(e, r) { var n = r.method, o = e.iterator[n]; if (o === t) return r.delegate = null, "throw" === n && e.iterator["return"] && (r.method = "return", r.arg = t, maybeInvokeDelegate(e, r), "throw" === r.method) || "return" !== n && (r.method = "throw", r.arg = new TypeError("The iterator does not provide a '" + n + "' method")), y; var i = tryCatch(o, e.iterator, r.arg); if ("throw" === i.type) return r.method = "throw", r.arg = i.arg, r.delegate = null, y; var a = i.arg; return a ? a.done ? (r[e.resultName] = a.value, r.next = e.nextLoc, "return" !== r.method && (r.method = "next", r.arg = t), r.delegate = null, y) : a : (r.method = "throw", r.arg = new TypeError("iterator result is not an object"), r.delegate = null, y); } function pushTryEntry(t) { var e = { tryLoc: t[0] }; 1 in t && (e.catchLoc = t[1]), 2 in t && (e.finallyLoc = t[2], e.afterLoc = t[3]), this.tryEntries.push(e); } function resetTryEntry(t) { var e = t.completion || {}; e.type = "normal", delete e.arg, t.completion = e; } function Context(t) { this.tryEntries = [{ tryLoc: "root" }], t.forEach(pushTryEntry, this), this.reset(!0); } function values(e) { if (e || "" === e) { var r = e[a]; if (r) return r.call(e); if ("function" == typeof e.next) return e; if (!isNaN(e.length)) { var o = -1, i = function next() { for (; ++o < e.length;) if (n.call(e, o)) return next.value = e[o], next.done = !1, next; return next.value = t, next.done = !0, next; }; return i.next = i; } } throw new TypeError(_typeof(e) + " is not iterable"); } return GeneratorFunction.prototype = GeneratorFunctionPrototype, o(g, "constructor", { value: GeneratorFunctionPrototype, configurable: !0 }), o(GeneratorFunctionPrototype, "constructor", { value: GeneratorFunction, configurable: !0 }), GeneratorFunction.displayName = define(GeneratorFunctionPrototype, u, "GeneratorFunction"), e.isGeneratorFunction = function (t) { var e = "function" == typeof t && t.constructor; return !!e && (e === GeneratorFunction || "GeneratorFunction" === (e.displayName || e.name)); }, e.mark = function (t) { return Object.setPrototypeOf ? Object.setPrototypeOf(t, GeneratorFunctionPrototype) : (t.__proto__ = GeneratorFunctionPrototype, define(t, u, "GeneratorFunction")), t.prototype = Object.create(g), t; }, e.awrap = function (t) { return { __await: t }; }, defineIteratorMethods(AsyncIterator.prototype), define(AsyncIterator.prototype, c, function () { return this; }), e.AsyncIterator = AsyncIterator, e.async = function (t, r, n, o, i) { void 0 === i && (i = Promise); var a = new AsyncIterator(wrap(t, r, n, o), i); return e.isGeneratorFunction(r) ? a : a.next().then(function (t) { return t.done ? t.value : a.next(); }); }, defineIteratorMethods(g), define(g, u, "Generator"), define(g, a, function () { return this; }), define(g, "toString", function () { return "[object Generator]"; }), e.keys = function (t) { var e = Object(t), r = []; for (var n in e) r.push(n); return r.reverse(), function next() { for (; r.length;) { var t = r.pop(); if (t in e) return next.value = t, next.done = !1, next; } return next.done = !0, next; }; }, e.values = values, Context.prototype = { constructor: Context, reset: function reset(e) { if (this.prev = 0, this.next = 0, this.sent = this._sent = t, this.done = !1, this.delegate = null, this.method = "next", this.arg = t, this.tryEntries.forEach(resetTryEntry), !e) for (var r in this) "t" === r.charAt(0) && n.call(this, r) && !isNaN(+r.slice(1)) && (this[r] = t); }, stop: function stop() { this.done = !0; var t = this.tryEntries[0].completion; if ("throw" === t.type) throw t.arg; return this.rval; }, dispatchException: function dispatchException(e) { if (this.done) throw e; var r = this; function handle(n, o) { return a.type = "throw", a.arg = e, r.next = n, o && (r.method = "next", r.arg = t), !!o; } for (var o = this.tryEntries.length - 1; o >= 0; --o) { var i = this.tryEntries[o], a = i.completion; if ("root" === i.tryLoc) return handle("end"); if (i.tryLoc <= this.prev) { var c = n.call(i, "catchLoc"), u = n.call(i, "finallyLoc"); if (c && u) { if (this.prev < i.catchLoc) return handle(i.catchLoc, !0); if (this.prev < i.finallyLoc) return handle(i.finallyLoc); } else if (c) { if (this.prev < i.catchLoc) return handle(i.catchLoc, !0); } else { if (!u) throw Error("try statement without catch or finally"); if (this.prev < i.finallyLoc) return handle(i.finallyLoc); } } } }, abrupt: function abrupt(t, e) { for (var r = this.tryEntries.length - 1; r >= 0; --r) { var o = this.tryEntries[r]; if (o.tryLoc <= this.prev && n.call(o, "finallyLoc") && this.prev < o.finallyLoc) { var i = o; break; } } i && ("break" === t || "continue" === t) && i.tryLoc <= e && e <= i.finallyLoc && (i = null); var a = i ? i.completion : {}; return a.type = t, a.arg = e, i ? (this.method = "next", this.next = i.finallyLoc, y) : this.complete(a); }, complete: function complete(t, e) { if ("throw" === t.type) throw t.arg; return "break" === t.type || "continue" === t.type ? this.next = t.arg : "return" === t.type ? (this.rval = this.arg = t.arg, this.method = "return", this.next = "end") : "normal" === t.type && e && (this.next = e), y; }, finish: function finish(t) { for (var e = this.tryEntries.length - 1; e >= 0; --e) { var r = this.tryEntries[e]; if (r.finallyLoc === t) return this.complete(r.completion, r.afterLoc), resetTryEntry(r), y; } }, "catch": function _catch(t) { for (var e = this.tryEntries.length - 1; e >= 0; --e) { var r = this.tryEntries[e]; if (r.tryLoc === t) { var n = r.completion; if ("throw" === n.type) { var o = n.arg; resetTryEntry(r); } return o; } } throw Error("illegal catch attempt"); }, delegateYield: function delegateYield(e, r, n) { return this.delegate = { iterator: values(e), resultName: r, nextLoc: n }, "next" === this.method && (this.arg = t), y; } }, e; }
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 _classPrivateMethodInitSpec(e, a) { _checkPrivateRedeclaration(e, a), a.add(e); }
function _classPrivateFieldInitSpec(e, t, a) { _checkPrivateRedeclaration(e, t), t.set(e, a); }
function _checkPrivateRedeclaration(e, t) { if (t.has(e)) throw new TypeError("Cannot initialize the same private elements twice on an object"); }
function _classPrivateFieldGet(s, a) { return s.get(_assertClassBrand(s, a)); }
function _classPrivateFieldSet(s, a, r) { return s.set(_assertClassBrand(s, a), r), r; }
function _assertClassBrand(e, t, n) { if ("function" == typeof e ? e === t : e.has(t)) return arguments.length < 3 ? t : n; throw new TypeError("Private element is not present on this object"); }
var fs = require("fs");
var path = require("path");
var crypto = require("crypto");
var _locksDir = /*#__PURE__*/new WeakMap();
var _instanceId = /*#__PURE__*/new WeakMap();
var _leaseTTLms = /*#__PURE__*/new WeakMap();
var _retryDelayMs = /*#__PURE__*/new WeakMap();
var _maxRetryDelayMs = /*#__PURE__*/new WeakMap();
var _jitterMs = /*#__PURE__*/new WeakMap();
var _staleGraceMs = /*#__PURE__*/new WeakMap();
var _maxStealAttempts = /*#__PURE__*/new WeakMap();
var _renewEveryMs = /*#__PURE__*/new WeakMap();
var _renewSkewMs = /*#__PURE__*/new WeakMap();
var _held = /*#__PURE__*/new WeakMap();
var _renewTimer = /*#__PURE__*/new WeakMap();
var _closing = /*#__PURE__*/new WeakMap();
var _ShardLockManager_brand = /*#__PURE__*/new WeakSet();
var ShardLockManager = /*#__PURE__*/function () {
function ShardLockManager(locksDir) {
var _ref = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : {},
_ref$instanceId = _ref.instanceId,
instanceId = _ref$instanceId === void 0 ? null : _ref$instanceId,
_ref$leaseTTLms = _ref.leaseTTLms,
leaseTTLms = _ref$leaseTTLms === void 0 ? 250 : _ref$leaseTTLms,
_ref$retryDelayMs = _ref.retryDelayMs,
retryDelayMs = _ref$retryDelayMs === void 0 ? 3 : _ref$retryDelayMs,
_ref$maxRetryDelayMs = _ref.maxRetryDelayMs,
maxRetryDelayMs = _ref$maxRetryDelayMs === void 0 ? 25 : _ref$maxRetryDelayMs,
_ref$jitterMs = _ref.jitterMs,
jitterMs = _ref$jitterMs === void 0 ? 2 : _ref$jitterMs,
_ref$staleGraceMs = _ref.staleGraceMs,
staleGraceMs = _ref$staleGraceMs === void 0 ? 10 : _ref$staleGraceMs,
_ref$maxStealAttempts = _ref.maxStealAttempts,
maxStealAttempts = _ref$maxStealAttempts === void 0 ? 2 : _ref$maxStealAttempts,
_ref$renewEveryMs = _ref.renewEveryMs,
renewEveryMs = _ref$renewEveryMs === void 0 ? null : _ref$renewEveryMs,
_ref$renewSkewMs = _ref.renewSkewMs,
renewSkewMs = _ref$renewSkewMs === void 0 ? 10 : _ref$renewSkewMs;
_classCallCheck(this, ShardLockManager);
_classPrivateMethodInitSpec(this, _ShardLockManager_brand);
_classPrivateFieldInitSpec(this, _locksDir, void 0);
_classPrivateFieldInitSpec(this, _instanceId, void 0);
_classPrivateFieldInitSpec(this, _leaseTTLms, void 0);
_classPrivateFieldInitSpec(this, _retryDelayMs, void 0);
_classPrivateFieldInitSpec(this, _maxRetryDelayMs, void 0);
_classPrivateFieldInitSpec(this, _jitterMs, void 0);
_classPrivateFieldInitSpec(this, _staleGraceMs, void 0);
_classPrivateFieldInitSpec(this, _maxStealAttempts, void 0);
_classPrivateFieldInitSpec(this, _renewEveryMs, void 0);
_classPrivateFieldInitSpec(this, _renewSkewMs, void 0);
_classPrivateFieldInitSpec(this, _held, void 0);
_classPrivateFieldInitSpec(this, _renewTimer, void 0);
_classPrivateFieldInitSpec(this, _closing, void 0);
if (typeof locksDir !== "string" || !locksDir) {
throw new Error("[Shard Locker] ~ requires a valid locksDir.");
}
_classPrivateFieldSet(_locksDir, this, locksDir);
_classPrivateFieldSet(_instanceId, this, typeof instanceId === "string" && instanceId ? instanceId : ShardLockManager.generateInstanceId());
_classPrivateFieldSet(_leaseTTLms, this, typeof leaseTTLms === "number" && Number.isFinite(leaseTTLms) ? Math.max(10, Math.floor(leaseTTLms)) : 250);
_classPrivateFieldSet(_retryDelayMs, this, typeof retryDelayMs === "number" && Number.isFinite(retryDelayMs) ? Math.max(0, Math.floor(retryDelayMs)) : 3);
_classPrivateFieldSet(_maxRetryDelayMs, this, typeof maxRetryDelayMs === "number" && Number.isFinite(maxRetryDelayMs) ? Math.max(_classPrivateFieldGet(_retryDelayMs, this), Math.floor(maxRetryDelayMs)) : 25);
_classPrivateFieldSet(_jitterMs, this, typeof jitterMs === "number" && Number.isFinite(jitterMs) ? Math.max(0, Math.floor(jitterMs)) : 2);
_classPrivateFieldSet(_staleGraceMs, this, typeof staleGraceMs === "number" && Number.isFinite(staleGraceMs) ? Math.max(0, Math.floor(staleGraceMs)) : 10);
_classPrivateFieldSet(_maxStealAttempts, this, typeof maxStealAttempts === "number" && Number.isFinite(maxStealAttempts) ? Math.max(0, Math.floor(maxStealAttempts)) : 2);
_classPrivateFieldSet(_renewSkewMs, this, typeof renewSkewMs === "number" && Number.isFinite(renewSkewMs) ? Math.max(0, Math.floor(renewSkewMs)) : 10);
_classPrivateFieldSet(_renewEveryMs, this, typeof renewEveryMs === "number" && Number.isFinite(renewEveryMs) ? Math.max(5, Math.floor(renewEveryMs)) : null);
_classPrivateFieldSet(_held, this, new Map());
_classPrivateFieldSet(_renewTimer, this, null);
_classPrivateFieldSet(_closing, this, false);
fs.mkdirSync(_classPrivateFieldGet(_locksDir, this), {
recursive: true
});
if (_classPrivateFieldGet(_renewEveryMs, this) === null) {
_classPrivateFieldSet(_renewEveryMs, this, Math.max(5, Math.floor(_classPrivateFieldGet(_leaseTTLms, this) / 2)));
}
_assertClassBrand(_ShardLockManager_brand, this, _startRenewLoop).call(this);
}
return _createClass(ShardLockManager, [{
key: "getInstanceId",
value: function getInstanceId() {
return _classPrivateFieldGet(_instanceId, this);
}
}, {
key: "getLocksDir",
value: function getLocksDir() {
return _classPrivateFieldGet(_locksDir, this);
}
}, {
key: "getLeaseTTLms",
value: function getLeaseTTLms() {
return _classPrivateFieldGet(_leaseTTLms, this);
}
}, {
key: "getRetryDelayMs",
value: function getRetryDelayMs() {
return _classPrivateFieldGet(_retryDelayMs, this);
}
}, {
key: "getMaxRetryDelayMs",
value: function getMaxRetryDelayMs() {
return _classPrivateFieldGet(_maxRetryDelayMs, this);
}
}, {
key: "getRenewEveryMs",
value: function getRenewEveryMs() {
return _classPrivateFieldGet(_renewEveryMs, this);
}
}, {
key: "getLockPath",
value: function getLockPath(collection, shard) {
var safe = _assertClassBrand(_ShardLockManager_brand, this, _safeLockName).call(this, collection);
var s = Number(shard);
return path.join(_classPrivateFieldGet(_locksDir, this), "".concat(safe, "__shard_").concat(s, ".lock"));
}
}, {
key: "acquire",
value: function () {
var _acquire = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee(collection, shard) {
var options,
ttlMs,
maxWaitMs,
deadline,
delay,
ok,
jitter,
waitMs,
_args = arguments;
return _regeneratorRuntime().wrap(function _callee$(_context) {
while (1) switch (_context.prev = _context.next) {
case 0:
options = _args.length > 2 && _args[2] !== undefined ? _args[2] : {};
if (!(_classPrivateFieldGet(_closing, this) === true)) {
_context.next = 3;
break;
}
throw new Error("[Shard Locker] ~ closing; cannot acquire new leases.");
case 3:
ttlMs = typeof options.leaseTTLms === "number" && Number.isFinite(options.leaseTTLms) ? Math.max(10, Math.floor(options.leaseTTLms)) : _classPrivateFieldGet(_leaseTTLms, this);
maxWaitMs = typeof options.maxWaitMs === "number" && Number.isFinite(options.maxWaitMs) ? Math.max(0, Math.floor(options.maxWaitMs)) : Infinity;
deadline = Number.isFinite(maxWaitMs) ? Date.now() + maxWaitMs : Infinity;
delay = _classPrivateFieldGet(_retryDelayMs, this);
case 7:
if (!(Date.now() < deadline)) {
_context.next = 22;
break;
}
if (!(_classPrivateFieldGet(_closing, this) === true)) {
_context.next = 10;
break;
}
throw new Error("[Shard Locker] ~ closing; cannot acquire new leases.");
case 10:
_context.next = 12;
return this.tryAcquire(collection, shard, {
leaseTTLms: ttlMs
});
case 12:
ok = _context.sent;
if (!(ok === true)) {
_context.next = 15;
break;
}
return _context.abrupt("return", true);
case 15:
jitter = _classPrivateFieldGet(_jitterMs, this) > 0 ? Math.floor(Math.random() * (_classPrivateFieldGet(_jitterMs, this) + 1)) : 0;
waitMs = Math.min(_classPrivateFieldGet(_maxRetryDelayMs, this), Math.max(0, delay + jitter));
_context.next = 19;
return _assertClassBrand(_ShardLockManager_brand, this, _sleep).call(this, waitMs);
case 19:
delay = Math.min(_classPrivateFieldGet(_maxRetryDelayMs, this), Math.max(_classPrivateFieldGet(_retryDelayMs, this), delay + 1));
_context.next = 7;
break;
case 22:
return _context.abrupt("return", false);
case 23:
case "end":
return _context.stop();
}
}, _callee, this);
}));
function acquire(_x, _x2) {
return _acquire.apply(this, arguments);
}
return acquire;
}()
}, {
key: "tryAcquire",
value: function () {
var _tryAcquire = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee2(collection, shard) {
var options,
lockPath,
ttlMs,
now,
payloadObj,
payload,
stealAttempts,
st,
staleAt,
_args2 = arguments;
return _regeneratorRuntime().wrap(function _callee2$(_context2) {
while (1) switch (_context2.prev = _context2.next) {
case 0:
options = _args2.length > 2 && _args2[2] !== undefined ? _args2[2] : {};
if (!(_classPrivateFieldGet(_closing, this) === true)) {
_context2.next = 3;
break;
}
return _context2.abrupt("return", false);
case 3:
lockPath = this.getLockPath(collection, shard);
ttlMs = typeof options.leaseTTLms === "number" && Number.isFinite(options.leaseTTLms) ? Math.max(10, Math.floor(options.leaseTTLms)) : _classPrivateFieldGet(_leaseTTLms, this);
now = Date.now();
payloadObj = {
owner: _classPrivateFieldGet(_instanceId, this),
pid: process.pid,
collection: String(collection),
shard: Number(shard),
issuedAt: now,
expiresAt: now + ttlMs
};
payload = JSON.stringify(payloadObj);
_context2.prev = 8;
_context2.next = 11;
return fs.promises.writeFile(lockPath, payload, {
flag: "wx"
});
case 11:
_classPrivateFieldGet(_held, this).set(lockPath, payloadObj);
return _context2.abrupt("return", true);
case 15:
_context2.prev = 15;
_context2.t0 = _context2["catch"](8);
if (!(_context2.t0 && _context2.t0.code === "EPERM")) {
_context2.next = 21;
break;
}
_context2.next = 20;
return _assertClassBrand(_ShardLockManager_brand, this, _sleep).call(this, 2);
case 20:
return _context2.abrupt("return", false);
case 21:
if (!(!_context2.t0 || _context2.t0.code !== "EEXIST")) {
_context2.next = 23;
break;
}
throw _context2.t0;
case 23:
stealAttempts = 0;
case 24:
if (!(stealAttempts <= _classPrivateFieldGet(_maxStealAttempts, this))) {
_context2.next = 46;
break;
}
_context2.next = 27;
return _assertClassBrand(_ShardLockManager_brand, this, _readLease).call(this, lockPath);
case 27:
st = _context2.sent;
if (!(!st || !Number.isFinite(st.expiresAt))) {
_context2.next = 35;
break;
}
_context2.next = 31;
return _assertClassBrand(_ShardLockManager_brand, this, _trySteal).call(this, lockPath, payload, payloadObj);
case 31:
if (!_context2.sent) {
_context2.next = 33;
break;
}
return _context2.abrupt("return", true);
case 33:
stealAttempts += 1;
return _context2.abrupt("break", 46);
case 35:
staleAt = Number(st.expiresAt) + _classPrivateFieldGet(_staleGraceMs, this);
if (!(staleAt <= now)) {
_context2.next = 43;
break;
}
_context2.next = 39;
return _assertClassBrand(_ShardLockManager_brand, this, _trySteal).call(this, lockPath, payload, payloadObj);
case 39:
if (!_context2.sent) {
_context2.next = 41;
break;
}
return _context2.abrupt("return", true);
case 41:
stealAttempts += 1;
return _context2.abrupt("continue", 24);
case 43:
return _context2.abrupt("return", false);
case 46:
return _context2.abrupt("return", false);
case 47:
case "end":
return _context2.stop();
}
}, _callee2, this, [[8, 15]]);
}));
function tryAcquire(_x3, _x4) {
return _tryAcquire.apply(this, arguments);
}
return tryAcquire;
}()
}, {
key: "release",
value: function () {
var _release = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee3(collection, shard) {
var lockPath;
return _regeneratorRuntime().wrap(function _callee3$(_context3) {
while (1) switch (_context3.prev = _context3.next) {
case 0:
lockPath = this.getLockPath(collection, shard);
_context3.next = 3;
return this.releaseByPath(lockPath);
case 3:
case "end":
return _context3.stop();
}
}, _callee3, this);
}));
function release(_x5, _x6) {
return _release.apply(this, arguments);
}
return release;
}()
}, {
key: "releaseByPath",
value: function () {
var _releaseByPath = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee4(lockPath) {
var p, st;
return _regeneratorRuntime().wrap(function _callee4$(_context4) {
while (1) switch (_context4.prev = _context4.next) {
case 0:
if (lockPath) {
_context4.next = 2;
break;
}
return _context4.abrupt("return");
case 2:
p = String(lockPath);
_context4.prev = 3;
_context4.next = 6;
return _assertClassBrand(_ShardLockManager_brand, this, _readLease).call(this, p);
case 6:
st = _context4.sent;
if (!(st && st.owner === _classPrivateFieldGet(_instanceId, this))) {
_context4.next = 15;
break;
}
_context4.prev = 8;
_context4.next = 11;
return fs.promises.unlink(p);
case 11:
_context4.next = 15;
break;
case 13:
_context4.prev = 13;
_context4.t0 = _context4["catch"](8);
case 15:
_context4.next = 19;
break;
case 17:
_context4.prev = 17;
_context4.t1 = _context4["catch"](3);
case 19:
_classPrivateFieldGet(_held, this)["delete"](p);
case 20:
case "end":
return _context4.stop();
}
}, _callee4, this, [[3, 17], [8, 13]]);
}));
function releaseByPath(_x7) {
return _releaseByPath.apply(this, arguments);
}
return releaseByPath;
}()
}, {
key: "renew",
value: function () {
var _renew = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee5(collection, shard) {
var options,
lockPath,
_args5 = arguments;
return _regeneratorRuntime().wrap(function _callee5$(_context5) {
while (1) switch (_context5.prev = _context5.next) {
case 0:
options = _args5.length > 2 && _args5[2] !== undefined ? _args5[2] : {};
lockPath = this.getLockPath(collection, shard);
return _context5.abrupt("return", this.renewByPath(lockPath, options));
case 3:
case "end":
return _context5.stop();
}
}, _callee5, this);
}));
function renew(_x8, _x9) {
return _renew.apply(this, arguments);
}
return renew;
}()
}, {
key: "renewByPath",
value: function () {
var _renewByPath = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee6(lockPath) {
var _ref2,
_ref2$leaseTTLms,
leaseTTLms,
p,
held,
ttlMs,
now,
st,
next,
data,
_args6 = arguments;
return _regeneratorRuntime().wrap(function _callee6$(_context6) {
while (1) switch (_context6.prev = _context6.next) {
case 0:
_ref2 = _args6.length > 1 && _args6[1] !== undefined ? _args6[1] : {}, _ref2$leaseTTLms = _ref2.leaseTTLms, leaseTTLms = _ref2$leaseTTLms === void 0 ? null : _ref2$leaseTTLms;
p = String(lockPath);
held = _classPrivateFieldGet(_held, this).get(p);
if (held) {
_context6.next = 5;
break;
}
return _context6.abrupt("return", false);
case 5:
ttlMs = typeof leaseTTLms === "number" && Number.isFinite(leaseTTLms) ? Math.max(10, Math.floor(leaseTTLms)) : _classPrivateFieldGet(_leaseTTLms, this);
now = Date.now();
_context6.next = 9;
return _assertClassBrand(_ShardLockManager_brand, this, _readLease).call(this, p);
case 9:
st = _context6.sent;
if (!(!st || st.owner !== _classPrivateFieldGet(_instanceId, this))) {
_context6.next = 13;
break;
}
_classPrivateFieldGet(_held, this)["delete"](p);
return _context6.abrupt("return", false);
case 13:
next = _objectSpread(_objectSpread({}, st), {}, {
owner: _classPrivateFieldGet(_instanceId, this),
pid: process.pid,
issuedAt: now,
expiresAt: now + ttlMs
});
data = JSON.stringify(next);
_context6.prev = 15;
_context6.next = 18;
return fs.promises.writeFile(p, data, "utf8");
case 18:
_context6.next = 23;
break;
case 20:
_context6.prev = 20;
_context6.t0 = _context6["catch"](15);
return _context6.abrupt("return", false);
case 23:
_classPrivateFieldGet(_held, this).set(p, next);
return _context6.abrupt("return", true);
case 25:
case "end":
return _context6.stop();
}
}, _callee6, this, [[15, 20]]);
}));
function renewByPath(_x10) {
return _renewByPath.apply(this, arguments);
}
return renewByPath;
}()
}, {
key: "close",
value: function () {
var _close = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee7() {
var paths, _i, _paths, p;
return _regeneratorRuntime().wrap(function _callee7$(_context7) {
while (1) switch (_context7.prev = _context7.next) {
case 0:
_classPrivateFieldSet(_closing, this, true);
if (_classPrivateFieldGet(_renewTimer, this) !== null) {
clearInterval(_classPrivateFieldGet(_renewTimer, this));
_classPrivateFieldSet(_renewTimer, this, null);
}
paths = Array.from(_classPrivateFieldGet(_held, this).keys());
_i = 0, _paths = paths;
case 4:
if (!(_i < _paths.length)) {
_context7.next = 11;
break;
}
p = _paths[_i];
_context7.next = 8;
return this.releaseByPath(p);
case 8:
_i++;
_context7.next = 4;
break;
case 11:
case "end":
return _context7.stop();
}
}, _callee7, this);
}));
function close() {
return _close.apply(this, arguments);
}
return close;
}()
}], [{
key: "generateInstanceId",
value: function generateInstanceId() {
try {
var b = crypto.randomBytes(8).toString("hex");
return "".concat(process.pid, "-").concat(Date.now(), "-").concat(b);
} catch (_unused4) {
return "".concat(process.pid, "-").concat(Date.now(), "-").concat(Math.random().toString(16).slice(2));
}
}
}]);
}();
function _startRenewLoop() {
var _this = this;
if (_classPrivateFieldGet(_renewEveryMs, this) <= 0) {
return;
}
_classPrivateFieldSet(_renewTimer, this, setInterval(function () {
_assertClassBrand(_ShardLockManager_brand, _this, _tickRenew).call(_this)["catch"](function () {});
}, _classPrivateFieldGet(_renewEveryMs, this)));
try {
_classPrivateFieldGet(_renewTimer, this).unref();
} catch (_unused5) {}
}
function _tickRenew() {
return _tickRenew2.apply(this, arguments);
}
function _tickRenew2() {
_tickRenew2 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee8() {
var now, heldPaths, _i2, _heldPaths, _heldPaths$_i, p, st, expiresAt, renewAt;
return _regeneratorRuntime().wrap(function _callee8$(_context8) {
while (1) switch (_context8.prev = _context8.next) {
case 0:
if (!(_classPrivateFieldGet(_closing, this) === true)) {
_context8.next = 2;
break;
}
return _context8.abrupt("return");
case 2:
now = Date.now();
heldPaths = Array.from(_classPrivateFieldGet(_held, this).entries());
_i2 = 0, _heldPaths = heldPaths;
case 5:
if (!(_i2 < _heldPaths.length)) {
_context8.next = 20;
break;
}
_heldPaths$_i = _slicedToArray(_heldPaths[_i2], 2), p = _heldPaths$_i[0], st = _heldPaths$_i[1];
if (!(!st || st.owner !== _classPrivateFieldGet(_instanceId, this))) {
_context8.next = 10;
break;
}
_classPrivateFieldGet(_held, this)["delete"](p);
return _context8.abrupt("continue", 17);
case 10:
expiresAt = Number(st.expiresAt);
if (Number.isFinite(expiresAt)) {
_context8.next = 13;
break;
}
return _context8.abrupt("continue", 17);
case 13:
renewAt = Math.max(0, expiresAt - Math.max(0, _classPrivateFieldGet(_renewSkewMs, this)));
if (!(now >= renewAt)) {
_context8.next = 17;
break;
}
_context8.next = 17;
return this.renewByPath(p);
case 17:
_i2++;
_context8.next = 5;
break;
case 20:
case "end":
return _context8.stop();
}
}, _callee8, this);
}));
return _tickRenew2.apply(this, arguments);
}
function _readLease(_x11) {
return _readLease2.apply(this, arguments);
}
function _readLease2() {
_readLease2 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee9(lockPath) {
var raw;
return _regeneratorRuntime().wrap(function _callee9$(_context9) {
while (1) switch (_context9.prev = _context9.next) {
case 0:
_context9.prev = 0;
_context9.next = 3;
return fs.promises.readFile(lockPath, "utf8");
case 3:
raw = _context9.sent;
if (raw) {
_context9.next = 6;
break;
}
return _context9.abrupt("return", null);
case 6:
return _context9.abrupt("return", JSON.parse(raw));
case 9:
_context9.prev = 9;
_context9.t0 = _context9["catch"](0);
return _context9.abrupt("return", null);
case 12:
case "end":
return _context9.stop();
}
}, _callee9, null, [[0, 9]]);
}));
return _readLease2.apply(this, arguments);
}
function _trySteal(_x12, _x13, _x14) {
return _trySteal2.apply(this, arguments);
}
function _trySteal2() {
_trySteal2 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee10(lockPath, payload, payloadObj) {
return _regeneratorRuntime().wrap(function _callee10$(_context10) {
while (1) switch (_context10.prev = _context10.next) {
case 0:
_context10.prev = 0;
_context10.next = 3;
return fs.promises.unlink(lockPath);
case 3:
_context10.next = 7;
break;
case 5:
_context10.prev = 5;
_context10.t0 = _context10["catch"](0);
case 7:
_context10.prev = 7;
_context10.next = 10;
return fs.promises.writeFile(lockPath, payload, {
flag: "wx"
});
case 10:
_classPrivateFieldGet(_held, this).set(lockPath, payloadObj);
return _context10.abrupt("return", true);
case 14:
_context10.prev = 14;
_context10.t1 = _context10["catch"](7);
if (!(_context10.t1 && _context10.t1.code === "EPERM")) {
_context10.next = 19;
break;
}
_context10.next = 19;
return _assertClassBrand(_ShardLockManager_brand, this, _sleep).call(this, 2);
case 19:
return _context10.abrupt("return", false);
case 20:
case "end":
return _context10.stop();
}
}, _callee10, this, [[0, 5], [7, 14]]);
}));
return _trySteal2.apply(this, arguments);
}
function _safeLockName(collection) {
return String(collection).replace(/\\/g, "/").replace(/\//g, "__").replace(/[^A-Za-z0-9._-]/g, "_");
}
function _sleep(ms) {
var m = Math.max(0, Math.floor(ms || 0));
if (m <= 0) {
return Promise.resolve();
}
return new Promise(function (r) {
return setTimeout(r, m);
});
}
;
module.exports = ShardLockManager;