aboutsummaryrefslogtreecommitdiffhomepage
path: root/server/models/job/job.ts
blob: c2d088090accc4068bce6d45694a69b10190a8ff (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
import { values } from 'lodash'
import * as Sequelize from 'sequelize'

import { JOB_STATES, JOB_CATEGORIES } from '../../initializers'

import { addMethodsToModel } from '../utils'
import {
  JobInstance,
  JobAttributes,

  JobMethods
} from './job-interface'
import { JobCategory, JobState } from '../../../shared/models/job.model'

let Job: Sequelize.Model<JobInstance, JobAttributes>
let listWithLimitByCategory: JobMethods.ListWithLimitByCategory

export default function defineJob (sequelize: Sequelize.Sequelize, DataTypes: Sequelize.DataTypes) {
  Job = sequelize.define<JobInstance, JobAttributes>('Job',
    {
      state: {
        type: DataTypes.ENUM(values(JOB_STATES)),
        allowNull: false
      },
      category: {
        type: DataTypes.ENUM(values(JOB_CATEGORIES)),
        allowNull: false
      },
      handlerName: {
        type: DataTypes.STRING,
        allowNull: false
      },
      handlerInputData: {
        type: DataTypes.JSON,
        allowNull: true
      }
    },
    {
      indexes: [
        {
          fields: [ 'state', 'category' ]
        }
      ]
    }
  )

  const classMethods = [ listWithLimitByCategory ]
  addMethodsToModel(Job, classMethods)

  return Job
}

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

listWithLimitByCategory = function (limit: number, state: JobState, jobCategory: JobCategory) {
  const query = {
    order: [
      [ 'id', 'ASC' ]
    ],
    limit: limit,
    where: {
      state,
      category: jobCategory
    }
  }

  return Job.findAll(query)
}