Init commit

server/core/lib/job-queue/handlers/live-video-start.ts

@@ -0,0 +1,64 @@
+import { LiveVideoStartPayload } from '@peertube/peertube-models'
+import { logger, loggerTagsFactory } from '@server/helpers/logger.js'
+import { startScheduledLiveSession } from '@server/controllers/api/live-stream.js'
+import { CONFIG } from '@server/initializers/config.js'
+import { LiveVideoProcessManager } from '@server/lib/live-videos/live-video-process-manager.js'
+import { LIVE_VIDEO_START_LEAD_TIME_MS } from '@server/lib/live-videos/shared.js'
+import { LiveVideoModel, LiveVideoStatus } from '@server/models/video/live-video.js'
+import { Job } from 'bullmq'
+import { setTimeout as wait } from 'node:timers/promises'
+
+const lTags = loggerTagsFactory('jobs', 'live-video-start')
+const LIVE_VIDEO_START_POLL_MS = 5_000
+
+export async function processLiveVideoStart (job: Job) {
+  const payload = job.data as LiveVideoStartPayload
+  let liveVideo: LiveVideoModel
+
+  while (true) {
+    const currentLiveVideo = await LiveVideoModel.loadById(payload.liveVideoId)
+    if (!currentLiveVideo) {
+      logger.warn('Cannot find live video %d for scheduled start job %s.', payload.liveVideoId, job.id, lTags())
+      return
+    }
+
+    if (currentLiveVideo.status !== LiveVideoStatus.SCHEDULED) {
+      logger.info('Skipping scheduled start of live video %d because status is %s.', currentLiveVideo.id, currentLiveVideo.status, lTags())
+      return
+    }
+
+    const now = Date.now()
+    const startAt = currentLiveVideo.startTime.getTime()
+    const endAt = currentLiveVideo.endTime.getTime()
+    if (endAt <= now) {
+      logger.info('Skipping scheduled start of live video %d because it is already finished.', currentLiveVideo.id, lTags())
+      return
+    }
+
+    const launchAt = startAt - LIVE_VIDEO_START_LEAD_TIME_MS
+    const waitMs = launchAt - now
+    if (waitMs <= 0) {
+      liveVideo = currentLiveVideo
+      break
+    }
+
+    await wait(Math.min(waitMs, LIVE_VIDEO_START_POLL_MS))
+  }
+
+  if (CONFIG.LIVE.ENABLE_SOURCE_START) {
+    await LiveVideoProcessManager.Instance.start(liveVideo)
+  } else {
+    logger.info('Skipping source-side scheduled live start for live video %d because live.enable_source_start is disabled.', liveVideo.id, lTags())
+  }
+
+  const sinkWaitMs = liveVideo.startTime.getTime() - Date.now()
+  if (sinkWaitMs > 0) {
+    await wait(sinkWaitMs)
+  }
+
+  if (CONFIG.LIVE.ENABLE_SINK_START) {
+    await startScheduledLiveSession(liveVideo.targetCamera, { keepAliveUntil: liveVideo.endTime })
+  } else {
+    logger.info('Skipping sink-side scheduled live start for live video %d because live.enable_sink_start is disabled.', liveVideo.id, lTags())
+  }
+}