Server: add job scheduler to transcode video files

[github/Chocobozzz/PeerTube.git] / server / models / job.js

'use strict'

const values = require('lodash/values')

const constants = require('../initializers/constants')

// ---------------------------------------------------------------------------

module.exports = function (sequelize, DataTypes) {
  const Job = sequelize.define('Job',
    {
      state: {
        type: DataTypes.ENUM(values(constants.JOB_STATES)),
        allowNull: false
      },
      handlerName: {
        type: DataTypes.STRING,
        allowNull: false
      },
      handlerInputData: {
        type: DataTypes.JSON,
        allowNull: true
      }
    },
    {
      indexes: [
        {
          fields: [ 'state' ]
        }
      ],
      classMethods: {
        listWithLimit
      }
    }
  )

  return Job
}

// ---------------------------------------------------------------------------

function listWithLimit (limit, callback) {
  const query = {
    order: [
      [ 'id', 'ASC' ]
    ],
    limit: limit,
    where: {
      state: constants.JOB_STATES.PENDING
    }
  }

  return this.findAll(query).asCallback(callback)
}