Mirrors/cobalt
1
0
Fork 0
mirror of https://github.com/wukko/cobalt.git synced 2024-11-15 12:50:01 +00:00
Code Issues Projects Releases Packages Wiki Activity

stream/internal: use pipe() to handle internal streams

Browse source
This commit is contained in:
dumbmoron 2024-06-06 14:45:36 +00:00
parent fe7d4974e4
commit 85bed9aa74
No known key found for this signature in database
1 changed files with 3 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
src/modules/stream/internal.js
View file

@ -1,7 +1,7 @@
import { request } from 'undici'; import { request } from 'undici';
import { Readable } from 'node:stream'; import { Readable } from 'node:stream';
import { assert } from 'console'; import { assert } from 'console';
import { getHeaders } from './shared.js'; import { getHeaders, pipe } from './shared.js';
const CHUNK_SIZE = BigInt(8e6); // 8 MB const CHUNK_SIZE = BigInt(8e6); // 8 MB
const min = (a, b) => a < b ? a : b; const min = (a, b) => a < b ? a : b;
@ -66,8 +66,7 @@ async function handleYoutubeStream(streamInfo, res) {
if (headerValue) res.setHeader(headerName, headerValue); if (headerValue) res.setHeader(headerName, headerValue);
} }
stream.pipe(res); pipe(stream, res, () => res.end());
stream.on('error', () => res.end());
} catch { } catch {
res.end(); res.end();
} }
@ -97,8 +96,7 @@ export async function internalStream(streamInfo, res) {
if (req.statusCode < 200 || req.statusCode > 299) if (req.statusCode < 200 || req.statusCode > 299)
return res.end(); return res.end();
req.body.pipe(res); pipe(req.body, res, () => res.end());
req.body.on('error', () => res.end());
} catch { } catch {
streamInfo.controller.abort(); streamInfo.controller.abort();
} }