Jerry/rc-plugin
1
0
Fork 0
mirror of https://github.com/Jerryplusy/rc-plugin.git synced 2025-10-14 16:19:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

🎈 pref: 优化直播切片，最大程度避免内存泄露

Browse Source
This commit is contained in:
zhiyu1998 2024-10-01 15:16:58 +08:00
parent 93a8966ff6
commit 44e3027f6c
1 changed files with 28 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
apps/tools.js
View File

@ -481,11 +481,18 @@ export class tools extends plugin {
* @param second * @param second
*/ */
async sendStreamSegment(e, stream_url, second = this.streamDuration) { async sendStreamSegment(e, stream_url, second = this.streamDuration) {
const outputFilePath = `${ this.getCurDownloadPath(e) }/stream_10s.flv`; const outputFilePath = `${ this.getCurDownloadPath(e) }/stream_${second}s.flv`;
await checkAndRemoveFile(outputFilePath); await checkAndRemoveFile(outputFilePath);
// 创建一个取消令牌
const CancelToken = axios.CancelToken;
const source = CancelToken.source();
try { try {
const response = await axios.get(stream_url, { responseType: 'stream' }); const response = await axios.get(stream_url, {
responseType: 'stream',
cancelToken: source.token,
});
logger.info("[R插件][发送直播流] 正在下载直播流..."); logger.info("[R插件][发送直播流] 正在下载直播流...");
const file = fs.createWriteStream(outputFilePath); const file = fs.createWriteStream(outputFilePath);
@ -493,13 +500,28 @@ export class tools extends plugin {
// 设置 streamDuration 秒后停止下载 // 设置 streamDuration 秒后停止下载
setTimeout(async () => { setTimeout(async () => {
logger.info(`[R插件][发送直播流] 直播下载 ${ this.streamDuration } 秒钟到，停止下载！`); logger.info(`[R插件][发送直播流] 直播下载 ${ second } 秒钟到，停止下载！`);
response.data.destroy(); // 销毁流 // 取消请求
source.cancel('下载时间到，停止请求');
response.data.unpipe(file); // 取消管道连接
file.end(); // 结束写入
await this.sendVideoToUpload(e, outputFilePath); await this.sendVideoToUpload(e, outputFilePath);
file.close();
}, second * 1000); }, second * 1000);
// 监听请求被取消的情况
response.data.on('error', (err) => {
if (axios.isCancel(err)) {
logger.info('请求已取消:', err.message);
} else {
logger.error('下载过程中发生错误:', err.message);
}
});
} catch (error) { } catch (error) {
logger.error(`下载失败: ${ error.message }`); if (axios.isCancel(error)) {
logger.info('请求已取消:', error.message);
} else {
logger.error(`下载失败: ${ error.message }`);
}
await fs.promises.unlink(outputFilePath); // 下载失败时删除文件 await fs.promises.unlink(outputFilePath); // 下载失败时删除文件
} }
} }