125 lines
3.1 KiB
JavaScript
125 lines
3.1 KiB
JavaScript
'use strict';
|
|
|
|
var util = require('util');
|
|
var EventEmitter = require('events').EventEmitter;
|
|
var uuid = require('node-uuid');
|
|
|
|
function JobBackend(metadataBackend) {
|
|
EventEmitter.call(this);
|
|
this.metadataBackend = metadataBackend;
|
|
this.db = 5;
|
|
}
|
|
util.inherits(JobBackend, EventEmitter);
|
|
|
|
JobBackend.prototype.create = function (username, sql, callback) {
|
|
var self = this;
|
|
var job_id = uuid.v4();
|
|
var now = new Date().toISOString();
|
|
var redisParams = [
|
|
'job:' + job_id,
|
|
'user', username,
|
|
'status', 'pending',
|
|
'query', sql,
|
|
'created_at', now,
|
|
'updated_at', now
|
|
];
|
|
|
|
this.metadataBackend.redisCmd(this.db, 'HMSET', redisParams , function (err) {
|
|
if (err) {
|
|
return callback(err);
|
|
}
|
|
|
|
self.get(job_id, callback);
|
|
});
|
|
};
|
|
|
|
JobBackend.prototype.get = function (job_id, callback) {
|
|
var redisParams = [
|
|
'job:' + job_id,
|
|
'user',
|
|
'status',
|
|
'query',
|
|
'created_at',
|
|
'updated_at',
|
|
'failed_reason'
|
|
];
|
|
|
|
this.metadataBackend.redisCmd(this.db, 'HMGET', redisParams , function (err, jobValues) {
|
|
if (err) {
|
|
return callback(err);
|
|
}
|
|
|
|
function isJobFound(jobValues) {
|
|
return jobValues[0] && jobValues[1] && jobValues[2] && jobValues[3] && jobValues[4];
|
|
}
|
|
|
|
if (!isJobFound(jobValues)) {
|
|
return callback(new Error('Job with id ' + job_id + ' not found'));
|
|
}
|
|
|
|
callback(null, {
|
|
job_id: job_id,
|
|
user: jobValues[0],
|
|
status: jobValues[1],
|
|
query: jobValues[2],
|
|
created_at: jobValues[3],
|
|
updated_at: jobValues[4],
|
|
failed_reason: jobValues[5] ? jobValues[5] : undefined
|
|
});
|
|
});
|
|
};
|
|
|
|
JobBackend.prototype.setRunning = function (job) {
|
|
var self = this;
|
|
var redisParams = [
|
|
'job:' + job.job_id,
|
|
'status', 'running',
|
|
'updated_at', new Date().toISOString()
|
|
];
|
|
|
|
this.metadataBackend.redisCmd(this.db, 'HMSET', redisParams, function (err) {
|
|
if (err) {
|
|
return self.emit('error', err);
|
|
}
|
|
|
|
self.emit('running', job);
|
|
});
|
|
};
|
|
|
|
JobBackend.prototype.setDone = function (job) {
|
|
var self = this;
|
|
var redisParams = [
|
|
'job:' + job.job_id,
|
|
'status', 'done',
|
|
'updated_at', new Date().toISOString()
|
|
];
|
|
|
|
this.metadataBackend.redisCmd(this.db, 'HMSET', redisParams , function (err) {
|
|
if (err) {
|
|
return self.emit('error', err);
|
|
}
|
|
|
|
self.emit('done', job);
|
|
});
|
|
};
|
|
|
|
JobBackend.prototype.setFailed = function (job, err) {
|
|
var self = this;
|
|
var redisParams = [
|
|
'job:' + job.job_id,
|
|
'status', 'failed',
|
|
'failed_reason', err.message,
|
|
'updated_at', new Date().toISOString()
|
|
];
|
|
|
|
this.metadataBackend.redisCmd(this.db, 'HMSET', redisParams , function (err) {
|
|
if (err) {
|
|
return self.emit('error', err);
|
|
}
|
|
|
|
self.emit('failed', job);
|
|
});
|
|
};
|
|
|
|
module.exports = JobBackend;
|