You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
92 lines
2.7 KiB
92 lines
2.7 KiB
1 year ago
|
'use strict';
|
||
|
|
||
|
var util = require('util');
|
||
|
var JobBase = require('./job_base');
|
||
|
var jobStatus = require('../job_status');
|
||
|
|
||
|
function JobMultiple(jobDefinition) {
|
||
|
JobBase.call(this, jobDefinition);
|
||
|
|
||
|
this.init();
|
||
|
}
|
||
|
util.inherits(JobMultiple, JobBase);
|
||
|
|
||
|
module.exports = JobMultiple;
|
||
|
|
||
|
JobMultiple.is = function (query) {
|
||
|
if (!Array.isArray(query)) {
|
||
|
return false;
|
||
|
}
|
||
|
|
||
|
// 1. From user: ['select * from ...', 'select * from ...']
|
||
|
// 2. From redis: [ { query: 'select * from ...', status: 'pending' },
|
||
|
// { query: 'select * from ...', status: 'pending' } ]
|
||
|
for (var i = 0; i < query.length; i++) {
|
||
|
if (typeof query[i] !== 'string') {
|
||
|
if (typeof query[i].query !== 'string') {
|
||
|
return false;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return true;
|
||
|
};
|
||
|
|
||
|
JobMultiple.prototype.init = function () {
|
||
|
|
||
|
if (!this.data.status) {
|
||
|
this.data.status = jobStatus.PENDING;
|
||
|
}
|
||
|
|
||
|
for (var i = 0; i < this.data.query.length; i++) {
|
||
|
if (!this.data.query[i].query && !this.data.query[i].status) {
|
||
|
this.data.query[i] = {
|
||
|
query: this.data.query[i],
|
||
|
status: jobStatus.PENDING
|
||
|
};
|
||
|
}
|
||
|
}
|
||
|
};
|
||
|
|
||
|
JobMultiple.prototype.getNextQuery = function () {
|
||
|
for (var i = 0; i < this.data.query.length; i++) {
|
||
|
if (this.data.query[i].status === jobStatus.PENDING) {
|
||
|
return this.data.query[i].query;
|
||
|
}
|
||
|
}
|
||
|
};
|
||
|
|
||
|
JobMultiple.prototype.setQuery = function (query) {
|
||
|
if (!JobMultiple.is(query)) {
|
||
|
throw new Error('You must indicate a valid SQL');
|
||
|
}
|
||
|
|
||
|
JobMultiple.super_.prototype.setQuery.call(this, query);
|
||
|
};
|
||
|
|
||
|
JobMultiple.prototype.setStatus = function (finalStatus, errorMesssage) {
|
||
|
var initialStatus = this.data.status;
|
||
|
// if transition is to "done" and there are more queries to run
|
||
|
// then job status must be "pending" instead of "done"
|
||
|
// else job status transition to done (if "running")
|
||
|
if (finalStatus === jobStatus.DONE && this.hasNextQuery()) {
|
||
|
JobMultiple.super_.prototype.setStatus.call(this, jobStatus.PENDING);
|
||
|
} else {
|
||
|
JobMultiple.super_.prototype.setStatus.call(this, finalStatus, errorMesssage);
|
||
|
}
|
||
|
|
||
|
for (var i = 0; i < this.data.query.length; i++) {
|
||
|
var isValid = JobMultiple.super_.prototype.isValidTransition(this.data.query[i].status, finalStatus);
|
||
|
|
||
|
if (isValid) {
|
||
|
this.data.query[i].status = finalStatus;
|
||
|
if (finalStatus === jobStatus.FAILED && errorMesssage) {
|
||
|
this.data.query[i].failed_reason = errorMesssage;
|
||
|
}
|
||
|
return;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
throw new Error('Cannot set status from ' + initialStatus + ' to ' + finalStatus);
|
||
|
};
|