X-Git-Url: https://git.immae.eu/?a=blobdiff_plain;f=server%2Flib%2Frequest%2Frequest-video-event-scheduler.ts;h=c9d1651177a89ce7bec12577da74bc521077e65a;hb=70c065d64c330196d371941d9294a55da6e3aa37;hp=6e5306c7d1ac31ad784097221bf7de8ee657302b;hpb=65fcc3119c334b75dd13bcfdebf186afdc580a8f;p=github%2FChocobozzz%2FPeerTube.git diff --git a/server/lib/request/request-video-event-scheduler.ts b/server/lib/request/request-video-event-scheduler.ts index 6e5306c7d..c9d165117 100644 --- a/server/lib/request/request-video-event-scheduler.ts +++ b/server/lib/request/request-video-event-scheduler.ts @@ -1,12 +1,21 @@ -const db = require('../../initializers/database') -import { BaseRequestScheduler } from './base-request-scheduler' +import * as Sequelize from 'sequelize' + +import { database as db } from '../../initializers/database' +import { AbstractRequestScheduler } from './abstract-request-scheduler' import { REQUESTS_VIDEO_EVENT_LIMIT_PODS, REQUESTS_VIDEO_EVENT_LIMIT_PER_POD, REQUEST_VIDEO_EVENT_ENDPOINT } from '../../initializers' -class RequestVideoEventScheduler extends BaseRequestScheduler { +export type RequestVideoEventSchedulerOptions = { + type: string + videoId: string + count?: number + transaction?: Sequelize.Transaction +} + +class RequestVideoEventScheduler extends AbstractRequestScheduler { constructor () { super() @@ -25,7 +34,7 @@ class RequestVideoEventScheduler extends BaseRequestScheduler { return db.RequestVideoEvent } - buildRequestObjects (eventsToProcess) { + buildRequestObjects (eventsToProcess: { [ toPodId: number ]: any }[]) { const requestsToMakeGrouped = {} /* Example: @@ -87,16 +96,10 @@ class RequestVideoEventScheduler extends BaseRequestScheduler { return requestsToMakeGrouped } - // { type, videoId, count?, transaction? } - createRequest (options, callback) { - const type = options.type - const videoId = options.videoId - const transaction = options.transaction - let count = options.count - + createRequest ({ type, videoId, count, transaction }: RequestVideoEventSchedulerOptions, callback: (err: Error) => void) { if (count === undefined) count = 1 - const dbRequestOptions: { transaction?: any } = {} + const dbRequestOptions: Sequelize.CreateOptions = {} if (transaction) dbRequestOptions.transaction = transaction const createQuery = {