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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
|
import { values } from 'lodash'
import * as Sequelize from 'sequelize'
import { JobCategory, JobState } from '../../../shared/models/job.model'
import { JOB_CATEGORIES, JOB_STATES } from '../../initializers'
import { addMethodsToModel, getSort } from '../utils'
import { JobAttributes, JobInstance, JobMethods } from './job-interface'
let Job: Sequelize.Model<JobInstance, JobAttributes>
let listWithLimitByCategory: JobMethods.ListWithLimitByCategory
let listForApi: JobMethods.ListForApi
let toFormattedJSON: JobMethods.ToFormattedJSON
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,
listForApi
]
const instanceMethods = [
toFormattedJSON
]
addMethodsToModel(Job, classMethods, instanceMethods)
return Job
}
toFormattedJSON = function (this: JobInstance) {
return {
id: this.id,
state: this.state,
category: this.category,
handlerName: this.handlerName,
handlerInputData: this.handlerInputData,
createdAt: this.createdAt,
updatedAt: this.updatedAt
}
}
// ---------------------------------------------------------------------------
listWithLimitByCategory = function (limit: number, state: JobState, jobCategory: JobCategory) {
const query = {
order: [
[ 'id', 'ASC' ]
],
limit: limit,
where: {
state,
category: jobCategory
}
}
return Job.findAll(query)
}
listForApi = function (start: number, count: number, sort: string) {
const query = {
offset: start,
limit: count,
order: [ getSort(sort) ]
}
return Job.findAndCountAll(query).then(({ rows, count }) => {
return {
data: rows,
total: count
}
})
}
|
