PeerTube/server/lib/job-queue/handlers/video-live-ending.ts

273 lines
9.1 KiB
TypeScript
Raw Normal View History

2022-08-08 03:42:08 -05:00
import { Job } from 'bullmq'
import { readdir, remove } from 'fs-extra'
import { join } from 'path'
import { getLocalVideoActivityPubUrl } from '@server/lib/activitypub/url'
import { federateVideoIfNeeded } from '@server/lib/activitypub/videos'
import { cleanupAndDestroyPermanentLive, cleanupTMPLiveFiles, cleanupUnsavedNormalLive } from '@server/lib/live'
import { generateHLSMasterPlaylistFilename, generateHlsSha256SegmentsFilename, getLiveReplayBaseDirectory } from '@server/lib/paths'
2020-11-20 10:16:55 -06:00
import { generateVideoMiniature } from '@server/lib/thumbnail'
import { generateHlsPlaylistResolutionFromTS } from '@server/lib/transcoding/hls-transcoding'
import { VideoPathManager } from '@server/lib/video-path-manager'
Add support for saving video files to object storage (#4290) * Add support for saving video files to object storage * Add support for custom url generation on s3 stored files Uses two config keys to support url generation that doesn't directly go to (compatible s3). Can be used to generate urls to any cache server or CDN. * Upload files to s3 concurrently and delete originals afterwards * Only publish after move to object storage is complete * Use base url instead of url template * Fix mistyped config field * Add rudenmentary way to download before transcode * Implement Chocobozzz suggestions https://github.com/Chocobozzz/PeerTube/pull/4290#issuecomment-891670478 The remarks in question: Try to use objectStorage prefix instead of s3 prefix for your function/variables/config names Prefer to use a tree for the config: s3.streaming_playlists_bucket -> object_storage.streaming_playlists.bucket Use uppercase for config: S3.STREAMING_PLAYLISTS_BUCKETINFO.bucket -> OBJECT_STORAGE.STREAMING_PLAYLISTS.BUCKET (maybe BUCKET_NAME instead of BUCKET) I suggest to rename moveJobsRunning to pendingMovingJobs (or better, create a dedicated videoJobInfo table with a pendingMove & videoId columns so we could also use this table to track pending transcoding jobs) https://github.com/Chocobozzz/PeerTube/pull/4290/files#diff-3e26d41ca4bda1de8e1747af70ca2af642abcc1e9e0bfb94239ff2165acfbde5R19 uses a string instead of an integer I think we should store the origin object storage URL in fileUrl, without base_url injection. Instead, inject the base_url at "runtime" so admins can easily change this configuration without running a script to update DB URLs * Import correct function * Support multipart upload * Remove import of node 15.0 module stream/promises * Extend maximum upload job length Using the same value as for redundancy downloading seems logical * Use dynamic part size for really large uploads Also adds very small part size for local testing * Fix decreasePendingMove query * Resolve various PR comments * Move to object storage after optimize * Make upload size configurable and increase default * Prune webtorrent files that are stored in object storage * Move files after transcoding jobs * Fix federation * Add video path manager * Support move to external storage job in client * Fix live object storage tests Co-authored-by: Chocobozzz <me@florianbigard.com>
2021-08-17 01:26:20 -05:00
import { moveToNextState } from '@server/lib/video-state'
import { VideoModel } from '@server/models/video/video'
2022-05-03 04:38:07 -05:00
import { VideoBlacklistModel } from '@server/models/video/video-blacklist'
2020-11-04 07:16:57 -06:00
import { VideoFileModel } from '@server/models/video/video-file'
2020-10-26 10:44:23 -05:00
import { VideoLiveModel } from '@server/models/video/video-live'
import { VideoLiveReplaySettingModel } from '@server/models/video/video-live-replay-setting'
2022-05-03 04:38:07 -05:00
import { VideoLiveSessionModel } from '@server/models/video/video-live-session'
import { VideoStreamingPlaylistModel } from '@server/models/video/video-streaming-playlist'
2022-05-03 04:38:07 -05:00
import { MVideo, MVideoLive, MVideoLiveSession, MVideoWithAllFiles } from '@server/types/models'
import { ffprobePromise, getAudioStream, getVideoStreamDimensionsInfo, getVideoStreamFPS } from '@shared/ffmpeg'
2020-11-06 03:57:40 -06:00
import { ThumbnailType, VideoLiveEndingPayload, VideoState } from '@shared/models'
import { logger, loggerTagsFactory } from '../../../helpers/logger'
const lTags = loggerTagsFactory('live', 'job')
2021-08-27 07:32:44 -05:00
async function processVideoLiveEnding (job: Job) {
const payload = job.data as VideoLiveEndingPayload
logger.info('Processing video live ending for %s.', payload.videoId, { payload, ...lTags() })
2020-11-04 07:16:57 -06:00
function logError () {
logger.warn('Video live %d does not exist anymore. Cannot process live ending.', payload.videoId, lTags())
2020-11-04 07:16:57 -06:00
}
const video = await VideoModel.load(payload.videoId)
2020-10-26 10:44:23 -05:00
const live = await VideoLiveModel.loadByVideoId(payload.videoId)
2022-05-03 04:38:07 -05:00
const liveSession = await VideoLiveSessionModel.load(payload.liveSessionId)
2020-10-26 10:44:23 -05:00
if (!video || !live || !liveSession) {
2020-11-04 07:16:57 -06:00
logError()
return
}
2022-10-04 04:17:37 -05:00
const permanentLive = live.permanentLive
liveSession.endingProcessed = true
await liveSession.save()
if (liveSession.saveReplay !== true) {
return cleanupLiveAndFederate({ permanentLive, video, streamingPlaylistId: payload.streamingPlaylistId })
2020-10-26 10:44:23 -05:00
}
if (permanentLive) {
await saveReplayToExternalVideo({
liveVideo: video,
liveSession,
publishedAt: payload.publishedAt,
replayDirectory: payload.replayDirectory
})
return cleanupLiveAndFederate({ permanentLive, video, streamingPlaylistId: payload.streamingPlaylistId })
}
return replaceLiveByReplay({
video,
liveSession,
live,
permanentLive,
replayDirectory: payload.replayDirectory
})
2020-10-26 10:44:23 -05:00
}
// ---------------------------------------------------------------------------
export {
2021-06-16 08:14:41 -05:00
processVideoLiveEnding
2020-10-26 10:44:23 -05:00
}
// ---------------------------------------------------------------------------
2022-05-03 04:38:07 -05:00
async function saveReplayToExternalVideo (options: {
liveVideo: MVideo
liveSession: MVideoLiveSession
publishedAt: string
replayDirectory: string
}) {
const { liveVideo, liveSession, publishedAt, replayDirectory } = options
const replaySettings = await VideoLiveReplaySettingModel.load(liveSession.replaySettingId)
const replayVideo = new VideoModel({
name: `${liveVideo.name} - ${new Date(publishedAt).toLocaleString()}`,
isLive: false,
state: VideoState.TO_TRANSCODE,
duration: 0,
remote: liveVideo.remote,
category: liveVideo.category,
licence: liveVideo.licence,
language: liveVideo.language,
commentsEnabled: liveVideo.commentsEnabled,
downloadEnabled: liveVideo.downloadEnabled,
2022-05-03 04:38:07 -05:00
waitTranscoding: true,
nsfw: liveVideo.nsfw,
description: liveVideo.description,
support: liveVideo.support,
privacy: replaySettings.privacy,
channelId: liveVideo.channelId
}) as MVideoWithAllFiles
replayVideo.Thumbnails = []
replayVideo.VideoFiles = []
replayVideo.VideoStreamingPlaylists = []
replayVideo.url = getLocalVideoActivityPubUrl(replayVideo)
await replayVideo.save()
liveSession.replayVideoId = replayVideo.id
2022-05-03 04:38:07 -05:00
await liveSession.save()
// If live is blacklisted, also blacklist the replay
const blacklist = await VideoBlacklistModel.loadByVideoId(liveVideo.id)
if (blacklist) {
await VideoBlacklistModel.create({
videoId: replayVideo.id,
unfederated: blacklist.unfederated,
reason: blacklist.reason,
type: blacklist.type
})
}
await assignReplayFilesToVideo({ video: replayVideo, replayDirectory })
await remove(replayDirectory)
for (const type of [ ThumbnailType.MINIATURE, ThumbnailType.PREVIEW ]) {
const image = await generateVideoMiniature({ video: replayVideo, videoFile: replayVideo.getMaxQualityFile(), type })
await replayVideo.addAndSaveThumbnail(image)
}
await moveToNextState({ video: replayVideo, isNewVideo: true })
}
2022-05-03 04:38:07 -05:00
async function replaceLiveByReplay (options: {
video: MVideo
2022-05-03 04:38:07 -05:00
liveSession: MVideoLiveSession
live: MVideoLive
permanentLive: boolean
2022-05-03 04:38:07 -05:00
replayDirectory: string
}) {
const { video, liveSession, live, permanentLive, replayDirectory } = options
2022-05-03 04:38:07 -05:00
const replaySettings = await VideoLiveReplaySettingModel.load(liveSession.replaySettingId)
const videoWithFiles = await VideoModel.loadFull(video.id)
const hlsPlaylist = videoWithFiles.getHLSPlaylist()
await cleanupTMPLiveFiles(videoWithFiles, hlsPlaylist)
2020-10-26 10:44:23 -05:00
2020-10-27 10:06:24 -05:00
await live.destroy()
videoWithFiles.isLive = false
videoWithFiles.privacy = replaySettings.privacy
videoWithFiles.waitTranscoding = true
videoWithFiles.state = VideoState.TO_TRANSCODE
2020-10-28 04:49:20 -05:00
await videoWithFiles.save()
2022-05-03 04:38:07 -05:00
liveSession.replayVideoId = videoWithFiles.id
2022-05-03 04:38:07 -05:00
await liveSession.save()
2020-10-26 10:44:23 -05:00
2020-11-03 08:33:30 -06:00
await VideoFileModel.removeHLSFilesOfVideoId(hlsPlaylist.id)
2021-07-23 04:20:00 -05:00
// Reset playlist
2020-11-03 08:33:30 -06:00
hlsPlaylist.VideoFiles = []
2021-07-23 04:20:00 -05:00
hlsPlaylist.playlistFilename = generateHLSMasterPlaylistFilename()
hlsPlaylist.segmentsSha256Filename = generateHlsSha256SegmentsFilename()
await hlsPlaylist.save()
2020-11-03 08:33:30 -06:00
2022-05-03 04:38:07 -05:00
await assignReplayFilesToVideo({ video: videoWithFiles, replayDirectory })
if (permanentLive) { // Remove session replay
await remove(replayDirectory)
} else { // We won't stream again in this live, we can delete the base replay directory
await remove(getLiveReplayBaseDirectory(videoWithFiles))
}
// Regenerate the thumbnail & preview?
if (videoWithFiles.getMiniature().automaticallyGenerated === true) {
const miniature = await generateVideoMiniature({
video: videoWithFiles,
videoFile: videoWithFiles.getMaxQualityFile(),
type: ThumbnailType.MINIATURE
})
2022-05-03 04:38:07 -05:00
await videoWithFiles.addAndSaveThumbnail(miniature)
}
if (videoWithFiles.getPreview().automaticallyGenerated === true) {
const preview = await generateVideoMiniature({
video: videoWithFiles,
videoFile: videoWithFiles.getMaxQualityFile(),
type: ThumbnailType.PREVIEW
})
2022-05-03 04:38:07 -05:00
await videoWithFiles.addAndSaveThumbnail(preview)
}
2022-05-03 04:38:07 -05:00
// We consider this is a new video
await moveToNextState({ video: videoWithFiles, isNewVideo: true })
}
2022-05-03 04:38:07 -05:00
async function assignReplayFilesToVideo (options: {
video: MVideo
replayDirectory: string
}) {
const { video, replayDirectory } = options
const concatenatedTsFiles = await readdir(replayDirectory)
for (const concatenatedTsFile of concatenatedTsFiles) {
const inputFileMutexReleaser = await VideoPathManager.Instance.lockFiles(video.uuid)
await video.reload()
2020-12-04 08:10:13 -06:00
const concatenatedTsFilePath = join(replayDirectory, concatenatedTsFile)
2020-12-02 03:07:26 -06:00
const probe = await ffprobePromise(concatenatedTsFilePath)
const { audioStream } = await getAudioStream(concatenatedTsFilePath, probe)
const { resolution } = await getVideoStreamDimensionsInfo(concatenatedTsFilePath, probe)
const fps = await getVideoStreamFPS(concatenatedTsFilePath, probe)
2020-12-02 03:07:26 -06:00
try {
await generateHlsPlaylistResolutionFromTS({
video,
inputFileMutexReleaser,
concatenatedTsFilePath,
resolution,
fps,
isAAC: audioStream?.codec_name === 'aac'
})
} catch (err) {
logger.error('Cannot generate HLS playlist resolution from TS files.', { err })
}
inputFileMutexReleaser()
2020-11-03 08:33:30 -06:00
}
2020-10-28 04:49:20 -05:00
return video
}
2020-11-06 03:57:40 -06:00
2022-05-03 04:38:07 -05:00
async function cleanupLiveAndFederate (options: {
video: MVideo
permanentLive: boolean
streamingPlaylistId: number
2022-05-03 04:38:07 -05:00
}) {
const { permanentLive, video, streamingPlaylistId } = options
2020-12-03 07:10:54 -06:00
const streamingPlaylist = await VideoStreamingPlaylistModel.loadWithVideo(streamingPlaylistId)
if (streamingPlaylist) {
if (permanentLive) {
await cleanupAndDestroyPermanentLive(video, streamingPlaylist)
} else {
await cleanupUnsavedNormalLive(video, streamingPlaylist)
}
}
2022-05-25 08:18:29 -05:00
try {
2022-06-28 07:57:51 -05:00
const fullVideo = await VideoModel.loadFull(video.id)
2022-05-25 08:18:29 -05:00
return federateVideoIfNeeded(fullVideo, false, undefined)
} catch (err) {
logger.warn('Cannot federate live after cleanup', { videoId: video.id, err })
}
}