yetangitu/Peertube
1
0
Fork 0
mirror of https://github.com/Chocobozzz/PeerTube.git synced 2025-10-03 01:39:37 +02:00
Code Issues Releases Wiki Activity

More robust checkLiveSegmentHash

Browse source
This commit is contained in:
Chocobozzz 2025-07-28 09:46:56 +02:00
parent 121d9029b9
commit 06d9c7a13d
No known key found for this signature in database
GPG key ID: 583A612D890159BE
1 changed files with 19 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

24
packages/tests/src/shared/streaming-playlists.ts
View file

@ -1,6 +1,6 @@
/* eslint-disable @typescript-eslint/no-unused-expressions,@typescript-eslint/require-await */
import { getHLS, removeFragmentedMP4Ext, uuidRegex } from '@peertube/peertube-core-utils'
import { getHLS, removeFragmentedMP4Ext, uuidRegex, wait } from '@peertube/peertube-core-utils'
import {
FileStorage,
HttpStatusCode,
@ -61,14 +61,28 @@ export async function checkLiveSegmentHash (options: {
segmentName: string
hlsPlaylist: VideoStreamingPlaylist
withRetry?: boolean
currentRetry?: number
}) {
const { server, baseUrlSegment, videoUUID, segmentName, hlsPlaylist, withRetry = false } = options
const { server, baseUrlSegment, videoUUID, segmentName, hlsPlaylist, withRetry = false, currentRetry = 1 } = options
const command = server.streamingPlaylists
const segmentBody = await command.getFragmentedSegment({ url: `${baseUrlSegment}/${videoUUID}/${segmentName}`, withRetry })
const shaBody = await command.getSegmentSha256({ url: hlsPlaylist.segmentsSha256Url, withRetry })
try {
const segmentBody = await command.getFragmentedSegment({ url: `${baseUrlSegment}/${videoUUID}/${segmentName}`, withRetry: false })
const shaBody = await command.getSegmentSha256({ url: hlsPlaylist.segmentsSha256Url, withRetry: false })
expect(sha256(segmentBody)).to.equal(shaBody[segmentName])
expect(sha256(segmentBody)).to.equal(shaBody[segmentName])
} catch (err) {
if (!withRetry || currentRetry > 10) throw err
await wait(250)
return this.checkLiveSegmentHash({
...options,
withRetry,
currentRetry: currentRetry + 1
})
}
}
export async function checkPlaylistInfohash (options: {