]>
Commit | Line | Data |
---|---|---|
1 | import { Op, QueryTypes, Transaction } from 'sequelize' | |
2 | import { AllowNull, BelongsTo, Column, CreatedAt, Default, ForeignKey, IsInt, Model, Table, Unique, UpdatedAt } from 'sequelize-typescript' | |
3 | import { AttributesOnly } from '@shared/core-utils' | |
4 | import { VideoModel } from './video' | |
5 | ||
6 | @Table({ | |
7 | tableName: 'videoJobInfo', | |
8 | indexes: [ | |
9 | { | |
10 | fields: [ 'videoId' ], | |
11 | where: { | |
12 | videoId: { | |
13 | [Op.ne]: null | |
14 | } | |
15 | } | |
16 | } | |
17 | ] | |
18 | }) | |
19 | ||
20 | export class VideoJobInfoModel extends Model<Partial<AttributesOnly<VideoJobInfoModel>>> { | |
21 | @CreatedAt | |
22 | createdAt: Date | |
23 | ||
24 | @UpdatedAt | |
25 | updatedAt: Date | |
26 | ||
27 | @AllowNull(false) | |
28 | @Default(0) | |
29 | @IsInt | |
30 | @Column | |
31 | pendingMove: number | |
32 | ||
33 | @AllowNull(false) | |
34 | @Default(0) | |
35 | @IsInt | |
36 | @Column | |
37 | pendingTranscode: number | |
38 | ||
39 | @ForeignKey(() => VideoModel) | |
40 | @Unique | |
41 | @Column | |
42 | videoId: number | |
43 | ||
44 | @BelongsTo(() => VideoModel, { | |
45 | foreignKey: { | |
46 | allowNull: false | |
47 | }, | |
48 | onDelete: 'cascade' | |
49 | }) | |
50 | Video: VideoModel | |
51 | ||
52 | static load (videoId: number, transaction: Transaction) { | |
53 | const where = { | |
54 | videoId | |
55 | } | |
56 | ||
57 | return VideoJobInfoModel.findOne({ where, transaction }) | |
58 | } | |
59 | ||
60 | static async increaseOrCreate (videoUUID: string, column: 'pendingMove' | 'pendingTranscode'): Promise<number> { | |
61 | const options = { type: QueryTypes.SELECT as QueryTypes.SELECT, bind: { videoUUID } } | |
62 | ||
63 | const [ { pendingMove } ] = await VideoJobInfoModel.sequelize.query<{ pendingMove: number }>(` | |
64 | INSERT INTO "videoJobInfo" ("videoId", "${column}", "createdAt", "updatedAt") | |
65 | SELECT | |
66 | "video"."id" AS "videoId", 1, NOW(), NOW() | |
67 | FROM | |
68 | "video" | |
69 | WHERE | |
70 | "video"."uuid" = $videoUUID | |
71 | ON CONFLICT ("videoId") DO UPDATE | |
72 | SET | |
73 | "${column}" = "videoJobInfo"."${column}" + 1, | |
74 | "updatedAt" = NOW() | |
75 | RETURNING | |
76 | "${column}" | |
77 | `, options) | |
78 | ||
79 | return pendingMove | |
80 | } | |
81 | ||
82 | static async decrease (videoUUID: string, column: 'pendingMove' | 'pendingTranscode'): Promise<number> { | |
83 | const options = { type: QueryTypes.SELECT as QueryTypes.SELECT, bind: { videoUUID } } | |
84 | ||
85 | const [ { pendingMove } ] = await VideoJobInfoModel.sequelize.query<{ pendingMove: number }>(` | |
86 | UPDATE | |
87 | "videoJobInfo" | |
88 | SET | |
89 | "${column}" = "videoJobInfo"."${column}" - 1, | |
90 | "updatedAt" = NOW() | |
91 | FROM "video" | |
92 | WHERE | |
93 | "video"."id" = "videoJobInfo"."videoId" AND "video"."uuid" = $videoUUID | |
94 | RETURNING | |
95 | "${column}"; | |
96 | `, options) | |
97 | ||
98 | return pendingMove | |
99 | } | |
100 | } |