vulpes
Version:
Job management framework
101 lines (81 loc) • 5.22 kB
JavaScript
;
var _createClass = function () { function defineProperties(target, props) { for (var i = 0; i < props.length; i++) { var descriptor = props[i]; descriptor.enumerable = descriptor.enumerable || false; descriptor.configurable = true; if ("value" in descriptor) descriptor.writable = true; Object.defineProperty(target, descriptor.key, descriptor); } } return function (Constructor, protoProps, staticProps) { if (protoProps) defineProperties(Constructor.prototype, protoProps); if (staticProps) defineProperties(Constructor, staticProps); return Constructor; }; }();
var _get = function get(object, property, receiver) { if (object === null) object = Function.prototype; var desc = Object.getOwnPropertyDescriptor(object, property); if (desc === undefined) { var parent = Object.getPrototypeOf(object); if (parent === null) { return undefined; } else { return get(parent, property, receiver); } } else if ("value" in desc) { return desc.value; } else { var getter = desc.get; if (getter === undefined) { return undefined; } return getter.call(receiver); } };
function _classCallCheck(instance, Constructor) { if (!(instance instanceof Constructor)) { throw new TypeError("Cannot call a class as a function"); } }
function _possibleConstructorReturn(self, call) { if (!self) { throw new ReferenceError("this hasn't been initialised - super() hasn't been called"); } return call && (typeof call === "object" || typeof call === "function") ? call : self; }
function _inherits(subClass, superClass) { if (typeof superClass !== "function" && superClass !== null) { throw new TypeError("Super expression must either be null or a function, not " + typeof superClass); } subClass.prototype = Object.create(superClass && superClass.prototype, { constructor: { value: subClass, enumerable: false, writable: true, configurable: true } }); if (superClass) Object.setPrototypeOf ? Object.setPrototypeOf(subClass, superClass) : subClass.__proto__ = superClass; }
var ms = require("ms");
var VError = require("verror");
var Helper = require("./Helper.js");
var _require = require("../time.js"),
getTimestamp = _require.getTimestamp;
var _require2 = require("../symbols.js"),
ERROR_CODE_INVALID_JOB_STATUS = _require2.ERROR_CODE_INVALID_JOB_STATUS,
JOB_RESULT_TYPE_TIMEOUT = _require2.JOB_RESULT_TYPE_TIMEOUT,
JOB_STATUS_RUNNING = _require2.JOB_STATUS_RUNNING;
var PROCESSING_LIMIT = 50;
/**
* Helper for timing-out jobs
* @augments Helper
* @memberof module:Vulpes
*/
var TimeoutHelper = function (_Helper) {
_inherits(TimeoutHelper, _Helper);
function TimeoutHelper() {
var interval = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : ms("15s");
_classCallCheck(this, TimeoutHelper);
var _this = _possibleConstructorReturn(this, (TimeoutHelper.__proto__ || Object.getPrototypeOf(TimeoutHelper)).call(this));
_this._interval = interval;
return _this;
}
_createClass(TimeoutHelper, [{
key: "attach",
value: function attach(service) {
var _this2 = this;
_get(TimeoutHelper.prototype.__proto__ || Object.getPrototypeOf(TimeoutHelper.prototype), "attach", this).call(this, service);
this._timer = setInterval(this.checkJobs.bind(this), this._interval);
setTimeout(function () {
return _this2.checkJobs();
}, 0);
}
}, {
key: "checkJobs",
value: function checkJobs() {
var _this3 = this;
this.service.queryJobs({
timeLimit: function timeLimit(_timeLimit) {
return _timeLimit !== null;
},
status: JOB_STATUS_RUNNING
}).then(function (runningJobs) {
return runningJobs.length > PROCESSING_LIMIT ? runningJobs.splice(0, PROCESSING_LIMIT) : runningJobs;
}).then(function (runningJobs) {
var work = Promise.resolve();
runningJobs.forEach(function (job) {
work = work.then(function () {
var now = getTimestamp();
if (now - job.times.started >= job.timeLimit) {
return _this3.service.stopJob(job.id, JOB_RESULT_TYPE_TIMEOUT).catch(function (err) {
var info = VError.info(err);
if (info && info.code === ERROR_CODE_INVALID_JOB_STATUS) {
// Bob status was changed while the helper was
// preparing - we can ignore this error..
return;
}
throw err;
});
}
});
});
});
}
}, {
key: "shutdown",
value: function shutdown() {
clearInterval(this._timer);
_get(TimeoutHelper.prototype.__proto__ || Object.getPrototypeOf(TimeoutHelper.prototype), "shutdown", this).call(this);
}
}]);
return TimeoutHelper;
}(Helper);
module.exports = TimeoutHelper;