增加指令本地备份功能

GB28181Session.js

@@ -4,7 +4,8 @@ const SDP = require('./sdp/parser');
 const Logger = require('./core/logger');
 const context = require('./core/ctx');
 const NtvAuthModule = require('./NtvAuthModule');
-
+const fs = require('fs');
+const cache_path = "/etc/noveltv/mserver/cache/";
 class NodeSipSession {
     constructor(config, userid, remote, uas) {
         
@@ -150,6 +151,7 @@ class NodeSipSession {
         });
         
         //ntv 尝试自动开启视频，只开启第一个通道。 TODO: 对于硬盘录像机，应可以开启多个通道
+        //如果要自动开启视频，把这个注释打开
         await this.startRealPlay(firstChid,auto_play);
     }
 
@@ -786,6 +788,50 @@ class NodeSipSession {
         });
     }
 
+//扩展函数////////////////////////////
+
+//ssrc生成规则
+genSSRC(id,chid){
+    let sub1 = 10000 + parseInt(id.substring(15));
+    let sub2 = chid.substring(15);
+    let ssrc = '' + sub1 + sub2;
+    return ssrc;
+}
+//设置关闭视频预览的指令缓存，如果要缓存更多指令，可以追加参数  NTV ADD
+setSessionData(key,json){
+    this.dialogs[key] = json;
+    var filename = cache_path + key;
+    fs.writeFile(filename, JSON.stringify(json), function (error) {
+        if (error) {
+            Logger.error('Failed to write cache file !');
+        }
+      });
+}
+//查询关闭视频预览的指令缓存  NTV ADD
+getSessionData(key){
+    var filename = cache_path + key;
+    if(this.dialogs[key]){
+        return this.dialogs[key];
+    }else if (fs.existsSync(filename)){
+        var data = fs.readFileSync(filename,'utf-8');
+        Logger.log('Get string from cache OK!');
+        return JSON.parse(data);
+    }
+
+    return null;
+}
+
+delSessionData(key){
+    var filename = cache_path + key;
+    delete this.dialogs[key];
+    fs.unlink(filename, (err) => {
+        if (err){
+            Logger.error('Failed to unlink file : ' + filename);
+        } 
+      });
+}
+//扩展函数////////////////////////////
+
     //预览 channelId 通道国标编码
     sendRealPlayMessage(channelId, rhost, rport, mode) {
         Logger.log("Start real play " + this.id + ":" + channelId + " " + rhost + ":" + rport + " mode:" + mode);
@@ -797,6 +843,7 @@ class NodeSipSession {
 
             let findssrc = "";
 
+            /** 替换 NTV
             for (var key in this.dialogs) {
                 let session = this.dialogs[key];
                 if (session.bye && session.port == rport && session.host == rhost && session.channelId == channelId && session.play == 'realplay') {
@@ -805,22 +852,26 @@ class NodeSipSession {
                     break;
                 }
             }
+            */
+            let ssrc = this.genSSRC(this.id,channelId)
+            var key  = ssrc + "p";
+            let session = this.dialogs[key] || this.getSessionData(key);
+            if(session){
+               isFinded = true;
+               findssrc = session.ssrc;
+            }
 
             //己存在会话,同一个流媒体不需要重复请求
             if (isFinded) {
+                this.playmode='realplay';
                 result.data = { ssrc: findssrc };
                 resolve(result);
+                Logger.log("Live stream is playing now, needn't send play cmd again!");
                 return;
             }
             //0: udp,1:tcp/passive ,2:tcp/active
             let selectMode = mode || 0;
 
-            //ssrc is here  -- by ntv wangjian
-            //let ssrc = "0" + channelId.substring(16, 20) + channelId.substring(3, 8);
-            let sub1 = 10000 + parseInt(this.id.substring(15));
-            let sub2 = channelId.substring(15);
-            let ssrc = '' + sub1 + sub2;
-
             let host = rhost || "127.0.0.1";
 
             let port = rport || 9200;
@@ -912,10 +963,10 @@ class NodeSipSession {
                                     });
 
                                     //会话标识
-                                    let key = [response.headers['call-id'], response.headers.from.params.tag, response.headers.to.params.tag].join(':');
+                                    //let key = [response.headers['call-id'], response.headers.from.params.tag, response.headers.to.params.tag].join(':');
 
                                     //创建会话
-                                    if (!that.dialogs[key]) {
+                                    //if (!that.dialogs[key]) {
                                         // 断开会话请求
                                         let byeRequest = {
                                             method: 'BYE',
@@ -929,11 +980,12 @@ class NodeSipSession {
                                             }
                                         }
 
-                                        that.dialogs[key] = { channelId: channelId, ssrc: ssrc, host: host, port: port, bye: byeRequest, play: 'realplay' };
-                                        
+                                        //that.dialogs[key] = { channelId: channelId, ssrc: ssrc, host: host, port: port, bye: byeRequest, play: 'realplay' };
+                                        let bayCmd        = { channelId: channelId, ssrc: ssrc, host: host, port: port, bye: byeRequest, play: 'realplay' };
+                                        that.setSessionData(key,bayCmd);
                                         //console.log("BYE INFO:");
                                         //console.log(that.dialogs[key]);
-                                    }
+                                    //}
 
                                     result.data = { ssrc: ssrc };
 
@@ -945,7 +997,6 @@ class NodeSipSession {
                                         }else{
                                             Logger.log("视频开启状态已上报失败！" + data.err_desc);
                                         }
-                                        
                                     });
 
                                     resolve(result);
