CartoDB-SQL-API/batch/job_backend.js

'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;