import { Op, QueryTypes, Transaction } from 'sequelize' import { AllowNull, BelongsTo, Column, CreatedAt, Default, ForeignKey, IsInt, Model, Table, Unique, UpdatedAt } from 'sequelize-typescript' import { forceNumber } from '@peertube/peertube-core-utils' import { AttributesOnly } from '@peertube/peertube-typescript-utils' import { VideoModel } from './video.js' export type VideoJobInfoColumnType = 'pendingMove' | 'pendingTranscode' @Table({ tableName: 'videoJobInfo', indexes: [ { fields: [ 'videoId' ], where: { videoId: { [Op.ne]: null } } } ] }) export class VideoJobInfoModel extends Model>> { @CreatedAt createdAt: Date @UpdatedAt updatedAt: Date @AllowNull(false) @Default(0) @IsInt @Column pendingMove: number @AllowNull(false) @Default(0) @IsInt @Column pendingTranscode: number @ForeignKey(() => VideoModel) @Unique @Column videoId: number @BelongsTo(() => VideoModel, { foreignKey: { allowNull: false }, onDelete: 'cascade' }) Video: Awaited static load (videoId: number, transaction?: Transaction) { const where = { videoId } return VideoJobInfoModel.findOne({ where, transaction }) } static async increaseOrCreate (videoUUID: string, column: VideoJobInfoColumnType, amountArg = 1): Promise { const options = { type: QueryTypes.SELECT as QueryTypes.SELECT, bind: { videoUUID } } const amount = forceNumber(amountArg) const [ result ] = await VideoJobInfoModel.sequelize.query<{ pendingMove: number }>(` INSERT INTO "videoJobInfo" ("videoId", "${column}", "createdAt", "updatedAt") SELECT "video"."id" AS "videoId", ${amount}, NOW(), NOW() FROM "video" WHERE "video"."uuid" = $videoUUID ON CONFLICT ("videoId") DO UPDATE SET "${column}" = "videoJobInfo"."${column}" + ${amount}, "updatedAt" = NOW() RETURNING "${column}" `, options) return result[column] } static async decrease (videoUUID: string, column: VideoJobInfoColumnType): Promise { const options = { type: QueryTypes.SELECT as QueryTypes.SELECT, bind: { videoUUID } } const result = await VideoJobInfoModel.sequelize.query(` UPDATE "videoJobInfo" SET "${column}" = "videoJobInfo"."${column}" - 1, "updatedAt" = NOW() FROM "video" WHERE "video"."id" = "videoJobInfo"."videoId" AND "video"."uuid" = $videoUUID RETURNING "${column}"; `, options) if (result.length === 0) return undefined return result[0][column] } static async abortAllTasks (videoUUID: string, column: VideoJobInfoColumnType): Promise { const options = { type: QueryTypes.UPDATE as QueryTypes.UPDATE, bind: { videoUUID } } await VideoJobInfoModel.sequelize.query(` UPDATE "videoJobInfo" SET "${column}" = 0, "updatedAt" = NOW() FROM "video" WHERE "video"."id" = "videoJobInfo"."videoId" AND "video"."uuid" = $videoUUID `, options) } }