90 أسطر
2.4 KiB
JavaScript
90 أسطر
2.4 KiB
JavaScript
'use strict';
|
|
|
|
const { PassThrough } = require('stream');
|
|
const ffmpeg = require('fluent-ffmpeg');
|
|
|
|
const os = require('os');
|
|
|
|
const platform = os.platform();
|
|
let ffmpegPath;
|
|
|
|
switch (platform) {
|
|
case 'darwin':
|
|
ffmpegPath = '/usr/local/bin/ffmpeg'; // macOS
|
|
break;
|
|
case 'linux':
|
|
ffmpegPath = '/usr/bin/ffmpeg'; // Linux
|
|
break;
|
|
case 'win32':
|
|
ffmpegPath = 'C:\\ffmpeg\\bin\\ffmpeg.exe'; // Windows
|
|
break;
|
|
default:
|
|
ffmpegPath = '/usr/bin/ffmpeg'; // Centos or others...
|
|
}
|
|
|
|
ffmpeg.setFfmpegPath(ffmpegPath);
|
|
|
|
console.log('FFmpeg', { platform, ffmpegPath });
|
|
|
|
class RtmpStreamer {
|
|
constructor(rtmpUrl, rtmpKey, socket) {
|
|
this.socket = socket;
|
|
this.rtmpUrl = rtmpUrl;
|
|
this.rtmpKey = rtmpKey;
|
|
this.stream = new PassThrough();
|
|
this.ffmpegStream = null;
|
|
this.initFFmpeg();
|
|
this.run = true;
|
|
}
|
|
|
|
initFFmpeg() {
|
|
this.ffmpegStream = ffmpeg()
|
|
.input(this.stream)
|
|
.inputOptions('-re')
|
|
.videoCodec('libx264')
|
|
.videoBitrate('3000k')
|
|
.size('1280x720')
|
|
.audioCodec('aac')
|
|
.audioBitrate('128k')
|
|
.outputOptions(['-f flv'])
|
|
.output(this.rtmpUrl)
|
|
.on('start', (commandLine) => console.info('ffmpeg command', { id: this.rtmpKey, cmd: commandLine }))
|
|
.on('error', (err, stdout, stderr) => {
|
|
if (!err.message.includes('Exiting normally')) {
|
|
console.error('FFmpeg error:', { id: this.rtmpKey, error: err.message });
|
|
this.socket.emit('error', err.message);
|
|
}
|
|
this.end();
|
|
})
|
|
.on('end', () => {
|
|
console.info('FFmpeg process ended', this.rtmpKey);
|
|
this.end();
|
|
})
|
|
.run();
|
|
}
|
|
|
|
write(data) {
|
|
if (this.stream) this.stream.write(data);
|
|
}
|
|
|
|
isRunning() {
|
|
return this.run;
|
|
}
|
|
|
|
end() {
|
|
if (this.stream) {
|
|
this.stream.end();
|
|
this.stream = null;
|
|
console.info('RTMP streaming stopped', this.rtmpKey);
|
|
}
|
|
if (this.ffmpegStream && !this.ffmpegStream.killed) {
|
|
this.ffmpegStream.kill('SIGTERM');
|
|
this.ffmpegStream = null;
|
|
console.info('FFMPEG closed successfully', this.rtmpKey);
|
|
}
|
|
this.run = false;
|
|
}
|
|
}
|
|
|
|
module.exports = RtmpStreamer;
|