admin/cobalt
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

stream: wrap abort controller in try-catch

Browse Source
This commit is contained in:
dumbmoron
2023-11-05 22:09:54 +00:00
parent 1508a0bff4
commit ed646b826d
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/modules/stream/types.js
View File

@@ -6,6 +6,10 @@ import { request } from "undici";
import { create as contentDisposition } from "content-disposition-header"; import { create as contentDisposition } from "content-disposition-header";
import { AbortController } from "abort-controller" import { AbortController } from "abort-controller"
function closeRequest(controller) {
try { controller.abort() } catch {}
}
function closeResponse(res) { function closeResponse(res) {
if (!res.headersSent) res.sendStatus(500); if (!res.headersSent) res.sendStatus(500);
return res.destroy(); return res.destroy();
@@ -21,7 +25,7 @@ function killProcess(p) {
export async function streamDefault(streamInfo, res) { export async function streamDefault(streamInfo, res) {
const abortController = new AbortController(); const abortController = new AbortController();
const shutdown = () => (abortController.abort(), closeResponse(res)); const shutdown = () => (closeRequest(abortController), closeResponse(res));
try { try {
const filename = streamInfo.isAudioOnly ? `${streamInfo.filename}.${streamInfo.audioFormat}` : streamInfo.filename; const filename = streamInfo.isAudioOnly ? `${streamInfo.filename}.${streamInfo.audioFormat}` : streamInfo.filename;
@@ -45,7 +49,7 @@ export async function streamDefault(streamInfo, res) {
export async function streamLiveRender(streamInfo, res) { export async function streamLiveRender(streamInfo, res) {
let abortController = new AbortController(), process; let abortController = new AbortController(), process;
const shutdown = () => (abortController.abort(), killProcess(process), closeResponse(res)); const shutdown = () => (closeRequest(abortController), killProcess(process), closeResponse(res));
try { try {
if (streamInfo.urls.length !== 2) return shutdown(); if (streamInfo.urls.length !== 2) return shutdown();