X-Git-Url: https://git.immae.eu/?a=blobdiff_plain;f=server%2Flib%2Frequest%2Frequest-video-qadu-scheduler.ts;fp=server%2Flib%2Frequest%2Frequest-video-qadu-scheduler.ts;h=5ec7de9c206e1291aa2a6bb4bd9f13d0473f3991;hb=6fcd19ba737f1f5614a56c6925adb882dea43b8d;hp=d7169cc815f46c73b3e69fb8018ef824b447d3c9;hpb=5fe7e898316e18369c3e1aba307b55077adc7bfb;p=github%2FChocobozzz%2FPeerTube.git diff --git a/server/lib/request/request-video-qadu-scheduler.ts b/server/lib/request/request-video-qadu-scheduler.ts index d7169cc81..5ec7de9c2 100644 --- a/server/lib/request/request-video-qadu-scheduler.ts +++ b/server/lib/request/request-video-qadu-scheduler.ts @@ -9,6 +9,7 @@ import { REQUEST_VIDEO_QADU_ENDPOINT, REQUEST_VIDEO_QADU_TYPES } from '../../initializers' +import { RequestsVideoQaduGrouped } from '../../models' import { RequestVideoQaduType } from '../../../shared' export type RequestVideoQaduSchedulerOptions = { @@ -17,7 +18,7 @@ export type RequestVideoQaduSchedulerOptions = { transaction?: Sequelize.Transaction } -class RequestVideoQaduScheduler extends AbstractRequestScheduler { +class RequestVideoQaduScheduler extends AbstractRequestScheduler { constructor () { super() @@ -36,7 +37,7 @@ class RequestVideoQaduScheduler extends AbstractRequestScheduler { return db.RequestVideoQadu } - buildRequestObjects (requests: { [ toPodId: number ]: any }[]) { + buildRequestObjects (requests: RequestsVideoQaduGrouped) { const requestsToMakeGrouped = {} Object.keys(requests).forEach(toPodId => { @@ -105,20 +106,18 @@ class RequestVideoQaduScheduler extends AbstractRequestScheduler { return requestsToMakeGrouped } - createRequest ({ type, videoId, transaction }: RequestVideoQaduSchedulerOptions, callback: (err: Error) => void) { + createRequest ({ type, videoId, transaction }: RequestVideoQaduSchedulerOptions) { const dbRequestOptions: Sequelize.BulkCreateOptions = {} if (transaction) dbRequestOptions.transaction = transaction // Send the update to all our friends - db.Pod.listAllIds(transaction, function (err, podIds) { - if (err) return callback(err) - + return db.Pod.listAllIds(transaction).then(podIds => { const queries = [] podIds.forEach(podId => { queries.push({ type, videoId, podId }) }) - return db.RequestVideoQadu.bulkCreate(queries, dbRequestOptions).asCallback(callback) + return db.RequestVideoQadu.bulkCreate(queries, dbRequestOptions) }) } }