5 Star 35 Fork 13

nygula / Node-Media-Server

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
node_fission_server.js 2.86 KB
一键复制 编辑 原始数据 按行查看 历史
Chen Mingliang 提交于 2020-07-16 23:18 . f
//
// Created by Chen Mingliang on 20/7/16.
// illuspas[a]msn.com
// Copyright (c) 2020 Nodemedia. All rights reserved.
//
const Logger = require('./node_core_logger');
const NodeFissionSession = require('./node_fission_session');
const context = require('./node_core_ctx');
const { getFFmpegVersion, getFFmpegUrl } = require('./node_core_utils');
const fs = require('fs');
const _ = require('lodash');
const mkdirp = require('mkdirp');
class NodeFissionServer {
constructor(config) {
this.config = config;
this.fissionSessions = new Map();
}
async run() {
try {
mkdirp.sync(this.config.http.mediaroot);
fs.accessSync(this.config.http.mediaroot, fs.constants.W_OK);
} catch (error) {
Logger.error(`Node Media Fission Server startup failed. MediaRoot:${this.config.http.mediaroot} cannot be written.`);
return;
}
try {
fs.accessSync(this.config.fission.ffmpeg, fs.constants.X_OK);
} catch (error) {
Logger.error(`Node Media Fission Server startup failed. ffmpeg:${this.config.fission.ffmpeg} cannot be executed.`);
return;
}
let version = await getFFmpegVersion(this.config.fission.ffmpeg);
if (version === '' || parseInt(version.split('.')[0]) < 4) {
Logger.error(`Node Media Fission Server startup failed. ffmpeg requires version 4.0.0 above`);
Logger.error('Download the latest ffmpeg static program:', getFFmpegUrl());
return;
}
context.nodeEvent.on('postPublish', this.onPostPublish.bind(this));
context.nodeEvent.on('donePublish', this.onDonePublish.bind(this));
Logger.log(`Node Media Fission Server started, MediaRoot: ${this.config.http.mediaroot}, ffmpeg version: ${version}`);
}
onPostPublish(id, streamPath, args) {
let regRes = /\/(.*)\/(.*)/gi.exec(streamPath);
let [app, name] = _.slice(regRes, 1);
for (let task of this.config.fission.tasks) {
regRes = /(.*)\/(.*)/gi.exec(task.rule);
let [ruleApp, ruleName] = _.slice(regRes, 1);
if ((app === ruleApp || ruleApp === "*") && (name === ruleName || ruleName === "*")) {
let s = context.sessions.get(id);
if (s.isLocal && name.split('_')[1]) {
continue;
}
let conf = task;
conf.ffmpeg = this.config.fission.ffmpeg;
conf.mediaroot = this.config.http.mediaroot;
conf.rtmpPort = this.config.rtmp.port;
conf.streamPath = streamPath;
conf.streamApp = app;
conf.streamName = name;
conf.args = args;
let session = new NodeFissionSession(conf);
this.fissionSessions.set(id, session);
session.on('end', () => {
this.fissionSessions.delete(id);
});
session.run();
}
}
}
onDonePublish(id, streamPath, args) {
let session = this.fissionSessions.get(id);
if (session) {
session.end();
}
}
}
module.exports = NodeFissionServer;
1
https://gitee.com/nygula/Node-Media-Server.git
git@gitee.com:nygula/Node-Media-Server.git
nygula
Node-Media-Server
Node-Media-Server
master

搜索帮助