@@ -966,14 +1017,17 @@ class NodeSipSession {
         return new Promise((resolve, reject) => {
             let result = { result: false, message: '没有找到视频通道！' };
 
-            for (var key in this.dialogs) {
+            //for (var key in this.dialogs) {
                 //搜索满足条件的会话
-                let session = this.dialogs[key];
+                //let session = this.dialogs[key];
                 //ntv  注释掉多余条件，因为数据类型不同导致找不到
-                if (session.bye && session.port == rport && session.host == rhost && session.channelId == channelId && session.play == 'realplay') {
+                //if (session.bye && session.port == rport && session.host == rhost && session.channelId == channelId && session.play == 'realplay') {
                     //ntv 挪到回调中，TODO 这个指令发不成功！
                     //context.nodeEvent.emit('stopPlayed', session.ssrc);
                     
+                let key    = this.genSSRC(this.id,channelId) + "p";
+                let session= this.getSessionData(key);
+                if(session){ 
                     //add by ntv  强制更新uri地址，因为终端地址会变。
                     session.bye.uri = 'sip:' + this.id + '@' + this.host + ':' + this.port;
 
@@ -985,7 +1039,8 @@ class NodeSipSession {
                             ntv remove rtp连接断开时再emit这条消息，放到了stream/session中执行
                             */
                             //context.nodeEvent.emit('stopPlayed', session.ssrc);
-                            delete this.dialogs[key];
+                            //delete this.dialogs[key];
+                            this.delSessionData(key);
                             this.authModule.play_status(this.id,0,(data)=>{
                                 if(data.code==0){
                                     Logger.log("视频关闭状态已上报！");
@@ -1002,8 +1057,9 @@ class NodeSipSession {
 
                     result.result = true;
                     result.message = 'OK';
+
                 }
-            }
+            //}
 
             resolve(result);
         });

NtvAuthModule.js

@@ -7,6 +7,8 @@
 
 const http = require('http');
 const qs   = require('querystring');
+const url  = require("url");
+const request = require('request');
 
 class NtvAuthModule {
 	
@@ -43,6 +45,16 @@ class NtvAuthModule {
 		callback(obj);
 	}
 
+	requestCall(url,callback){
+		var that = this;
+		request(url, function(error, response, body) {
+			if (!error && response.statusCode == 200) {
+				that.parseData(body,callback);
+			}else{
+				callback(that.errorMsg('无法连接业务服务器！'));
+			}
+		});
+	}
 	/**
 	 * 请求远程接口
 	 * @param {接口的action} action 
@@ -81,7 +93,7 @@ class NtvAuthModule {
 
 		req.on('error', function(e) {
 			if(typeof(callback)=="function"){
-				that.callback(that.errorMsg(5,'无法连接业务服务器！'));
+				callback(that.errorMsg('无法连接业务服务器！'));
 			}
 		});
 
@@ -128,6 +140,19 @@ class NtvAuthModule {
 		var port = config.port;
 		this.postCall(host,port,path,paras,callback);
 	}
+
+	/**
+	 * 废弃
+	 * 调用RTP_SERVER接口，启动rtmp推流 
+	 * http://__RTP_SERVER/index/api/addStreamPusherProxy?secret=035c73f7-bb6b-4889-a715-d9eb2d1925cc&schema=rtmp&vhost=__defaultVhost__&app=rtp
+	 * &stream=__STREAM&dst_url=rtmp://__RTMP_SERVER/show2/__STREAM
+	*/
+	start_rtmp(api,stream,callback){
+		stream = stream.toLocaleUpperCase();
+		api    = api.replace(/__STREAM/g, stream);
+		console.log("query rtmp push api : "+ api);
+		this.requestCall(api,callback);
+	}
 }
 
 module.exports = NtvAuthModule;