Merge pull request #767 from sunhui2013/master

增加支持tcp收流暂停/恢复流检测功能

server/WebApi.cpp

@@ -851,7 +851,7 @@ void installWebApi() {
             return;
         }
         auto server = it->second;
-        server->pauseRtpCheck();
+        server->pauseRtpCheck(allArgs["stream_id"]);
         val["hit"] = 1;
     });
 
@@ -866,7 +866,7 @@ void installWebApi() {
             return;
         }
         auto server = it->second;
-        server->resumeRtpCheck();
+        server->resumeRtpCheck(allArgs["stream_id"]);
         val["hit"] = 1;
     });
 

src/Rtp/RtpServer.cpp

@@ -160,15 +160,30 @@ uint16_t RtpServer::getPort() {
     return _udp_server ? _udp_server->get_local_port() : 0;
 }
 
-void RtpServer::pauseRtpCheck(){    
+void RtpServer::pauseRtpCheck(const string &stream_id){    
     if(_rtp_process)
         _rtp_process->setStopCheckRtp(true);
+    else{
+        if(!stream_id.empty()){
+            auto rtp_process = RtpSelector::Instance().getProcess(stream_id,false);
+            if(rtp_process)
+                rtp_process->setStopCheckRtp(true);
+        }
+    }
+    
 }
 
-void RtpServer::resumeRtpCheck(){
+void RtpServer::resumeRtpCheck(const string &stream_id){
     if(_rtp_process)
         _rtp_process->setStopCheckRtp(false);
-
+    else{
+        //解决不指定流或者TCP收流无法暂停
+        if(!stream_id.empty()){
+        auto rtp_process = RtpSelector::Instance().getProcess(stream_id,false);
+        if(rtp_process)
+            rtp_process->setStopCheckRtp(false);
+        }
+    }
 }
 
 }//namespace mediakit

src/Rtp/RtpServer.h

@@ -60,12 +60,12 @@ public:
     /**
      * 暂停Rtp服务的RTP流检测
      */
-    void pauseRtpCheck();
+    void pauseRtpCheck(const string &stream_id = "");
 
     /**
      * 恢复Rtp服务的RTP流检测
      */
-    void resumeRtpCheck();
+    void resumeRtpCheck(const string &stream_id = "");
 
 protected:
     Socket::Ptr _udp_server;