基于node-media-server实现的http-flv转推流程序

    以前给网页展示的rtsp流，使用crtmpserver做了稍加改动后转推过去的，但随着flash的淘汰，videojs这个项目组也打算在这次升级中淘汰掉，改用flvjs。

    问题就是flvjs不支持rtmp，而要在crtmpserver上添加新的转流代码，一是实力不够，二是时间不够，于是选了一个成熟的http-flv框架，稍加修改。

    大概流程就会是，网页要看一个rtsp的流视频，按着约定构造一个http://serverIp:serverport/live/rtsp的base64编码.flv，

    例如：http://127.0.0.1:8000/live/cnRzcDovL2FkbWluOmFkbWluMTIzQDEwLjIuMy4xMDE=.flv

    这样，在转推流的server收到这个请求，会截取到这个base64，解码后调用ffmpeg做一个推流，推个rtmp到流回来。

    应为nodejs可以注册事件，而且可以在链接connect和disconnect的时候对ffmpeg的调用经行管理。

const { globalAgent } = require('http');
const NodeMediaServer  = require('node-media-server');
const numCPUs = require('os').cpus().length;
var HashMap = require('hashmap').HashMap;

const config = {
  rtmp: {
    port: 1935,
    chunk_size: 60000,
    gop_cache: true,
    ping: 60,
    ping_timeout: 30
  },

  http: {
    port: 8000,
    allow_origin: '*'
  },
  cluster: {
    num: numCPUs
  }
}

var nms = new NodeMediaServer(config)
nms.run();

var pmap = new HashMap();
var cmap = new HashMap();
const exec = require('child_process').execFile;

function strsub( str ) {
  var list = str.split( "/" );
  return list[2]
}

nms.on('preConnect', (id, args)=>{
  console.log('[NodeEvent on preConnect]', `id=${id} args=${JSON.stringify(args)}`);
  jobj = JSON.stringify(args);
  console.log(args['streamPath']);
  if ( args['streamPath'] != undefined ) {
    var base64 = strsub( args['streamPath']) ;
    var tmp = new Buffer( base64, 'base64' );
    var rtsp = tmp.toString();
    console.log( base64, rtsp );
    if ( cmap.has( base64 ) ) {
      num = cmap.get( base64 );
      num = num + 1;
      cmap.set( base64, num );
    } else {
      pro = exec( "ffmpeg.exe", ["-rtsp_transport", "tcp", "-i", rtsp, "-vcodec", "copy", "-acodec", "aac", "-f", "flv", "rtmp://127.0.0.1" + args['streamPath'] ],function(err,stdout,stderr){
        if(err){
            console.error(err);
        }
      });
      cmap.set( base64, 1 );
      pmap.set( base64, pro.pid );
    }
  }
});

nms.on('doneConnect', (id, args) => {
  console.log('[NodeEvent on doneConnect]', `id=${id} args=${JSON.stringify(args)}`);
  if ( args['streamPath'] != undefined ) {
    var base64 = strsub( args['streamPath']) ;
    console.log( base64 );
    if ( cmap.has( base64 ) ) {
      num = cmap.get( base64 );
      pro = pmap.get( base64 );
      num -= 1;
      cmap.set( base64, num );
      console.log( num, pro );
      if ( num == 0 ) {
        exec( "taskkill", [ "/F", "/PID", pro ] );
        cmap.remove( base64 );
        pmap.remove( base64 );
      }
    }
  }
});